main.go

package main

import (
	"fmt"
	"go-websocket/define"
	"go-websocket/pkg/redis"
	"go-websocket/routers"
	"go-websocket/servers"
	_ "go-websocket/tools/log"
	"go-websocket/tools/readconfig"
	"go-websocket/tools/util"
	"net/http"
	"os"
)

func main() {
	port := getPort()

	//初始化配置文件
	if err := readconfig.InitConfig(); err != nil {
		panic(err)
	}

	//初始化RPC服务
	initRPCServer(port)

	//记录本机内网IP地址
	define.LocalHost = util.GetIntranetIp()

	//将服务器地址、端口注册到redis中
	registerServer()

	//初始化路由
	routers.Init()

	//启动一个定时器用来发送心跳
	servers.PingTimer()

	fmt.Printf("服务器启动成功，端口号：%s\n", port)

	if err := http.ListenAndServe(":"+port, nil); err != nil {
		panic(err)
	}
}

func initRPCServer(port string) {
	//如果是集群，则启用RPC进行通讯
	if util.IsCluster() {
		//初始化RPC服务
		rpcPort := util.GenRpcPort(port)
		servers.InitRpcServer(rpcPort)
		fmt.Printf("启动RPC，端口号：%s\n", rpcPort)
	}
}

//将服务器地址、端口注册到redis中
func registerServer() {
	if util.IsCluster() {
		_, err := redis.SetAdd(define.REDIS_KEY_SERVER_LIST, define.LocalHost+":"+define.RPCPort)
		if err != nil {
			panic(err)
		}
	}
}

func getPort() string {
	port := "666"

	args := os.Args //获取用户输入的所有参数
	if args != nil && len(args) >= 2 && len(args[1]) != 0 {
		port = args[1]
	}

	define.Port = port
	return port
}