mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-10 19:30:25 +08:00
80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
package controllers
|
||
|
||
import (
|
||
"encoding/json"
|
||
"mayfly-go/base/biz"
|
||
"mayfly-go/base/ctx"
|
||
"mayfly-go/base/ginx"
|
||
"mayfly-go/base/rediscli"
|
||
"mayfly-go/base/utils"
|
||
"mayfly-go/mock-server/controllers/form"
|
||
)
|
||
|
||
const key = "ccbscf:mock:data"
|
||
|
||
// @router /api/mock-datas/:method [get]
|
||
func GetMockData(rc *ctx.ReqCtx) {
|
||
g := rc.GinCtx
|
||
method := g.Param("method")
|
||
params := utils.MapBuilder("method", method).ToMap()
|
||
// 调用该mock数据的用户，若该数据指定了生效用户，则需要校验是否可访问
|
||
username := g.Query("username")
|
||
if username != "" {
|
||
params["username"] = username
|
||
}
|
||
// 记录日志使用
|
||
rc.ReqParam = params
|
||
|
||
mockData := &form.MockData{}
|
||
// 从redis中获取key为 ‘ccbscf:mock:data’，field为‘method’的hash值
|
||
json.Unmarshal([]byte(rediscli.HGet(key, method)), mockData)
|
||
// 数据不存在或者状态为禁用
|
||
biz.IsTrue(mockData.Enable == 1, "无该mock数据")
|
||
|
||
eu := mockData.EffectiveUser
|
||
// 如果设置的生效用户为空，则表示所有用户都生效
|
||
if len(eu) == 0 {
|
||
rc.ResData = mockData.Data
|
||
return
|
||
}
|
||
biz.IsTrue(utils.StrLen(username) != 0, "该用户无法访问该mock数据")
|
||
// 判断该用户是否在该数据指定的生效用户中
|
||
for _, e := range eu {
|
||
if username == e {
|
||
rc.ResData = mockData.Data
|
||
return
|
||
}
|
||
}
|
||
panic(biz.NewBizErr("该用户无法访问该mock数据"))
|
||
}
|
||
|
||
// @router /api/mock-datas [put]
|
||
func UpdateMockData(rc *ctx.ReqCtx) {
|
||
mockData := &form.MockData{}
|
||
ginx.BindJsonAndValid(rc.GinCtx, mockData)
|
||
rc.ReqParam = mockData.Method
|
||
val, _ := json.Marshal(mockData)
|
||
rediscli.HSet(key, mockData.Method, val)
|
||
}
|
||
|
||
// @router /api/mock-datas [post]
|
||
func CreateMockData(rc *ctx.ReqCtx) {
|
||
mockData := &form.MockData{}
|
||
ginx.BindJsonAndValid(rc.GinCtx, mockData)
|
||
biz.IsTrue(!rediscli.HExist(key, mockData.Method), "该方法已存在")
|
||
val, _ := json.Marshal(mockData)
|
||
rediscli.HSet(key, mockData.Method, val)
|
||
}
|
||
|
||
// @router /api/mock-datas [get]
|
||
func GetAllData(rc *ctx.ReqCtx) {
|
||
rc.ResData = rediscli.HGetAll(key)
|
||
}
|
||
|
||
// @router /api/mock-datas/:method [delete]
|
||
func DeleteMockData(rc *ctx.ReqCtx) {
|
||
method := rc.GinCtx.Param("method")
|
||
rc.ReqParam = method
|
||
rediscli.HDel(key, method)
|
||
}
|