上传机器人代码

cmd/robot/pack.go

@@ -0,0 +1,33 @@
+package robot
+
+import (
+	"go_dreamfactory/comm"
+	"go_dreamfactory/pb"
+	"log"
+)
+
+func (r *Robot) handlePackMsg(msg *pb.UserMessage) {
+	switch msg.SubType {
+	case "queryuserpackresp":
+		r.handleFriendAdd(msg)
+	}
+}
+
+//添加好友
+func (r *Robot) QueryUserPack() {
+	req := &pb.QueryUserPackReq{IType: 1}
+	head := &pb.UserMessage{MainType: "pack", SubType: "queryuserpackreq"}
+	defer traceFunc(head.MainType, head.SubType, r.user.UserData.GetUserId(), req)
+	err := r.SendToClient(head, req)
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func (r *Robot) handleQueryUserPack(msg *pb.UserMessage) {
+	rsp := &pb.QueryUserPackResp{}
+	if !comm.ProtoDecode(msg, rsp) {
+		return
+	}
+	printReply(msg, rsp)
+}

cmd/robot/robot.go

@@ -69,6 +69,8 @@ func (r *Robot) handleMsg(msg *pb.UserMessage) {
 		r.handleUserMsg(msg)
 	case "friend":
 		r.handleFriendMsg(msg)
+	case "pack":
+		r.handlePackMsg(msg)
 	default:
 		log.Fatal("module route no exist")
 	}
@@ -78,8 +80,10 @@ func (r *Robot) handleMsg(msg *pb.UserMessage) {
 func (r *Robot) onUserLoaded() {
 	//user
 	r.CreateUser("user671")
+
 	//friend
-	r.FriendAdd()
+	// r.FriendAdd()
+	r.QueryUserPack()
 }
 
 func (r *Robot) SendToClient(msg *pb.UserMessage, rsp proto.Message) error {

comm/core.go

@@ -34,8 +34,9 @@ const ( //Rpc
 )
 
 const (
-	Event_UserLogin  core.Event_Key = "Event_UserLogin"  //登录事件
-	Event_CreateUser core.Event_Key = "Event_CreateUser" //创建角色事件
+	Event_UserLogin   core.Event_Key = "Event_UserLogin"   //登录事件
+	Event_CreateUser  core.Event_Key = "Event_CreateUser"  //创建角色事件
+	Event_UserOffline core.Event_Key = "Event_UserOffline" //用户离线事件
 )
 
 type ISC_GateRouteComp interface {