统一消息系统

小明：嘿，小李，最近我在研究一个项目，是关于消息管理系统的，你对这个有了解吗？

小李：哦，消息管理系统啊，我之前接触过一些。它主要是用来处理、存储和分发消息的系统，对吧？比如像即时通讯、邮件通知、日志记录这些场景。

小明：没错！我们团队正在开发一个轻量级的消息管理系统，用于内部各个模块之间的通信。不过，我有点担心版权问题，听说这个系统需要申请软著。

小李：对的，如果你的系统是原创的，而且具备一定的技术含量，那么申请软件著作权（简称“软著”）是很必要的。它可以保护你的知识产权，防止别人盗用。

小明：那我应该怎么做呢？有没有什么具体的步骤或者需要注意的地方？

小李：首先，你需要准备一份完整的软件文档，包括功能说明、技术实现、用户手册等。然后，提交到国家版权局进行登记。不过，这可能不是你现在的重点，你现在更关心的是怎么实现这个系统吧？

小明：是的，我想先把这个系统做出来，然后再考虑软著的问题。那你能帮我看看怎么写一个简单的消息管理系统吗？最好能展示出核心代码。

小李：当然可以！我们可以用Java来实现一个基础版本。这个系统主要包含几个部分：消息的生成、存储、发送和接收。我们可以使用队列结构来处理消息。

小明：听起来不错。那具体怎么实现呢？

小李：我们可以定义一个Message类，用来表示消息的内容。然后创建一个MessageQueue类，用来管理消息队列。最后再写一个MessageHandler类，用来处理消息的发送和接收。

小明：好的，那我们先从Message类开始写吧。

小李：没问题，以下是Message类的代码：

public class Message {
    private String content;
    private String sender;
    private String receiver;
    private long timestamp;

    public Message(String content, String sender, String receiver) {
        this.content = content;
        this.sender = sender;
        this.receiver = receiver;
        this.timestamp = System.currentTimeMillis();
    }

    // Getter and Setter methods
    public String getContent() {
        return content;
    }

    public void setContent(String content) {
        this.content = content;
    }

    public String getSender() {
        return sender;
    }

    public void setSender(String sender) {
        this.sender = sender;
    }

    public String getReceiver() {
        return receiver;
    }

    public void setReceiver(String receiver) {
        this.receiver = receiver;
    }

    public long getTimestamp() {
        return timestamp;
    }
}
    

小明：这个类看起来挺简单的，但确实能保存消息的基本信息。

小李：是的，接下来我们来写消息队列类，也就是MessageQueue。

小明：那这个类有什么功能呢？

小李：MessageQueue的作用是管理消息的存储和分发。我们可以使用一个List来存储消息，然后提供添加、获取和删除消息的方法。

小明：明白了，那代码应该怎么写呢？

小李：下面是MessageQueue的实现代码：

import java.util.ArrayList;
import java.util.List;

public class MessageQueue {
    private List messages = new ArrayList<>();

    public void addMessage(Message message) {
        messages.add(message);
    }

    public Message getMessageByIndex(int index) {
        if (index >= 0 && index < messages.size()) {
            return messages.get(index);
        } else {
            return null;
        }
    }

    public void removeMessageByIndex(int index) {
        if (index >= 0 && index < messages.size()) {
            messages.remove(index);
        }
    }

    public int getMessageCount() {
        return messages.size();
    }
}
    

小明：这个类的功能很明确，能管理消息的添加和获取。那接下来是不是要处理消息的发送和接收？

小李：对，我们可以创建一个MessageHandler类，负责将消息发送给对应的接收者。这里我们可以模拟一个简单的消息处理机制。

小明：那代码怎么写呢？

小李：以下是MessageHandler类的示例代码：

public class MessageHandler {
    private MessageQueue queue;

    public MessageHandler(MessageQueue queue) {
        this.queue = queue;
    }

    public void sendMessage(String content, String sender, String receiver) {
        Message message = new Message(content, sender, receiver);
        queue.addMessage(message);
        System.out.println("消息已发送: " + message.getContent());
    }

    public void receiveMessages(String receiver) {
        for (int i = 0; i < queue.getMessageCount(); i++) {
            Message message = queue.getMessageByIndex(i);
            if (message != null && message.getReceiver().equals(receiver)) {
                System.out.println("收到消息: " + message.getContent() + " (来自: " + message.getSender() + ")");
                queue.removeMessageByIndex(i);
                i--; // 因为移除后索引会变化，所以需要调整
            }
        }
    }
}
    

