IM服务端

hefeixia

2021-02-18 5b8c95c760840f09910730943b21391e47187315

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
package chat.handler;
 
import java.util.Date;
import java.util.List;
 
import chat.module.ChannelBucket;
import chat.module.entity.Channel;
import chat.module.entity.MessageRecord;
import chat.module.entity.PublishOperator;
import chat.server.call.CallObject;
import cn.wildfirechat.proto.WFCMessage;
 
public class ChannelHandler extends CallObject {
 
    private static ChannelBucket channelBucket;
    
    static {
        channelBucket = ChannelBucket.getInstance();
    }
    
    @Override
    protected void publishMethod() {
        addOneMethod("create");
        addOneMethod("destroy");
        addOneMethod("getInfo");
        addOneMethod("getRoomList");
        addOneMethod("join");
        addOneMethod("leave");
        addOneMethod("kickoff");
        addOneMethod("pull");
        addOneMethod("publish");
    }
    
    public void create() {
        String channelName = dataPool.getString("channelName");
        
        //1. create 
//        channelBucket.createOne(channelName, user);
        
        //2. result
        resultPool.success();
    }
    
    public void destroy() {
        String channelId = dataPool.getString("channelId");
        
        Channel channel = channelBucket.getOne(channelId);
        
        if (channel == null) {
            return;
        }
        
        //1. notify members
        channel.notify(PublishOperator.Destroy, user);
        
        //2. delete room
        channelBucket.deleteOne(channelId);
        
        //3. result
        resultPool.success();
    }
    
    public void pullChannel() throws Exception {
        //头
        WFCMessage.PullChannelInfo request = (WFCMessage.PullChannelInfo) dataPool.getWFCMessage(WFCMessage.PullChannelInfo.class);
        
        //尾
        WFCMessage.ChannelInfo info = null;//m_messagesStore.getChannelInfo(request.getChannelId());
        byte[] resBytes = info.toByteArray();
        resultPool.setBytes(resBytes);
    }
 
    public void getInfo() {
        String channelId = dataPool.getString("channelId");
        Channel channel = channelBucket.getOne(channelId);
        
        resultPool.add(channel);
    }
    
    public void getChannelList() {
        List<Channel> channeList = channelBucket.getAll();
        resultPool.add(channeList);
    }
    
    public void subscribe() {
        String channelId = dataPool.getString("channelId");
        
        Channel channel = channelBucket.getOne(channelId);
        
        if (channel == null) {
            return;
        }
        
        //1. add subscriber
        channel.join(channelId, user);
    }
    
    public void unsubscribe() {
        String channelId = dataPool.getString("channelId");
        
        Channel channel = channelBucket.getOne(channelId);
        
        if (channel == null) {
            return;
        }
        
        //1. delete member
        channel.leave(user);
    }
    
    protected void pull() {
        String channelId = dataPool.getString("channelId");
        long timeStamp = Long.parseLong(dataPool.getString("timePoint"));
        
        Channel channel = channelBucket.getOne(channelId);
        
        if (channel == null) {
            return;
        }
        
        //1. get message list
        List<MessageRecord> messageList = channel.getMessageList(timeStamp);
        
        //2. result
        resultPool.add(messageList);
    }
    
    protected void publish() {
        //1.
        String channelId = dataPool.getString("channelId");
        MessageRecord record = dataPool.getMessageRecord();
        
        Channel channel = channelBucket.getOne(channelId);
        
        if (channel == null) {
            return;
        }
        
        //1. add to message list
        //channel.publishOneMessage(record, user, true, null);
        
        //2. publish
        channel.notify(PublishOperator.PublishMessage, record, user);
        
        //3. result
        resultPool.success();
    }
}