[wip] base options, command, service

service/service.go

@@ -0,0 +1,148 @@
+package service
+
+import (
+	"net"
+	"os"
+	"sync"
+
+	grpcmiddleware "github.com/grpc-ecosystem/go-grpc-middleware"
+	"github.com/jinzhu/gorm"
+	"github.com/spf13/cobra"
+	"google.golang.org/grpc"
+)
+
+type Service interface {
+	Options() Options
+	Server() *grpc.Server
+	DB() *gorm.DB
+	Start() error
+	Stop() error
+	Close() error
+	Cmd() *cobra.Command
+}
+
+func New(opts ...Option) (Service, error) {
+	return newService(opts...)
+}
+
+type service struct {
+	cmd    *cobra.Command
+	opts   *options
+	server *grpc.Server
+	list   net.Listener
+	mu     sync.Mutex
+	running bool
+}
+
+func newService(opts ...Option) (*service, error) {
+	cmd.ParseFlags(os.Args)
+	s := &service{
+		opts: parseFlags(NewOptions()),
+		cmd:  cmd,
+	}
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	for _, f := range opts {
+		f(s.opts)
+	}
+	go func() {
+		for {
+			select {
+			case <-s.opts.ctx.Done():
+				s.Stop()
+			}
+		}
+	}()
+	if err := s.opts.parseTLSConfig(); err != nil {
+		return nil, err
+	}
+	cmd.Use = s.opts.name
+	cmd.RunE = func(cmd *cobra.Command, args []string) error {
+		if cmd.Use == "" {
+			cmd.Use = os.Args[0]
+		}
+		return s.run()
+	}
+	gopts := []grpc.ServerOption{grpc.UnaryInterceptor(grpcmiddleware.ChainUnaryServer(s.opts.serverInterceptors...))}
+	// TODO : check tls config and tls auth
+	// grpc.Creds(credentials.NewTLS(&s.opts.tlsConfig))
+	s.server = grpc.NewServer(append(gopts, s.opts.serverOpts...)...)
+	return s, nil
+}
+
+func (s *service) Options() Options {
+	return s.opts
+}
+
+func (s *service) DB() *gorm.DB {
+	return s.opts.db
+}
+
+func (s *service) Server() *grpc.Server {
+	return s.server
+}
+
+func (s *service) Cmd() *cobra.Command {
+	return s.cmd
+}
+
+func (s *service) run() error {
+	s.mu.Lock()
+	for i := range s.opts.beforeStart {
+		if err := s.opts.beforeStart[i](); err != nil {
+			return err
+		}
+	}
+	var err error
+	s.running = true
+	s.list, err = net.Listen("tcp", s.opts.address)
+	if err != nil {
+		s.mu.Unlock()
+		return err
+	}
+	s.opts.address = s.list.Addr().String()
+	errs := make(chan error)
+	go func() {
+		errs <- s.server.Serve(s.list)
+	}()
+	for i := range s.opts.afterStart {
+		if err := s.opts.afterStart[i](); err != nil {
+			s.mu.Unlock()
+			s.Stop()
+			return err
+		}
+	}
+	s.mu.Unlock()
+	return <- errs
+}
+
+func (s *service) Start() error {
+	return s.cmd.Execute()
+}
+
+func (s *service) Stop() error {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	if ! s.running {
+		return nil
+	}
+	for i := range s.opts.beforeStop {
+		if err := s.opts.beforeStop[i](); err != nil {
+			return err
+		}
+	}
+	s.server.GracefulStop()
+	s.running = false
+	for i := range s.opts.afterStop {
+		if err := s.opts.afterStop[i](); err != nil {
+
+		}
+	}
+	return nil
+}
+
+func (s *service) Close() error {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	return s.Stop()
+}