package main

import (
	"fmt"
	"qiniu/handler"

	qiniu "qiniu/proto/qiniu"

	"github.com/micro/go-micro/v2"
	"github.com/micro/go-micro/v2/config"
	"github.com/micro/go-micro/v2/config/source/env"
	log "github.com/micro/go-micro/v2/logger"
	"github.com/micro/go-micro/v2/server/grpc"
)

func main() {
	// New Service
	service := micro.NewService(
		micro.Name("com.zhigui.xian.kodo.qiniu"),
		micro.Version("latest"),
	)

	// Load configuration from env
	if err := config.Load(env.NewSource(env.WithStrippedPrefix("QINIU_KODO"))); err != nil {
		log.Fatal(err)
	}

	var maxMsgSize int
	if err := config.Get("max", "message", "size", "mb").Scan(&maxMsgSize); err != nil {
		log.Fatal(err)
	}

	//GRPC 默认大小4mb
	if maxMsgSize > 0 {
		_ = service.Server().Init(grpc.MaxMsgSize(1024 * 1024 * maxMsgSize))
		log.Info(fmt.Sprintf("Set services max message %d mb", maxMsgSize))
	}

	// Initialise service
	service.Init()

	// Register Handler
	qiniu.RegisterQiniuHandler(service.Server(), new(handler.Qiniu))

	// Register Struct as Subscriber
	//micro.RegisterSubscriber("go.micro.xian.Kodo.qiniu", service.Server(), new(subscriber.Qiniu))

	// Run service
	if err := service.Run(); err != nil {
		log.Fatal(err)
	}
}