main.go

package main

import (
	"dkv/api/kv"
	"dkv/api/oss"
	"dkv/api/redis"
	"dkv/component/cache"
	"dkv/component/config"
	"dkv/component/constant"
	"dkv/component/logger"
	"dkv/component/metrics"
	"dkv/component/pprof"
	"dkv/store"
	_ "dkv/store/synchronous"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/sirupsen/logrus"
)

func main() {
	// 启动存储引擎
	defer store.S.Close()
	defer cache.C.Close()

	// 启动http服务
	gin.SetMode(constant.Mode)
	engine := gin.New()
	engine.Use(gin.LoggerWithConfig(gin.LoggerConfig{Output: logger.L.WriterLevel(logrus.DebugLevel)}), gin.RecoveryWithWriter(logger.L.WriterLevel(logrus.ErrorLevel)))
	kv.Route(engine)     // 注册KV服务接口
	oss.Route(engine)    // 注册OSS服务接口
	metrics.Use(engine)  // 注册普罗米修斯接口
	pprof.Wrap(engine)   // 注册pprof接口
	config.Route(engine) // 配置服务接口
	redis.ListenAndServe()
	addr := fmt.Sprintf(":%d", constant.Port)
	logger.L.Infof("Listening and serving HTTP on : %s", addr)
	if err := engine.Run(addr); err != nil {
		logger.L.Fatalf("Fatal error gin: %v\n", err)
	}
}