add topic store for mongo

oikomi · Jun 3, 2015 · 65a8a1c · 65a8a1c
1 parent 8adbd6e
commit 65a8a1c
Show file tree

Hide file tree

Showing 4 changed files with 48 additions and 1 deletion.
diff --git a/base/base.go b/base/base.go
@@ -31,7 +31,7 @@ var ChannleList []string
 
 func init() {
 	ChannleList = []string{protocol.SYSCTRL_CLIENT_STATUS, protocol.SYSCTRL_TOPIC_STATUS, protocol.SYSCTRL_TOPIC_SYNC, 
-		protocol.SYSCTRL_SEND, protocol.SYSCTRL_MONITOR, protocol.STORE_CLIENT_INFO}
+		protocol.SYSCTRL_SEND, protocol.SYSCTRL_MONITOR, protocol.STORE_CLIENT_INFO, protocol.STORE_TOPIC_INFO}
 }
 
 type ChannelState struct {

diff --git a/msg_server/proto_proc.go b/msg_server/proto_proc.go
@@ -301,6 +301,27 @@ func (self *ProtoProc)procCreateTopic(cmd protocol.Cmd, session *libnet.Session)
 		}
 	}
 
+	// store topic
+	topicStoreData := mongo_store.NewTopicStoreData(topicName, session.State.(*base.SessionState).ClientID, 
+		self.msgServer.cfg.LocalIP)
+
+	args = make([]string, 0)
+	args = append(args, topicName)
+	CCmd = protocol.NewCmdInternal(protocol.STORE_TOPIC_CMD, args, topicStoreData)
+	member := mongo_store.NewMember(session.State.(*base.SessionState).ClientID)
+	CCmd.AnyData.(*mongo_store.TopicStoreData).MemberList = append(CCmd.AnyData.(*mongo_store.TopicStoreData).MemberList, member)
+
+	log.Info(CCmd)
+
+	if self.msgServer.channels[protocol.STORE_TOPIC_INFO] != nil {
+		_, err = self.msgServer.channels[protocol.STORE_TOPIC_INFO].Channel.Broadcast(libnet.Json(CCmd))
+		if err != nil {
+			log.Error(err.Error())
+			return err
+		}
+	}	
+
+
 	return nil
 }
 

diff --git a/protocol/channel.go b/protocol/channel.go
@@ -25,4 +25,5 @@ const (
 
 const (
 	STORE_CLIENT_INFO     = "/store/client-info" 
+	STORE_TOPIC_INFO      = "/store/topic-info" 
 )
diff --git a/storage/mongo_store/topic_store.go b/storage/mongo_store/topic_store.go
@@ -15,3 +15,28 @@
 
 package mongo_store
 
+type Member struct {
+	ID   string
+}
+
+func NewMember(ID string) *Member {
+	return &Member {
+		ID : ID,
+	}
+}
+
+type TopicStoreData struct {	
+	TopicName      string      `bson:"TopicName"`
+	CreaterID      string      `bson:"CreaterID"`
+	MemberList     []*Member   `bson:"MemberList"`
+	MsgServerAddr  string      `bson:"MsgServerAddr"`
+}
+
+func NewTopicStoreData(topicName string, createrID string, msgServerAddr string) *TopicStoreData {
+	return &TopicStoreData{
+		TopicName     : topicName,
+		CreaterID     : createrID,
+		MemberList    : make([]*Member, 0),
+		MsgServerAddr : msgServerAddr,
+	}
+}