添加分区逻辑代码

legend/service/areaService.go

@@ -0,0 +1,98 @@
+package service
+
+import (
+	"github.com/rs/xid"
+	"legend/models/legend"
+	"legend/request"
+	"legend/response"
+	"legend/utils"
+)
+
+type AreaService struct {
+	BaseService
+}
+
+func (c *AreaService) AddArea(req *request.AreaReq) *response.BaseResp {
+	resp := new(response.BaseResp)
+
+	area := new(legend.Area)
+	area.AreaName = req.AreaName
+	area.GroupName = req.GroupName
+	area.TemplateName = req.TemplateName
+	area.NotifyUrl = req.NotifyUrl
+	area.AttachParams = req.AttachParams
+	area.UpdateTime = utils.GetNowTime()
+	area.CreateTime = utils.GetNowTime()
+	area.Uid = xid.New().String()
+
+	if legend.InsertArea(area) {
+		resp.Code = 0
+	} else {
+		resp.Code = -1
+		resp.Msg = "添加分区失败"
+	}
+
+	return resp
+}
+
+func (c *AreaService) AreaList(page, limit int) *response.AreaListResp {
+	offset := utils.CountOffset(page, limit)
+	count := legend.GetAreaAllCount()
+	areas := legend.GetAreaList(offset, limit)
+
+	for i, _ := range areas {
+		areas[i].Id = offset + i + 1
+	}
+
+	areaResp := new(response.AreaListResp)
+	areaResp.Code = 0
+	areaResp.Count = count
+	areaResp.Data = areas
+
+	return areaResp
+}
+
+func (c *AreaService) DeleteArea(uid string) *response.BaseResp {
+	resp := new(response.BaseResp)
+
+	if legend.DeleteAreaByUid(uid) {
+		resp.Code = 0
+	} else {
+		resp.Code = -1
+		resp.Msg = "删除分区失败"
+	}
+	return resp
+}
+
+func (c *AreaService) GetArea(uid string) *response.AreaInfoResp {
+	resp := new(response.AreaInfoResp)
+	resp.Code = 0
+	resp.Msg = "请求成功"
+
+	area := legend.GetAreaByUid(uid)
+	resp.Area = area
+
+	return resp
+}
+
+func (c *AreaService) EditArea(req *request.AreaReq, uid string) *response.BaseResp {
+	resp := new(response.BaseResp)
+	resp.Code = -1
+
+	area := legend.GetAreaByUid(uid)
+	if area == nil || area.AreaName == "" {
+		resp.Msg = "更新失败"
+	} else {
+		area.UpdateTime = utils.GetNowTime()
+		area.GroupName = req.GroupName
+		area.TemplateName = req.TemplateName
+		area.NotifyUrl = req.NotifyUrl
+		area.AttachParams = req.AttachParams
+
+		if legend.UpdateArea(area) {
+			resp.Msg = "更新失败"
+		}
+	}
+
+	return resp
+}