136 lines
2.4 KiB
Go
136 lines
2.4 KiB
Go
package robot
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/pb"
|
|
"go_dreamfactory/utils"
|
|
"io/ioutil"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/golang/protobuf/proto"
|
|
"github.com/gorilla/websocket"
|
|
jsoniter "github.com/json-iterator/go"
|
|
)
|
|
|
|
type Robot struct {
|
|
ws *websocket.Conn
|
|
Opts *Options
|
|
}
|
|
|
|
func NewRobot(opts *Options) *Robot {
|
|
ws, _, err := websocket.DefaultDialer.Dial(opts.WsUrl, nil)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
r := &Robot{
|
|
ws: ws,
|
|
Opts: opts,
|
|
}
|
|
|
|
return r
|
|
}
|
|
|
|
func (r *Robot) Run() {
|
|
log.Print("Robot running...")
|
|
log.Printf("websocket %s \n", r.Opts.WsUrl)
|
|
|
|
if r.Opts.Create { //创建新用户
|
|
r.AccountRegister()
|
|
} else {
|
|
//login user
|
|
r.AccountLogin()
|
|
}
|
|
|
|
for {
|
|
var msg *pb.UserMessage = &pb.UserMessage{}
|
|
_, data, err := r.ws.ReadMessage()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
if err = proto.Unmarshal(data, msg); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
r.handleMsg(msg)
|
|
}
|
|
}
|
|
|
|
func (r *Robot) handleMsg(msg *pb.UserMessage) {
|
|
m, f, ok := utils.ParseP(msg.ServiceMethod)
|
|
if !ok {
|
|
log.Fatal("route error")
|
|
}
|
|
switch m {
|
|
case "login":
|
|
r.handleLogin(f)
|
|
default:
|
|
log.Fatal("module route no exist")
|
|
}
|
|
}
|
|
|
|
func (r *Robot) SendToClient(data []byte) error {
|
|
return r.ws.WriteMessage(websocket.BinaryMessage, data)
|
|
}
|
|
|
|
//注册账号
|
|
func (r *Robot) AccountRegister() {
|
|
//http
|
|
regReq := &pb.UserRegisterReq{}
|
|
jsonByte, _ := json.Marshal(regReq)
|
|
req, err := http.NewRequest("POST", r.Opts.RegUrl, bytes.NewReader(jsonByte))
|
|
if err != nil {
|
|
log.Fatalf("account register err %v", err)
|
|
}
|
|
req.Header.Set("Content-Type", "application/json;charset=UTF-8")
|
|
httpClient := &http.Client{}
|
|
rsp, err := httpClient.Do(req)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer rsp.Body.Close()
|
|
|
|
body, _ := ioutil.ReadAll(rsp.Body)
|
|
regRsp := &pb.UserRegisterRsp{}
|
|
err = jsoniter.Unmarshal(body, regRsp)
|
|
|
|
if regRsp.Code == int32(pb.ErrorCode_Success) { //注册成功
|
|
//登录
|
|
loginReg := &pb.UserLoginReq{
|
|
Name: regReq.Account,
|
|
}
|
|
|
|
head := &pb.UserMessage{
|
|
ServiceMethod: "login.login",
|
|
}
|
|
if comm.ProtoEncode(loginReg, head) {
|
|
err = r.SendToClient(head.Data)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
//登录认证
|
|
func (r *Robot) AccountLogin() {
|
|
//登录
|
|
loginReg := &pb.UserLoginReq{
|
|
Name: r.Opts.Account,
|
|
}
|
|
|
|
head := &pb.UserMessage{
|
|
ServiceMethod: "login.login",
|
|
}
|
|
if comm.ProtoEncode(loginReg, head) {
|
|
err := r.SendToClient(head.Data)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|