小明：这段代码看起来能实现消息的发送和接收功能，但有没有什么优化的空间？

小李：确实有一些地方可以改进。比如，目前的receiveMessages方法是直接遍历整个队列，效率不高。如果消息数量很大，可能会导致性能问题。我们可以引入一个更高效的结构，比如使用Map来按接收者分类存储消息。

小明：那我们可以修改一下MessageQueue的结构，用Map来存储不同接收者的消息，这样在接收时就不用每次都遍历整个队列了。

小李：没错，这是一个很好的优化方向。那我们来修改一下MessageQueue类，使其支持按接收者分类存储消息。

小明：那新的MessageQueue类该怎么写呢？

小李：我们可以使用一个Map>来存储消息，其中键是接收者的名称，值是对应的消息列表。

小明：那代码应该是这样的：

import java.util.*;

public class MessageQueue {
    private Map> messagesByReceiver = new HashMap<>();

    public void addMessage(Message message) {
        String receiver = message.getReceiver();
        if (!messagesByReceiver.containsKey(receiver)) {
            messagesByReceiver.put(receiver, new ArrayList<>());
        }
        messagesByReceiver.get(receiver).add(message);
    }

    public List getMessagesByReceiver(String receiver) {
        return messagesByReceiver.getOrDefault(receiver, new ArrayList<>());
    }

    public void clearMessagesByReceiver(String receiver) {
        messagesByReceiver.remove(receiver);
    }

    public int getTotalMessageCount() {
        int count = 0;
        for (List list : messagesByReceiver.values()) {
            count += list.size();
        }
        return count;
    }
}
    

小明：这个版本的MessageQueue确实更高效了，特别是当有很多消息的时候，只需要根据接收者来查找即可。

小李：没错，这样也提高了系统的可扩展性。现在，我们来看一下MessageHandler类是否也需要修改。

小明：是的，原来的receiveMessages方法可能需要重新设计，因为它不再需要遍历整个队列，而是可以直接获取特定接收者的消息。

小李：对，我们可以修改MessageHandler类如下：

public class MessageHandler {
    private MessageQueue queue;

    public MessageHandler(MessageQueue queue) {
        this.queue = queue;
    }

    public void sendMessage(String content, String sender, String receiver) {
        Message message = new Message(content, sender, receiver);
        queue.addMessage(message);
        System.out.println("消息已发送: " + message.getContent());
    }

    public void receiveMessages(String receiver) {
        List messages = queue.getMessagesByReceiver(receiver);
        for (Message message : messages) {
            System.out.println("收到消息: " + message.getContent() + " (来自: " + message.getSender() + ")");
        }
        queue.clearMessagesByReceiver(receiver); // 收到后清空
    }
}
    

小明：这样就更合理了，不需要每次都遍历所有消息，直接根据接收者获取即可。

小李：是的，这样也更符合实际应用场景。现在，我们可以测试一下这个系统。

小明：那我们写一个测试类来验证一下功能吧。

小李：好的，以下是一个简单的测试类示例：

public class MessageSystemTest {
    public static void main(String[] args) {
        MessageQueue queue = new MessageQueue();
        MessageHandler handler = new MessageHandler(queue);

        // 发送消息
        handler.sendMessage("你好！", "Alice", "Bob");
        handler.sendMessage("今天天气不错！", "Charlie", "Bob");
        handler.sendMessage("下午三点开会", "David", "Eve");

        // 接收消息
        System.out.println("Bob 收到的消息：");
        handler.receiveMessages("Bob");

        System.out.println("\nEve 收到的消息：");
        handler.receiveMessages("Eve");
    }
}
    

小明：运行结果应该显示Bob收到了两条消息，Eve收到了一条消息。

小李：没错，这就是一个简单的消息管理系统的实现。当然，这只是基础版本，实际应用中可能还需要加入线程安全、持久化、分布式等功能。

小明：是的，但我现在只是想先有一个原型，之后再逐步完善。

小李：很好，那你现在可以开始编写代码了。另外，记得在系统完成后申请软著，以保护你的劳动成果。

小明：明白了，谢谢你的帮助，小李！

小李：不客气，有问题随时问我！