RoServer/RO_Server_Trunk-branch_bt/roserver/server_aoi/main.go

package main

import (
	"github.com/pkg/profile"
	"io/ioutil"
	"os"
	"rocommon"
	"rocommon/service"
	_ "rocommon/socket"
	_ "rocommon/socket/tcp"
	"roserver/baseserver"
	"roserver/baseserver/model"
	_ "roserver/server_aoi/model"
	model2 "roserver/server_aoi/model"
	_ "roserver/server_aoi/msg"
	"runtime"
	"strconv"
	"syscall"
)

func main() {
	//记录gate pid用来做关闭操作
	sysType := runtime.GOOS
	if sysType != "windows" {
		if pid := syscall.Getpid(); pid != 1 {
			fileName := "aoi_server.pid" + strconv.Itoa(pid)
			ioutil.WriteFile(fileName, []byte(strconv.Itoa(pid)), 0777)
			defer os.Remove(fileName)
		}
	}

	//CPU
	prof := profile.Start(profile.CPUProfile, profile.ProfilePath("./pprof/serveraoi.pprof"), profile.NoShutdownHook)

	baseserver.Init(model.SERVICE_NODE_TYPE_AOI_STR, nil, &model2.AoiUpdate{})

	//先建立服务器对应的连接，在监听客户端
	sConfig := service.GetServiceConfig()
	var acceNode rocommon.ServerNode = nil
	if sConfig.Node.Addr != "" {
		//创建客户端监听
		acceNode = baseserver.CreateAcceptor(baseserver.ServiceParam{
			ServiceName: model.SERVICE_NODE_TYPE_AOI_STR,

			ServiceType: "tcpAcceptor",
			ProcName:    "aoi.backend",
			LisAddr:     sConfig.Node.Addr,
		}, sConfig)
	}

	for _, concern := range sConfig.Node.Concern {
		//建立需要链接的服务器,可以通过服务器发现etcd来处理(包含在了CreateConnector中)
		baseserver.CreateConnector(baseserver.ServiceParam{
			DiscoveryServiceName: concern,
			DiscoveryServiceZone: sConfig.Node.Zone,
			ServiceType:          "tcpConnector",
			ServiceName:          model.SERVICE_NODE_TYPE_AOI_STR,
			ProcName:             "common.backend",
		})
	}

	baseserver.Wait()
	//CPU
	prof.Stop()
	baseserver.Exit(acceNode)

	/*
		//内存使用详情
		f, err := os.Create("./pprof/serveraoi.pprof/aoi_mem_profile.pro")
		if err != nil {
			log.Println("err:", err)
		}
		pprof.WriteHeapProfile(f)
		f.Close()
	*/
}