@@ -1,58 +1,124 @@
|
||||
package grpc
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"github.com/micro/go-micro"
|
||||
broker "github.com/micro/go-micro/broker"
|
||||
client "github.com/micro/go-micro/client/grpc"
|
||||
server "github.com/micro/go-micro/server/grpc"
|
||||
"github.com/micro/go-micro/client"
|
||||
gclient "github.com/micro/go-micro/client/grpc"
|
||||
"github.com/micro/go-micro/server"
|
||||
gserver "github.com/micro/go-micro/server/grpc"
|
||||
"github.com/micro/go-micro/service"
|
||||
)
|
||||
|
||||
type grpcService struct {
|
||||
opts service.Options
|
||||
}
|
||||
|
||||
func newService(opts ...service.Option) service.Service {
|
||||
options := service.NewOptions(opts...)
|
||||
|
||||
return &grpcService{
|
||||
opts: options,
|
||||
}
|
||||
}
|
||||
|
||||
func (s *grpcService) Name() string {
|
||||
return s.opts.Server.Options().Name
|
||||
}
|
||||
|
||||
// Init initialises options. Additionally it calls cmd.Init
|
||||
// which parses command line flags. cmd.Init is only called
|
||||
// on first Init.
|
||||
func (s *grpcService) Init(opts ...service.Option) {
|
||||
// process options
|
||||
for _, o := range opts {
|
||||
o(&s.opts)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *grpcService) Options() service.Options {
|
||||
return s.opts
|
||||
}
|
||||
|
||||
func (s *grpcService) Client() client.Client {
|
||||
return s.opts.Client
|
||||
}
|
||||
|
||||
func (s *grpcService) Server() server.Server {
|
||||
return s.opts.Server
|
||||
}
|
||||
|
||||
func (s *grpcService) String() string {
|
||||
return "grpc"
|
||||
}
|
||||
|
||||
func (s *grpcService) Start() error {
|
||||
for _, fn := range s.opts.BeforeStart {
|
||||
if err := fn(); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
if err := s.opts.Server.Start(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, fn := range s.opts.AfterStart {
|
||||
if err := fn(); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *grpcService) Stop() error {
|
||||
var gerr error
|
||||
|
||||
for _, fn := range s.opts.BeforeStop {
|
||||
if err := fn(); err != nil {
|
||||
gerr = err
|
||||
}
|
||||
}
|
||||
|
||||
if err := s.opts.Server.Stop(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, fn := range s.opts.AfterStop {
|
||||
if err := fn(); err != nil {
|
||||
gerr = err
|
||||
}
|
||||
}
|
||||
|
||||
return gerr
|
||||
}
|
||||
|
||||
func (s *grpcService) Run() error {
|
||||
if err := s.Start(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// wait on context cancel
|
||||
<-s.opts.Context.Done()
|
||||
|
||||
return s.Stop()
|
||||
}
|
||||
|
||||
// NewService returns a grpc service compatible with go-micro.Service
|
||||
func NewService(opts ...micro.Option) micro.Service {
|
||||
func NewService(opts ...service.Option) service.Service {
|
||||
// our grpc client
|
||||
c := client.NewClient()
|
||||
c := gclient.NewClient()
|
||||
// our grpc server
|
||||
s := server.NewServer()
|
||||
// our grpc broker
|
||||
b := broker.NewBroker()
|
||||
s := gserver.NewServer()
|
||||
|
||||
// create options with priority for our opts
|
||||
options := []micro.Option{
|
||||
micro.Client(c),
|
||||
micro.Server(s),
|
||||
micro.Broker(b),
|
||||
options := []service.Option{
|
||||
service.Client(c),
|
||||
service.Server(s),
|
||||
}
|
||||
|
||||
// append passed in opts
|
||||
options = append(options, opts...)
|
||||
|
||||
// generate and return a service
|
||||
return micro.NewService(options...)
|
||||
}
|
||||
|
||||
// NewFunction returns a grpc service compatible with go-micro.Function
|
||||
func NewFunction(opts ...micro.Option) micro.Function {
|
||||
// our grpc client
|
||||
c := client.NewClient()
|
||||
// our grpc server
|
||||
s := server.NewServer()
|
||||
// our grpc broker
|
||||
b := broker.NewBroker()
|
||||
|
||||
// create options with priority for our opts
|
||||
options := []micro.Option{
|
||||
micro.Client(c),
|
||||
micro.Server(s),
|
||||
micro.Broker(b),
|
||||
micro.RegisterTTL(time.Minute),
|
||||
micro.RegisterInterval(time.Second * 30),
|
||||
}
|
||||
|
||||
// append passed in opts
|
||||
options = append(options, opts...)
|
||||
|
||||
// generate and return a function
|
||||
return micro.NewFunction(options...)
|
||||
return newService(options...)
|
||||
}
|
||||
|
Reference in New Issue
Block a user