init

pkg/config/config.go

@@ -0,0 +1,19 @@
+package config
+
+import (
+	"github.com/spf13/viper"
+)
+
+var Config *viper.Viper
+
+func Init(configPath string) error {
+	Config = viper.New()
+	Config.SetConfigFile(configPath)
+	Config.SetConfigType("yaml")
+
+	if err := Config.ReadInConfig(); err != nil {
+		return err
+	}
+
+	return nil
+}

pkg/response/response.go

@@ -0,0 +1,60 @@
+package response
+
+import (
+	"github.com/gin-gonic/gin"
+	"net/http"
+)
+
+type Response struct {
+	Code int         `json:"code"`
+	Msg  string      `json:"msg"`
+	Data interface{} `json:"data"`
+}
+
+const (
+	SUCCESS     = 200
+	ERROR       = 500
+	UNAUTHORIZED = 401
+	NOT_FOUND   = 404
+	BAD_REQUEST = 400
+)
+
+var MsgFlags = map[int]string{
+	SUCCESS:     "操作成功",
+	ERROR:       "操作失败",
+	UNAUTHORIZED: "未授权",
+	NOT_FOUND:   "资源不存在",
+	BAD_REQUEST: "请求参数错误",
+}
+
+func GetMsg(code int) string {
+	msg, ok := MsgFlags[code]
+	if !ok {
+		return MsgFlags[ERROR]
+	}
+	return msg
+}
+
+func Success(c *gin.Context, data interface{}) {
+	c.JSON(http.StatusOK, Response{
+		Code: SUCCESS,
+		Msg:  "success",
+		Data: data,
+	})
+}
+
+func Error(c *gin.Context, code int, msg string) {
+	c.JSON(http.StatusOK, Response{
+		Code: code,
+		Msg:  msg,
+		Data: nil,
+	})
+}
+
+func Unauthorized(c *gin.Context) {
+	c.JSON(http.StatusUnauthorized, Response{
+		Code: UNAUTHORIZED,
+		Msg:  "未授权",
+		Data: nil,
+	})
+}