cc/server/api/v1/autocode/unit.go

package autocode

import (
	"github.com/flipped-aurora/gin-vue-admin/server/global"
	"github.com/flipped-aurora/gin-vue-admin/server/model/autocode"
	autocodeReq "github.com/flipped-aurora/gin-vue-admin/server/model/autocode/request"
	"github.com/flipped-aurora/gin-vue-admin/server/model/common/request"
	"github.com/flipped-aurora/gin-vue-admin/server/model/common/response"
	"github.com/flipped-aurora/gin-vue-admin/server/service"
	"github.com/gin-gonic/gin"
	"go.uber.org/zap"
	"time"
)

type UnitApi struct {
}

var unitService = service.ServiceGroupApp.AutoCodeServiceGroup.UnitService

// CreateUnit 创建Unit
// @Tags Unit
// @Summary 创建Unit
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data body autocode.Unit true "创建Unit"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"获取成功"}"
// @Router /unit/createUnit [post]
func (unitApi *UnitApi) CreateUnit(c *gin.Context) {
	var unit autocode.Unit
	_ = c.ShouldBindJSON(&unit)
	if err, nowUnit := unitService.GetUnitTop(); err != nil {
		//新建期数
		unit.Period = time.Now().Format("20060102150405")
		integral := 1000
		unit.UnitIntegral = &integral
		unit.UnitMaxIntegral = &integral
		unitService.CreateUnit(unit)
		response.OkWithMessage("创建成功", c)
	} else {
		unit.UnitIntegral = nowUnit.UnitMaxIntegral
		unit.UnitMaxIntegral = nowUnit.UnitMaxIntegral
		unit.Period = nowUnit.Period
		if err := unitService.CreateUnit(unit); err != nil {
			global.GVA_LOG.Error("创建失败!", zap.Any("err", err))
			response.FailWithMessage("创建失败", c)
		} else {
			response.OkWithMessage("创建成功", c)
		}
	}
}

// DeleteUnit 删除Unit
// @Tags Unit
// @Summary 删除Unit
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data body autocode.Unit true "删除Unit"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"删除成功"}"
// @Router /unit/deleteUnit [delete]
func (unitApi *UnitApi) DeleteUnit(c *gin.Context) {
	var unit autocode.Unit
	_ = c.ShouldBindJSON(&unit)
	if err := unitService.DeleteUnit(unit); err != nil {
		global.GVA_LOG.Error("删除失败!", zap.Any("err", err))
		response.FailWithMessage("删除失败", c)
	} else {
		response.OkWithMessage("删除成功", c)
	}
}

// DeleteUnitByIds 批量删除Unit
// @Tags Unit
// @Summary 批量删除Unit
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data body request.IdsReq true "批量删除Unit"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"批量删除成功"}"
// @Router /unit/deleteUnitByIds [delete]
func (unitApi *UnitApi) DeleteUnitByIds(c *gin.Context) {
	var IDS request.IdsReq
	_ = c.ShouldBindJSON(&IDS)
	if err := unitService.DeleteUnitByIds(IDS); err != nil {
		global.GVA_LOG.Error("批量删除失败!", zap.Any("err", err))
		response.FailWithMessage("批量删除失败", c)
	} else {
		response.OkWithMessage("批量删除成功", c)
	}
}

// UpdateUnit 更新Unit
// @Tags Unit
// @Summary 更新Unit
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data body autocode.Unit true "更新Unit"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"更新成功"}"
// @Router /unit/updateUnit [put]
func (unitApi *UnitApi) UpdateUnit(c *gin.Context) {
	var unit autocode.Unit
	_ = c.ShouldBindJSON(&unit)
	//记录历史,更新所有
	if err, units := unitService.GetUnitInfoListAll(); err != nil {
		response.FailWithMessage("更新失败", c)
		return
	} else {
		for _, temp := range units {
			unitHistoryService.CreateUnitHistory(autocode.UnitHistory{UnitId: &temp.ID, UnitName: temp.UnitName, UnitIntegral: temp.UnitIntegral, UnitMaxIntegral: temp.UnitMaxIntegral, Period: temp.Period})
		}
	}

	if err := unitService.UpdateUnitNext(unit); err != nil {
		global.GVA_LOG.Error("更新失败!", zap.Any("err", err))
		response.FailWithMessage("更新失败", c)
	} else {
		response.OkWithMessage("更新成功", c)
	}
}

// FindUnit 用id查询Unit
// @Tags Unit
// @Summary 用id查询Unit
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data query autocode.Unit true "用id查询Unit"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"查询成功"}"
// @Router /unit/findUnit [get]
func (unitApi *UnitApi) FindUnit(c *gin.Context) {
	var unit autocode.Unit
	_ = c.ShouldBindQuery(&unit)
	if err, reunit := unitService.GetUnit(unit.ID); err != nil {
		global.GVA_LOG.Error("查询失败!", zap.Any("err", err))
		response.FailWithMessage("查询失败", c)
	} else {
		response.OkWithData(gin.H{"reunit": reunit}, c)
	}
}

// GetUnitList 分页获取Unit列表
// @Tags Unit
// @Summary 分页获取Unit列表
// @Security ApiKeyAuth
// @accept application/json
// @Produce application/json
// @Param data query autocodeReq.UnitSearch true "分页获取Unit列表"
// @Success 200 {string} string "{"success":true,"data":{},"msg":"获取成功"}"
// @Router /unit/getUnitList [get]
func (unitApi *UnitApi) GetUnitList(c *gin.Context) {
	var pageInfo autocodeReq.UnitSearch
	_ = c.ShouldBindQuery(&pageInfo)
	if err, list, total := unitService.GetUnitInfoList(pageInfo); err != nil {
		global.GVA_LOG.Error("获取失败!", zap.Any("err", err))
		response.FailWithMessage("获取失败", c)
	} else {
		response.OkWithDetailed(response.PageResult{
			List:     list,
			Total:    total,
			Page:     pageInfo.Page,
			PageSize: pageInfo.PageSize,
		}, "获取成功", c)
	}
}