增加管理后台

admin/resp.go

@@ -0,0 +1,62 @@
+package admin
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"runtime"
+
+	"github.com/bjdgyc/anylink/base"
+)
+
+type Resp struct {
+	Code     int         `json:"code"`
+	Msg      string      `json:"msg"`
+	Location string      `json:"location"`
+	Data     interface{} `json:"data"`
+}
+
+func respHttp(w http.ResponseWriter, respCode int, data interface{}, errS ...interface{}) {
+	resp := Resp{
+		Code: respCode,
+		Msg:  "success",
+		Data: data,
+	}
+	_, file, line, _ := runtime.Caller(2)
+	resp.Location = fmt.Sprintf("%v:%v", file, line)
+
+	if respCode != 0 {
+		resp.Msg = ""
+		if v, ok := RespMap[respCode]; ok {
+			resp.Msg += v
+		}
+
+		if len(errS) > 0 {
+			resp.Msg += fmt.Sprint(errS...)
+		}
+	}
+
+	b, err := json.Marshal(resp)
+	if err != nil {
+		base.Error(err, resp)
+	}
+
+	w.Header().Set("Content-Type", "application/json; charset=utf-8")
+	w.WriteHeader(http.StatusOK)
+	w.Write(b)
+
+	// 记录返回数据
+	// logger.Category("response").Debug(string(b))
+}
+
+func RespSucess(w http.ResponseWriter, data interface{}) {
+	respHttp(w, 0, data, "")
+}
+
+func RespError(w http.ResponseWriter, respCode int, errS ...interface{}) {
+	respHttp(w, respCode, nil, errS...)
+}
+
+func RespData(w http.ResponseWriter, data interface{}, err error) {
+	respHttp(w, http.StatusOK, data, "")
+}