update game agent

Author: yiv

agent/client/main.go

@@ -0,0 +1,131 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"log"
+	"net/url"
+	"strconv"
+
+	"github.com/gogo/protobuf/proto"
+	"github.com/gorilla/websocket"
+
+	"github.com/yiv/yivgame/game/pb"
+)
+
+var (
+	agentAddr string = ":10050"
+)
+
+type client struct {
+	sk        *websocket.Conn
+	uid       int64
+	token     string
+	roomClass int32
+	player    *pb.Player
+}
+
+func newClient(uid int64) (c *client) {
+	c = &client{
+		sk:        newWebSocket(),
+		uid:       uid,
+		token:     strconv.FormatInt(uid, 10),
+		roomClass: 3,
+	}
+
+	return
+}
+
+func (c *client) recv(errChan chan error) {
+	for {
+		_, message, err := c.sk.ReadMessage()
+		if err != nil {
+			log.Println("read:", err)
+			errChan <- err
+			return
+		}
+		code, pbBytes := splitBytes(message)
+		if code == 10018 {
+			res := &pb.GeRes{}
+			proto.Unmarshal(pbBytes, res)
+			fmt.Printf("[返回]发消息[code: %v]\n", res.Code)
+		} else if code == 20019 {
+			res := &pb.ChatMsg{}
+			proto.Unmarshal(pbBytes, res)
+			fmt.Printf("[通知]信息[发者: %v,编号：%v]\n", res.Sid, res.Mid)
+		} else {
+			res := &pb.GeRes{}
+			proto.Unmarshal(pbBytes, res)
+			fmt.Printf("[返回]未知[code: %v]\n", res.Code)
+		}
+		fmt.Printf("\n 17=消息\n\n")
+	}
+}
+func (c *client) send(errChan chan error) {
+	var cmdCode uint32
+	fmt.Println("=======命令行启动=====")
+	for {
+		fmt.Scanln(&cmdCode)
+
+		if cmdCode == 100 {
+			c.sk.Close()
+			continue
+		} else if cmdCode == 99 {
+			c.enterTable()
+			continue
+		} else {
+			cmdCode += 10000
+		}
+
+		fmt.Printf("输入命令: %v\n", cmdCode)
+		err := c.sk.WriteMessage(websocket.BinaryMessage, toBytes(cmdCode, nil))
+		if err != nil {
+			fmt.Println("命令发送错误：", err)
+			errChan <- err
+			return
+		}
+	}
+
+}
+
+func newWebSocket() *websocket.Conn {
+	u := url.URL{Scheme: "ws", Host: agentAddr, Path: "/ws"}
+	log.Printf("connecting to %s", u.String())
+	c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
+	if err != nil {
+		log.Fatal("dial:", err)
+	}
+	fmt.Println("socket connected ")
+	return c
+}
+func splitBytes(payload []byte) (code uint32, pbBytes []byte) {
+	code = uint32(payload[0])<<24 | uint32(payload[1])<<16 | uint32(payload[2])<<8 | uint32(payload[3])
+	pbBytes = payload[4:]
+	return
+}
+func toBytes(i uint32, pbBytes []byte) (payload []byte) {
+	payload = append(payload, byte(i>>24), byte(i>>16), byte(i>>8), byte(i))
+	payload = append(payload, pbBytes...)
+	return
+}
+
+var (
+	uid = flag.Int("uid", 1, "the client uid")
+)
+
+var (
+	userMap = map[int]int64{
+		1: 953685341995009,
+		2: 2232870895617,
+	}
+)
+
+func main() {
+	flag.Parse()
+	uid := userMap[*uid]
+	errChan := make(chan error)
+	cli := newClient(uid)
+	go cli.recv(errChan)
+	go cli.send(errChan)
+	fmt.Println("terminate by err :", <-errChan)
+}

agent/main.go

