Vagrant lab

Dockerfile

@@ -0,0 +1,40 @@
+# 使用官方的 Go 镜像作为基础镜像
+FROM golang:1.20 AS builder
+
+# 设置工作目录
+WORKDIR /app
+
+# 拷贝 Go 项目代码到容器中
+COPY . .
+
+# 为 Linux 构建 plato 可执行文件
+# 为了在mac上能够执行，指定了 GOOS=linux GOARCH=amd64
+# 其他linux环境编译时不需要添加该命令，直接 go build -o plato . 即可
+RUN GOOS=linux GOARCH=amd64 go build -o plato .
+
+
+# 创建最终的生产环境镜像
+FROM alpine:latest
+
+# 设置工作目录
+WORKDIR /app
+
+# 拷贝构建好的可执行文件到生产镜像中
+COPY --from=builder /app/plato .
+
+# 赋予 plato-gateway 和 plato-state 可执行权限
+RUN #chmod +x ./plato-gateway && chmod +x ./plato-state
+RUN chmod +x ./plato
+
+# 拷贝项目的配置文件到容器中
+COPY ./plato.yaml .
+
+# 声明 gateway 服务运行时监听的端口
+EXPOSE 8900
+EXPOSE 8901
+
+# 声明 state 服务运行时监听的端口
+EXPOSE 8902
+
+# 启动应用程序
+#CMD ["./plato gateway"]

Makefile

@@ -0,0 +1,6 @@
+all:build
+
+build:
+	@echo "go build plato begin"
+	go build -o plato .
+	@echo "go build plato success"

chat.log

@@ -1,5 +1,42 @@
-me: hello
-logic: hello
-me: 你好！ plato
-logic: 你好！ plato
+me-clientID:0: 123
+logic: 123
+me-clientID:0: 345566
+me-clientID:1: sadklasjdl
+logic: sadklasjdl
+me-clientID:2: sdksajdlsa
+logic: sdksajdlsa
+me-clientID:3: djsalkjdlsa
+logic: djsalkjdlsa
+me-clientID:4: dhaskljdlks
+logic: dhaskljdlks
+me-clientID:5: dhjakshdks
+logic: dhjakshdks
+me-clientID:6: dhjksahdks
+logic: dhjksahdks
+me-clientID:7: 123daksjdklas
+logic: 123daksjdklas
+me-clientID:8: dmaklsdklsa
+logic: dmaklsdklsa
+me-clientID:9: dmksamdlsa
+logic: dmksamdlsa
+me-clientID:10: dals;dk;asd
+logic: dals;dk;asd
+me-clientID:11: dhaskjhdkasdk
+logic: dhaskjhdkasdk
+me-clientID:12: sdklajdklsad
+logic: sdklajdklsad
+me-clientID:13: djaslkdlsa
+logic: djaslkdlsa
+me-clientID:14: asdkljsaljdslajd
+logic: asdkljsaljdslajd
+me-clientID:15: djskahdkhsakd
+logic: djskahdkhsakd
+me-clientID:16: dklasjdklsajkld
+logic: dklasjdklsajkld
+me-clientID:17: djklsajdlsajdl
+logic: djklsajdlsajdl
+me-clientID:18: djaksljdlasdjl
+logic: djaksljdlasdjl
+me-clientID:19: djaslkjdlas
+logic: djaslkjdlas
 

client/cui.go

@@ -2,13 +2,14 @@ package client
 
 import (
 	"fmt"
+	"github.com/hardcore-os/plato/common/sdk"
 	"io/ioutil"
 	"log"
 	"math/rand"
+	"net"
 	"time"
 
 	"github.com/gookit/color"
-	"github.com/hardcore-os/plato/client/sdk"
 	"github.com/rocket049/gocui"
 )
 
@@ -59,16 +60,19 @@ func viewPrint(g *gocui.Gui, name, msg string, newline bool) {
 	g.Update(out.Show)
 }
 
-//doRecv work in goroutine
+// doRecv work in goroutine
 func doRecv(g *gocui.Gui) {
 	recvChannel := chat.Recv()
 	for msg := range recvChannel {
-		switch msg.Type {
-		case sdk.MsgTypeText:
-			viewPrint(g, msg.Name, msg.Content, false)
+		if msg != nil {
+			switch msg.Type {
+			case sdk.MsgTypeText:
+				viewPrint(g, msg.Name, msg.Content, false)
+			case sdk.MsgTypeAck:
+				//TODO 默认不处理
+			}
 		}
 	}
-	g.Close()
 }
 
 func quit(g *gocui.Gui, v *gocui.View) error {
@@ -91,7 +95,8 @@ func doSay(g *gocui.Gui, cv *gocui.View) {
 				ToUserID:   "222222",
 				Content:    string(p)}
 			// 先把自己说的话显示到消息流中
-			viewPrint(g, "me", msg.Content, false)
+			idKey := fmt.Sprintf("%d", chat.GetCurClientID())
+			viewPrint(g, "me-clientID:"+idKey, msg.Content, false)
 			chat.Send(msg)
 		}
 		v.Autoscroll = true
@@ -168,7 +173,7 @@ func viewHead(g *gocui.Gui, x0, y0, x1, y1 int) error {
 		}
 		v.Wrap = false
 		v.Overwrite = true
-		msg := "开始聊天了!"
+		msg := "Plato: IM系统聊天对话框"
 		setHeadText(g, msg)
 	}
 	return nil
