cms-template/service/service.go

package service

import (
	"context"
	"fmt"

	cmsservice "go.unistack.org/cms-service"
	cmsstorage "go.unistack.org/cms-service/storage"
	mpb "go.unistack.org/cms-template-proto/micro"
	serviceConfig "go.unistack.org/cms-template/config"
	"go.unistack.org/cms-template/handler"
	storage "go.unistack.org/cms-template/storage"
	grpcsrv "go.unistack.org/micro-server-grpc/v3"
	"go.unistack.org/micro/v3"
	"go.unistack.org/micro/v3/broker"
	"go.unistack.org/micro/v3/config"
	"go.unistack.org/micro/v3/logger"
	"go.unistack.org/micro/v3/register"
	"go.unistack.org/micro/v3/router"
	"go.unistack.org/micro/v3/server"
)

func NewService(ctx context.Context) (micro.Service, error) {
	var err error
	var svc micro.Service
	var reg register.Register
	var brk broker.Broker
	var rtr router.Router

	if ctx == nil {
		ctx = context.Background()
	}

	cfg := serviceConfig.NewConfig()
	cs := cmsservice.NewConfigLocal(cfg) // service.NewConfigRemote(cfg)...)

	if r, ok := register.FromContext(ctx); ok && r != nil {
		reg = r
	} else {
		reg = register.NewRegister()
	}
	if b, ok := broker.FromContext(ctx); ok && b != nil {
		brk = b
	} else {
		brk = broker.NewBroker()
	}
	if r, ok := router.FromContext(ctx); ok && r != nil {
		rtr = r
	} else {
		rtr = router.NewRouter()
	}

	// create grpc server
	mgsrv := grpcsrv.NewServer(
		server.Register(reg),
	)

	// create service
	svc = micro.NewService(
		micro.Server(mgsrv),
		micro.Register(reg),
		micro.Broker(brk),
		micro.Router(rtr),
		micro.Config(cs...),
	)

	h := handler.NewHandler(svc)

	// init service
	if err := svc.Init(
		micro.AfterStart(func(_ context.Context) error {
			return h.Init(svc.Options().Context)
		}),
		micro.BeforeStart(func(ctx context.Context) error {
			if err = config.Load(ctx, cs, config.LoadOverride(true)); err != nil {
				return err
			}
			if err = config.Validate(ctx, cfg); err != nil {
				return err
			}

			if err = svc.Init(
				micro.Name(cfg.Service.Name),
				micro.Version(cfg.Service.Version),
			); err != nil {
				return err
			}

			if err = svc.Server("grpc").Init(
				server.Address(cfg.App.Address),
				server.Name(cfg.Service.Name),
				server.Version(cfg.Service.Version),
			); err != nil {
				return err
			}

			return nil
		}),
		micro.BeforeStart(func(_ context.Context) error {
			level := logger.InfoLevel
			if v, ok := cfg.Logger.Level[cfg.Service.Name]; ok {
				level = logger.ParseLevel(v)
			} else if v, ok := cfg.Logger.Level["all"]; ok {
				level = logger.ParseLevel(v)
			}
			log := logger.NewLogger(
				logger.WithLevel(level),
				logger.WithCallerSkipCount(3),
			)
			return svc.Init(micro.Logger(log))
		}),
		micro.BeforeStart(func(_ context.Context) error {
			var scheme string
			var connstr string
			var dsn string
			if v, ok := cfg.Storage.DSN[cfg.Service.Name]; ok {
				connstr = v
			} else if v, ok := cfg.Storage.DSN["all"]; ok {
				connstr = v
			}
			fmt.Printf("SSSS %s\n", connstr)
			scheme, dsn, err = cmsstorage.StorageOptions(connstr)
			if err != nil {
				return err
			}
			db, dbok := cmsstorage.FromContext(svc.Options().Context)
			if !dbok {
				var err error
				db, err = cmsstorage.NewStorage(scheme, dsn)
				if err != nil {
					return err
				}
			}
			store, err := storage.NewStorage(scheme, db)
			if err != nil {
				return err
			}
			return svc.Init(micro.Context(cmsstorage.InterfaceNewContext(svc.Options().Context, store)))
		}),
	); err != nil {
		return nil, err
	}

	if err = mpb.RegisterAccountServiceServer(mgsrv, h); err != nil {
		logger.Fatalf(ctx, "failed to register handler: %v", err)
	}

	return svc, nil
}