Blame view

test/client.go 2.04 KB
ee23102d   zhangqijia   支持mongo, grpc接服务器
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
  package main
  
  import (
  	"context"
  	"fmt"
  	"google.golang.org/grpc"
  	_ "pro2d/conf"
  	"pro2d/protos/pb"
  	"pro2d/utils"
  )
  
  func Register(c pb.AccountClient, phone, password string)  error {
  	r, err := c.RegisterHandler(context.Background(), &pb.Register{
  		Phone:                "17683852936",
  		Password:             "123456",
  		Code:     0,
  	})
  
  	if err != nil {
  		return err
  	}
  	if r.Code != 0 {
  		return fmt.Errorf("%s", r.Msg)
  	}
  	utils.Sugar.Debug("register success")
  	return nil
  }
  
  func Login(loginUri, token string)  {
  	gameConn, err := grpc.Dial(loginUri, grpc.WithInsecure() )
  	if err != nil {
  		utils.Sugar.Errorf("game conn err: %v", err)
  		return
  	}
  	defer gameConn.Close()
  
  	client:= pb.NewGameClient(gameConn)
f7f4beb5   zhangqijia   新增grpc平滑关闭,修复crea...
38
  	var role *pb.Role
ee23102d   zhangqijia   支持mongo, grpc接服务器
39
40
41
42
43
44
45
46
  	loginRsp, err := client.LoginHandler(context.Background(), &pb.Token{
  		Token: token,
  	})
  
  	if err != nil {
  		utils.Sugar.Errorf("login error: %v", err)
  		return
  	}
f7f4beb5   zhangqijia   新增grpc平滑关闭,修复crea...
47
  	role = loginRsp.Role
ee23102d   zhangqijia   支持mongo, grpc接服务器
48
  
f7f4beb5   zhangqijia   新增grpc平滑关闭,修复crea...
49
50
51
52
53
54
55
56
57
  	if loginRsp.Rsp.Code != 0 {
  		utils.Sugar.Debugf("login rsp: %v", loginRsp.Rsp.Msg)
  		createRole, err := client.CreateRoleHandler(context.Background(), &pb.Token{Token: token})
  		if err != nil {
  			utils.Sugar.Errorf("create role err: %v", err)
  			return
  		}
  		utils.Sugar.Debug("create role rsp: ", createRole.Rsp.Code, createRole.Rsp.Msg)
  		role = createRole.Role
ee23102d   zhangqijia   支持mongo, grpc接服务器
58
  	}
ee23102d   zhangqijia   支持mongo, grpc接服务器
59
  
f7f4beb5   zhangqijia   新增grpc平滑关闭,修复crea...
60
  	utils.Sugar.Debugf("login successful role: %v", role)
ee23102d   zhangqijia   支持mongo, grpc接服务器
61
62
63
64
65
66
67
68
69
70
  }
  
  func main() {
  	conn, err := grpc.Dial("localhost:8848", grpc.WithInsecure())
  	if err != nil {
  		utils.Sugar.Errorf("conn err: %v", err)
  		return
  	}
  	defer conn.Close()
  	c := pb.NewAccountClient(conn)
f7f4beb5   zhangqijia   新增grpc平滑关闭,修复crea...
71
72
73
74
75
  	//err = Register(c,"17683852936", "123456")
  	//if err != nil {
  	//	utils.Sugar.Errorf("register err: %v", err)
  	//	return
  	//}
ee23102d   zhangqijia   支持mongo, grpc接服务器
76
77
78
79
80
  	rsp, err := c.CreateTokenHandler(context.Background(), &pb.AccountInfo{
  		Phone:                "17683852936",
  		Password:             "123456",
  	})
  
1584eb4b   zhangqijia   修复创建唯一索引的bug
81
82
83
84
85
86
87
88
89
90
  	if err != nil {
  		utils.Sugar.Errorf("createtoken err: %v", err)
  		return
  	}
  
  	if rsp.Rsp.Code != 0 {
  		utils.Sugar.Errorf("createtoken err: %v", rsp.Rsp.Msg)
  		return
  	}
  
f7f4beb5   zhangqijia   新增grpc平滑关闭,修复crea...
91
92
93
  	if len(rsp.GameService) >0 {
  		Login(rsp.GameService[0].Address, rsp.Token)
  	}
ee23102d   zhangqijia   支持mongo, grpc接服务器
94
  }