@@ -223,7 +228,7 @@ func pasteDown(g *gocui.Gui, cv *gocui.View) error {
 
 func RunMain() {
 	// step1 创建caht的核心对象
-	chat = sdk.NewChat("127.0.0.1:8080", "logic", "12312321", "2131")
+	chat = sdk.NewChat(net.ParseIP("0.0.0.0"), 8900, "logic", "12312321", "2131")
 	// step2 创建 GUI 图层对象并进行参与与回调函数的配置
 	g, err := gocui.NewGui(gocui.OutputNormal)
 	if err != nil {
@@ -256,6 +261,15 @@ func RunMain() {
 	if err := g.SetKeybinding("main", gocui.KeyArrowUp, gocui.ModNone, pasteUP); err != nil {
 		log.Panicln(err)
 	}
+
+	// emit disconnection
+	//go func() {
+	//	time.Sleep(10 * time.Second)
+	//	// reconnection
+	//	chat.ReConn()
+	//	go doRecv(g)
+	//}()
+
 	// 启动消费函数
 	go doRecv(g)
 	if err := g.MainLoop(); err != nil {

cmd/gateway.go

@@ -0,0 +1,19 @@
+package cmd
+
+import (
+	"github.com/hardcore-os/plato/gateway"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	rootCmd.AddCommand(gatewayCmd)
+}
+
+var gatewayCmd = &cobra.Command{
+	Use: "gateway",
+	Run: GatewayHandle,
+}
+
+func GatewayHandle(cmd *cobra.Command, args []string) {
+	gateway.RunMain(ConfigPath)
+}

cmd/ipconf.go

@@ -0,0 +1,19 @@
+package cmd
+
+import (
+	"github.com/hardcore-os/plato/ipconf"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	rootCmd.AddCommand(ipConfCmd)
+}
+
+var ipConfCmd = &cobra.Command{
+	Use: "ipconf",
+	Run: IpConfHandle,
+}
+
+func IpConfHandle(cmd *cobra.Command, args []string) {
+	ipconf.RunMain(ConfigPath)
+}

cmd/perf.go

@@ -0,0 +1,20 @@
+package cmd
+
+import (
+	"github.com/hardcore-os/plato/perf"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	rootCmd.AddCommand(perfCmd)
+	perfCmd.PersistentFlags().Int32Var(&perf.TcpConnNum, "tcp_conn_num", 10000, "tcp 连接的数量，默认10000")
+}
+
+var perfCmd = &cobra.Command{
+	Use: "perf",
+	Run: PerfHandle,
+}
+
+func PerfHandle(cmd *cobra.Command, args []string) {
+	perf.RunMain()
+}

cmd/plato.go

@@ -7,8 +7,18 @@ import (
 	"github.com/spf13/cobra"
 )
 
+var (
+	ConfigPath string
+)
+
 func init() {
 	cobra.OnInitialize(initConfig)
+	rootCmd.PersistentFlags().StringVar(
+		&ConfigPath,
+		"config",
+		"./plato.yaml",
+		"config file (default is ./plato.yaml)",
+	)
 }
 
 var rootCmd = &cobra.Command{

cmd/state.go

@@ -0,0 +1,19 @@
+package cmd
+
+import (
+	"github.com/hardcore-os/plato/state"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	rootCmd.AddCommand(stateCmd)
+}
+
+var stateCmd = &cobra.Command{
+	Use: "state",
+	Run: StateHandle,
+}
+
+func StateHandle(cmd *cobra.Command, args []string) {
+	state.RunMain(ConfigPath)
+}

common/cache/const.go

@@ -0,0 +1,10 @@
+package cache
+
+import "time"
+
+const (
+	MaxClientIDKey  = "max_client_id_{%d}_%d" // slot_connID
+	LastMsgKey      = "last_msg_{%d}_%d"      // slot_connID
+	LoginSlotSetKey = "login_slot_set_{%d}"   // 通过 hash tag保证在cluster模式上 key都在一个shard上 // slot
+	TTL7D           = 7 * 24 * time.Hour
+)