cmdserver.go

package cmdservice

import (
	"google.golang.org/grpc"
	"sync"

	"net"

	"errors"
	"google.golang.org/grpc/reflection"
	"log"

	"github.com/BASChain/go-bas-mail-server/app/cmdpb"
	"github.com/BASChain/go-bas-mail-server/app/cmdservice/api"
	"github.com/BASChain/go-bas-mail-server/config"
	"github.com/BASChain/go-bas-mail-server/bmtpserver"
)

type cmdServer struct {
	localaddr  string
	grpcServer *grpc.Server
}

type CmdServerInter interface {
	StartCmdService()
	StopCmdService()
}

var (
	cmdServerInst     CmdServerInter
	cmdServerInstLock sync.Mutex
)

func GetCmdServerInst() CmdServerInter {
	if cmdServerInst == nil {
		cmdServerInstLock.Lock()
		defer cmdServerInstLock.Unlock()
		if cmdServerInst == nil {
			cfg := config.GetBMSCfg()
			cmdServerInst = &cmdServer{localaddr: cfg.CmdListenPort}
		}
	}

	return cmdServerInst
}

func (cs *cmdServer) checklocaladdress() error {
	if cs.localaddr == "" {
		return errors.New("No Server Listen address")
	}

	return nil
}

func (cs *cmdServer) StartCmdService() {
	if err := cs.checklocaladdress(); err != nil {
		log.Fatal("Start Cmd Service Failed", err)
		return
	}

	lis, err := net.Listen("tcp", cs.localaddr)
	if err != nil {
		log.Fatalf("failed to listen: %v", err)
	}
	cs.grpcServer = grpc.NewServer()

	cmdpb.RegisterDefaultcmdsrvServer(cs.grpcServer, &api.CmdDefaultServer{stop})
	cmdpb.RegisterStringopsrvServer(cs.grpcServer, &api.CmdStringOPSrv{})

	reflection.Register(cs.grpcServer)
	log.Println("Commamd line server will start at", cs.localaddr)
	if err := cs.grpcServer.Serve(lis); err != nil {
		log.Fatalf("failed to serve: %s", err)
	}
}

func (cs *cmdServer) StopCmdService() {
	config.GetBMSCfg().Save()
	//server.DNSServerStop()
	//dohserver.GetDohDaemonServer().ShutDown()
	//mem.MemStateStop()
	cs.grpcServer.Stop()
	log.Println("Command line server stoped")
}

func stop() {

	bmtpserver.GetBMTPServer().StopTCPServer()

	GetCmdServerInst().StopCmdService()

}