完成重构数据库操作代码

server/api/api.go

@@ -0,0 +1,75 @@
+package api
+
+import (
+	"github.com/labstack/echo/v4"
+	"next-terminal/server/constant"
+	"next-terminal/server/global"
+	"next-terminal/server/model"
+)
+
+type H map[string]interface{}
+
+func Fail(c echo.Context, code int, message string) error {
+	return c.JSON(200, H{
+		"code":    code,
+		"message": message,
+	})
+}
+
+func FailWithData(c echo.Context, code int, message string, data interface{}) error {
+	return c.JSON(200, H{
+		"code":    code,
+		"message": message,
+		"data":    data,
+	})
+}
+
+func Success(c echo.Context, data interface{}) error {
+	return c.JSON(200, H{
+		"code":    1,
+		"message": "success",
+		"data":    data,
+	})
+}
+
+func NotFound(c echo.Context, message string) error {
+	return c.JSON(200, H{
+		"code":    -1,
+		"message": message,
+	})
+}
+
+func GetToken(c echo.Context) string {
+	token := c.Request().Header.Get(Token)
+	if len(token) > 0 {
+		return token
+	}
+	return c.QueryParam(Token)
+}
+
+func GetCurrentAccount(c echo.Context) (model.User, bool) {
+	token := GetToken(c)
+	cacheKey := BuildCacheKeyByToken(token)
+	get, b := global.Cache.Get(cacheKey)
+	if b {
+		return get.(Authorization).User, true
+	}
+	return model.User{}, false
+}
+
+func HasPermission(c echo.Context, owner string) bool {
+	// 检测是否登录
+	account, found := GetCurrentAccount(c)
+	if !found {
+		return false
+	}
+	// 检测是否为管理人员
+	if constant.TypeAdmin == account.Type {
+		return true
+	}
+	// 检测是否为所有者
+	if owner == account.ID {
+		return true
+	}
+	return false
+}