package main

import (
	"gitee.com/micro-tools/micro/module"
	"gitee.com/micro-tools/micro/module/base"
	pf "gitee.com/micro-tools/wf/extend/pb/protobuf"
	"gitee.com/micro-tools/wf/extend/utils/gmicro/server"
	"google.golang.org/grpc"
)

type Server struct {
	base.Module
}

func New() module.Module {
	this := new(Server)
	return this
}

func (s *Server) GetType() string {
	return "server"
}

func (s *Server) OnInit(app module.App) {
	s.Module.OnInit(s, app, server.RPCBind(func(sv *grpc.Server) {
		pf.RegisterUserServer(sv, &User{})
		pf.RegisterOrderServer(sv, &Order{})
		app.Logger().Infof("protocol user register")
		app.Logger().Infof("protocol order register")
	}))
}

func (s *Server) Run(closeSig chan bool) {
	s.App.Logger().Infof("%s模块运行中...", s.GetType())
	<-closeSig
	s.App.Logger().Infof("%s模块已停止...", s.GetType())
}

func (s *Server) OnDestroy() {
	s.Module.OnDestroy()
	s.App.Logger().Infof("%s模块已回收...", s.GetType())
}