@@ -0,0 +1,88 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"net/http"
+	"net/http/pprof"
+	"os"
+	"os/signal"
+	"syscall"
+
+	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"google.golang.org/grpc"
+
+	"github.com/go-kit/kit/log"
+	"github.com/go-kit/kit/log/level"
+
+	"github.com/yiv/yivgame/agent/service"
+	"github.com/yiv/yivgame/game/pb"
+)
+
+var (
+	webSocketAddr = flag.String("websocket.addr", ":10050", "game agent webSocket address")
+	gameServer    = flag.String("grpc.gameserver", ":10020", "game server gRPC server address")
+	debugAddr     = flag.String("debug.addr", ":10051", "Debug and metrics listen address")
+)
+
+func main() {
+	flag.Parse()
+
+	// Logging domain.
+	var logger log.Logger
+	{
+		logger = log.NewLogfmtLogger(os.Stdout)
+		logger = level.NewFilter(logger, level.AllowDebug())
+		logger = log.With(logger, "ts", log.DefaultTimestampUTC)
+		logger = log.With(logger, "caller", log.DefaultCaller)
+	}
+	logger.Log("msg", "hello")
+	defer logger.Log("msg", "goodbye")
+
+	// Mechanical domain.
+	errc := make(chan error)
+	// Interrupt handler.
+	go func() {
+		c := make(chan os.Signal)
+		signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
+		errc <- fmt.Errorf("%s", <-c)
+	}()
+
+	// gRPC transport.
+	conn, err := grpc.Dial(*gameServer, grpc.WithInsecure())
+	if err != nil {
+		errc <- err
+		return
+	}
+	defer conn.Close()
+	client := pb.NewGameServiceClient(conn)
+
+	// Business domain.
+	agentService := service.NewAgentService(client, logger)
+
+	// webSocket transport.
+	go func() {
+		logger := log.With(logger, "transport", "webSocket")
+		m := http.NewServeMux()
+		m.HandleFunc("/", agentService.WebSocketServer)
+		logger.Log("addr", *webSocketAddr)
+		errc <- http.ListenAndServe(*webSocketAddr, m)
+	}()
+
+	// Debug listener.
+	go func() {
+		logger := log.With(logger, "transport", "debug")
+
+		m := http.NewServeMux()
+		m.Handle("/debug/pprof/", http.HandlerFunc(pprof.Index))
+		m.Handle("/debug/pprof/cmdline", http.HandlerFunc(pprof.Cmdline))
+		m.Handle("/debug/pprof/profile", http.HandlerFunc(pprof.Profile))
+		m.Handle("/debug/pprof/symbol", http.HandlerFunc(pprof.Symbol))
+		m.Handle("/debug/pprof/trace", http.HandlerFunc(pprof.Trace))
+		m.Handle("/metrics", promhttp.Handler())
+
+		logger.Log("addr", *debugAddr)
+		errc <- http.ListenAndServe(*debugAddr, m)
+	}()
+	logger.Log("terminated", <-errc)
+}

agent/service/frame.go

@@ -0,0 +1,27 @@
+package service
+
+import (
+	"github.com/yiv/yivgame/game/pb"
+)
+
+func code2bytes(i uint32) (b []byte) {
+	b = append(b, byte(i>>24), byte(i>>16), byte(i>>8), byte(i))
+	return
+}
+func bytes2code(b []byte) (i uint32) {
+	i = uint32(b[0])<<24 | uint32(b[1])<<16 | uint32(b[2])<<8 | uint32(b[3])
+	return
+}
+
+func toframe(i uint32, pbBytes []byte) *pb.Frame {
+	payload := append(code2bytes(i), pbBytes...)
+	return &pb.Frame{Payload: payload}
+}
+func frameCode(f *pb.Frame) uint32 {
+	payload := f.Payload
+	return bytes2code(payload[0:4])
+}
+func framePBbytes(f *pb.Frame) []byte {
+	payload := f.Payload
+	return payload[4:]
+}