【消息队列开发】虚拟主机设计——操作交换机

public VirtualHost(String name) {
    this.virtualHostName = name;
    // 对于 MemoryDataCenter 来说, 不需要额外的初始化操作的. 只要对象 new 出来就行了
    // 但是, 针对 DiskDataCenter 来说, 则需要进行初始化操作. 建库建表和初始数据的设定.
    diskDataCenter.init();
    // 另外还需要针对硬盘的数据, 进行恢复到内存中.
    try {
        memoryDataCenter.recovery(diskDataCenter);
    } catch (IOException | MqException | ClassNotFoundException e) {
        e.printStackTrace();
        System.out.println("[VirtualHost] 恢复内存数据失败!");
    }
}

// 创建交换机
// 如果交换机不存在, 就创建. 如果存在, 直接返回.
// 返回值是 boolean. 创建成功, 返回 true. 失败返回 false
public boolean exchangeDeclare(String exchangeName, ExchangeType exchangeType, boolean durable, boolean autoDelete,
                               Map<String, Object> arguments) {
    // 把交换机的名字, 加上虚拟主机作为前缀.
    exchangeName = virtualHostName + exchangeName;
    try {
        synchronized (exchangeLocker) {
            // 1. 判定该交换机是否已经存在. 直接通过内存查询.
            Exchange existsExchange = memoryDataCenter.getExchange(exchangeName);
            if (existsExchange != null) {
                // 该交换机已经存在!
                System.out.println("[VirtualHost] 交换机已经存在! exchangeName=" + exchangeName);
                return true;
            }
            // 2. 真正创建交换机. 先构造 Exchange 对象
            Exchange exchange = new Exchange();
            exchange.setName(exchangeName);
            exchange.setType(exchangeType);
            exchange.setDurable(durable);
            exchange.setAutoDelete(autoDelete);
            exchange.setArguments(arguments);
            // 3. 把交换机对象写入硬盘
            if (durable) {
                diskDataCenter.insertExchange(exchange);
            }
            // 4. 把交换机对象写入内存
            memoryDataCenter.insertExchange(exchange);
            System.out.println("[VirtualHost] 交换机创建完成! exchangeName=" + exchangeName);
            // 上述逻辑, 先写硬盘, 后写内存. 目的就是因为硬盘更容易写失败. 如果硬盘写失败了, 内存就不写了.
            // 要是先写内存, 内存写成功了, 硬盘写失败了, 还需要把内存的数据给再删掉. 就比较麻烦了.
        }
        return true;
    } catch (Exception e) {
        System.out.println("[VirtualHost] 交换机创建失败! exchangeName=" + exchangeName);
        e.printStackTrace();
        return false;
    }
}

// 删除交换机
public boolean exchangeDelete(String exchangeName) {
    exchangeName = virtualHostName + exchangeName;
    try {
        synchronized (exchangeLocker) {
            // 1. 先找到对应的交换机.
            Exchange toDelete = memoryDataCenter.getExchange(exchangeName);
            if (toDelete == null) {
                throw new MqException("[VirtualHost] 交换机不存在无法删除!");
            }
            // 2. 删除硬盘上的数据
            if (toDelete.isDurable()) {
                diskDataCenter.deleteExchange(exchangeName);
            }
            // 3. 删除内存中的交换机数据
            memoryDataCenter.deleteExchange(exchangeName);
            System.out.println("[VirtualHost] 交换机删除成功! exchangeName=" + exchangeName);
        }
        return true;
    } catch (Exception e) {
        System.out.println("[VirtualHost] 交换机删除失败! exchangeName=" + exchangeName);
        e.printStackTrace();
        return false;
    }
}