socket.go

package socket

import (
	"fmt"
	"github.com/hulutech-web/goravel-socket/define"
	"github.com/hulutech-web/goravel-socket/pkg/etcd"
	"github.com/hulutech-web/goravel-socket/servers"
	"github.com/hulutech-web/goravel-socket/setting"
	"github.com/hulutech-web/goravel-socket/tools/log"
	"github.com/hulutech-web/goravel-socket/tools/util"
	"net"
)

type Socket struct{}

func init() {
	setting.Setup()
	log.Setup()
	Boot()
}

func Boot() {
	//初始化RPC服务
	initRPCServer()

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

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

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

// ETCD注册发现服务
func registerServer() {
	if util.IsCluster() {
		//注册租约
		ser, err := etcd.NewServiceReg(setting.EtcdSetting.Endpoints, 5)
		if err != nil {
			panic(err)
		}

		hostPort := net.JoinHostPort(setting.GlobalSetting.LocalHost, setting.CommonSetting.RPCPort)
		//添加key
		err = ser.PutService(define.ETCD_SERVER_LIST+hostPort, hostPort)
		if err != nil {
			panic(err)
		}

		cli, err := etcd.NewClientDis(setting.EtcdSetting.Endpoints)
		if err != nil {
			panic(err)
		}
		_, err = cli.GetService(define.ETCD_SERVER_LIST)
		if err != nil {
			panic(err)
		}
	}
}