package actions

import (
	"context"
	"google.golang.org/grpc"
	"google.golang.org/grpc/reflection"
	"pro2d/conf"
	"pro2d/models"
	"pro2d/protos/pb"
	"pro2d/utils"
)

type AccountServer struct{
	pb.UnimplementedAccountServer
	*BasicServer
}

func NewAccountServer() *AccountServer {
	return &AccountServer{
		BasicServer: NewServer(conf.GlobalConf.AccountConf.DBName),
	}
}

//拦截器
func AccountServerInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo,
	handler grpc.UnaryHandler) (interface{}, error) {

	utils.Sugar.Debugf("gRPC method: %s, %v", info.FullMethod, req)
	resp, err := handler(ctx, req)
	return resp, err
}

func (s *AccountServer)Start() error {
	lis, err := s.BasicServer.Start(conf.GlobalConf.AccountConf)
	if err != nil {
		return err
	}

	models.InitAccountServerModels()

	//new一个grpc
	gs := grpc.NewServer(grpc.UnaryInterceptor(AccountServerInterceptor))

	pb.RegisterAccountServer(gs, s)
	reflection.Register(gs) //在给定的gRPC服务器上注册服务器反射服务

	// Serve方法在lis上接受传入连接，为每个连接创建一个ServerTransport和server的goroutine。
	// 该goroutine读取gRPC请求，然后调用已注册的处理程序来响应它们。
	utils.Sugar.Debugf("Start AccountServer listening on %d", conf.GlobalConf.AccountConf.Port)

	return gs.Serve(lis)
}

func (s *AccountServer)Stop()  {
	s.BasicServer.Close()
}

type GameServer struct{
	pb.UnimplementedGameServer
	*BasicServer
}

func NewGameServer() *GameServer {
	return &GameServer{
		BasicServer: NewServer(conf.GlobalConf.GameConf.DBName),
	}
}
//拦截器
func GameServerInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo,
	handler grpc.UnaryHandler) (interface{}, error) {

	utils.Sugar.Debugf("gRPC method: %s, %v", info.FullMethod, req)
	resp, err := handler(ctx, req)
	return resp, err
}

func (s *GameServer)Start() error {
	lis, err := s.BasicServer.Start(conf.GlobalConf.GameConf)
	if err != nil {
		return err
	}

	models.InitGameServerModels()

	//new一个grpc
	gs := grpc.NewServer(grpc.UnaryInterceptor(GameServerInterceptor))

	pb.RegisterGameServer(gs, s)
	reflection.Register(gs) //在给定的gRPC服务器上注册服务器反射服务

	// Serve方法在lis上接受传入连接，为每个连接创建一个ServerTransport和server的goroutine。
	// 该goroutine读取gRPC请求，然后调用已注册的处理程序来响应它们。
	utils.Sugar.Debugf("Start GameServer listening on %d", conf.GlobalConf.GameConf.Port)
	return gs.Serve(lis)
}

func (s *GameServer)Stop()  {
	s.BasicServer.Close()
}