func (self *ProtoProc) procCreateTopic(cmd protocol.Cmd, session *libnet.Session) error { glog.Info("procCreateTopic") var err error if len(cmd.GetArgs()) != 1 { return CMD_NOT_CORRECT } topicName := cmd.GetArgs()[0] topicStoreData := storage.NewTopicStoreData(topicName, session.State.(*base.SessionState).ClientID, self.msgServer.cfg.LocalIP) t := protocol.NewTopic(topicName, self.msgServer.cfg.LocalIP, session.State.(*base.SessionState).ClientID, session) t.ClientIDList = append(t.ClientIDList, session.State.(*base.SessionState).ClientID) t.TSD = topicStoreData self.msgServer.topics[topicName] = t self.msgServer.topics[topicName].Channel = libnet.NewChannel(self.msgServer.server.Protocol()) self.msgServer.topics[topicName].Channel.Join(session, nil) glog.Info(topicStoreData) args := make([]string, 0) args = append(args, topicName) CCmd := protocol.NewCmdInternal(protocol.STORE_TOPIC_CMD, args, topicStoreData) m := storage.NewMember(session.State.(*base.SessionState).ClientID) CCmd.AnyData.(*storage.TopicStoreData).MemberList = append(CCmd.AnyData.(*storage.TopicStoreData).MemberList, m) glog.Info(CCmd) if self.msgServer.channels[protocol.SYSCTRL_TOPIC_STATUS] != nil { _, err = self.msgServer.channels[protocol.SYSCTRL_TOPIC_STATUS].Channel.Broadcast(libnet.Json(CCmd)) if err != nil { glog.Error(err.Error()) return err } } return nil }
func (self *ProtoProc) procJoinTopic(cmd protocol.Cmd, session *libnet.Session) error { glog.Info("procJoinTopic") var err error if len(cmd.GetArgs()) != 2 { return CMD_NOT_CORRECT } topicName := cmd.GetArgs()[0] clientID := cmd.GetArgs()[1] if self.msgServer.topics[topicName] == nil { glog.Warning("no topic in this server :" + topicName) t, err := self.findTopicMsgAddr(topicName) if err != nil { glog.Warningf("no topicName : %s", topicName) return err } resp := protocol.NewCmdSimple(protocol.LOCATE_TOPIC_MSG_ADDR_CMD) resp.AddArg(t.MsgServerAddr) resp.AddArg(topicName) err = session.Send(libnet.Json(resp)) if err != nil { glog.Error(err.Error()) return err } return err } else { t, err := self.findTopicMsgAddr(topicName) if err != nil { glog.Warningf("no topicName : %s", topicName) return err } resp := protocol.NewCmdSimple(protocol.LOCATE_TOPIC_MSG_ADDR_CMD) resp.AddArg(t.MsgServerAddr) resp.AddArg(topicName) err = session.Send(libnet.Json(resp)) if err != nil { glog.Error(err.Error()) return err } } m := storage.NewMember(clientID) self.msgServer.topics[topicName].ClientIDList = append(self.msgServer.topics[topicName].ClientIDList, clientID) self.msgServer.topics[topicName].AddMember(m) self.msgServer.topics[topicName].Channel.Join(session, nil) args := make([]string, 0) args = append(args, topicName) CCmd := protocol.NewCmdInternal(protocol.STORE_TOPIC_CMD, args, self.msgServer.topics[topicName].TSD) glog.Info(CCmd) if self.msgServer.channels[protocol.SYSCTRL_TOPIC_STATUS] != nil { _, err = self.msgServer.channels[protocol.SYSCTRL_TOPIC_STATUS].Channel.Broadcast(libnet.Json(CCmd)) if err != nil { glog.Error(err.Error()) return err } } return nil }