cold-delivery/app/admin/controller/cooler_box.go

package controller

import (
	"cold-delivery/app/admin/model"
	"cold-delivery/app/admin/service"
	"cold-delivery/app/admin/service/dto"
	"cold-delivery/common/actions"
	"cold-delivery/common/nats/nats_server"
	"github.com/gin-gonic/gin"
	"github.com/gin-gonic/gin/binding"
	"gogs.baozhida.cn/zoie/OAuth-core/api"
	"gogs.baozhida.cn/zoie/OAuth-core/pkg/jwtauth/user"
	_ "gogs.baozhida.cn/zoie/OAuth-core/pkg/response"
)

type CoolerBoxController struct {
	api.Api
}

// GetPage 获取保温箱列表
// @Summary 获取保温箱列表
// @Description 获取保温箱列表
// @Tags 保温箱
// @Param name query string false "保温箱名称"
// @Param pageSize query int false "页条数"
// @Param page query int false "页码"
// @Success 200 {object} response.Response{data=response.Page{list=[]model.CoolerBox}} "{"code": 200, "data": [...]}"
// @Router /api/CoolerBox [get]
// @Security Bearer
func (e CoolerBoxController) GetPage(c *gin.Context) {
	s := service.CoolerBox{}
	req := dto.CoolerBoxGetPageReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.Query).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	//数据权限检查
	p := actions.GetPermissionFromContext(c)

	list := make([]model.CoolerBox, 0)
	var count int64

	err = s.GetPage(&req, &list, &count, p)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	e.PageOK(list, int(count), req.GetPageIndex(), req.GetPageSize(), "查询成功")
}

// Get 通过id获取保温箱
// @Summary 通过id获取保温箱
// @Description 通过id获取保温箱
// @Tags 保温箱
// @Param id path string true "保温箱id"
// @Success 200 {object} response.Response{data=model.CoolerBox} "{"code": 200, "data": [...]}"
// @Router /api/CoolerBox/{id} [get]
// @Security Bearer
func (e CoolerBoxController) Get(c *gin.Context) {
	s := service.CoolerBox{}
	req := dto.CoolerBoxGetReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, nil).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	var object model.CoolerBox
	p := actions.GetPermissionFromContext(c)

	//数据权限检查
	err = s.Get(&req, &object, p)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}

	e.OK(object, "查询成功")
}

// Insert 添加保温箱
// @Summary 添加保温箱
// @Description 添加保温箱
// @Tags 保温箱
// @Accept  application/json
// @Product application/json
// @Param data body dto.CoolerBoxInsertReq true "data"
// @Success 200 {string} string	"{"code": 200, "message": "添加成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "添加失败"}"
// @Router /api/CoolerBox [post]
// @Security Bearer
func (e CoolerBoxController) Insert(c *gin.Context) {
	s := service.CoolerBox{}
	req := dto.CoolerBoxInsertReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	p := actions.GetPermissionFromContext(c)
	// 设置创建人
	req.SetCreateBy(user.GetUserId(c))
	req.SetDeptId(p.DeptId)
	err = s.Insert(&req)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	e.OK(req.GetId(), "创建成功")
}

// Update 修改保温箱
// @Summary 修改保温箱
// @Description 修改保温箱
// @Tags 保温箱
// @Accept  application/json
// @Product application/json
// @Param data body dto.CoolerBoxUpdateReq true "body"
// @Success 200 {string} string	"{"code": 200, "message": "添加成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "添加失败"}"
// @Router /api/CoolerBox [put]
// @Security Bearer
func (e CoolerBoxController) Update(c *gin.Context) {
	s := service.CoolerBox{}
	req := dto.CoolerBoxUpdateReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	p := actions.GetPermissionFromContext(c)

	req.SetUpdateBy(user.GetUserId(c))
	err = s.Update(&req, p)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	e.OK(req.GetId(), "更新成功")
}

// Delete 删除保温箱
// @Summary 删除保温箱
// @Description 删除保温箱
// @Tags 保温箱
// @Accept  application/json
// @Product application/json
// @Param data body dto.CoolerBoxDeleteReq true "body"
// @Success 200 {string} string	"{"code": 200, "message": "删除成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "删除失败"}"
// @Router /api/CoolerBox [delete]
// @Security Bearer
func (e CoolerBoxController) Delete(c *gin.Context) {
	s := service.CoolerBox{}
	req := dto.CoolerBoxDeleteReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON, nil).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	//数据权限检查
	p := actions.GetPermissionFromContext(c)
	err = s.Remove(&req, p)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	e.OK(req.GetId(), "删除成功")
}

// GetPage 获取保温箱列表
// @Summary 获取保温箱列表
// @Description 获取保温箱列表
// @Tags 保温箱
// @Param name query string false "保温箱名称"
// @Param pageSize query int false "页条数"
// @Param page query int false "页码"
// @Success 200 {object} response.Response{data=response.Page{list=[]model.CoolerBox}} "{"code": 200, "data": [...]}"
// @Router /api/CoolerBox [get]
// @Security Bearer
func (e CoolerBoxController) GetDevicePage(c *gin.Context) {
	s := service.CoolerBox{}
	req := dto.DeviceGetPageReq{}
	err := e.MakeContext(c).
		MakeOrm().
		MakeService(&s.Service).
		Bind(&req, binding.Query).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	list := make([]nats_server.Device_R, 0)
	p := actions.GetPermissionFromContext(c)

	var coolerBoxCount int64
	coolerBoxList := make([]model.CoolerBox, 0)

	coolerBoxReq := dto.CoolerBoxGetPageReq{}
	coolerBoxReq.PageSize = 9999
	err = s.GetPage(&coolerBoxReq, &coolerBoxList, &coolerBoxCount, p)

	var count int64
	company, err := model.GetCompanyById(p.DeptId)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	list, count, err = nats_server.Cold_ReadDevice_List(req.Name, company.ColdKey, req.GetPageIndex(), 9999)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	var coolerBoxMap = make(map[string]model.CoolerBox)
	for _, box := range coolerBoxList {
		coolerBoxMap[box.Sn] = box
	}
	// Fixme 测试完成之后删除
	//for i := 0; i < 20; i++ {
	//	list = append(list, nats_server.Device_R{
	//		T_devName: fmt.Sprintf("保温箱%03d", i),
	//		T_sn:      fmt.Sprintf("2024114956064%03d", i),
	//	})
	//}

	list2 := make([]nats_server.Device_R, 0)
	for i := 0; i < len(list); i++ {
		if _, ok := coolerBoxMap[list[i].T_sn]; ok {
			count--
			continue
		}
		list2 = append(list2, list[i])
	}

	//if len(list2) == 0 {
	//	e.PageOK(list2, len(list2), req.GetPageIndex(), req.GetPageSize(), "查询成功")
	//}

	// 计算总页数
	totalPages := (len(list2) + req.GetPageSize() - 1) / req.GetPageSize()

	// 如果n超出范围，返回错误
	if req.GetPageIndex() > totalPages {
		req.Page = totalPages
	}
	// 计算当前页的数据
	start := (req.GetPageIndex() - 1) * req.GetPageSize()
	end := start + req.GetPageSize()
	if end > len(list2) {
		end = len(list2)
	}
	currentPage := list2[start:end]

	e.PageOK(currentPage, len(list2), req.GetPageIndex(), req.GetPageSize(), "查询成功")
}

// BatchInsert 批量添加保温箱
// @Summary 批量添加保温箱
// @Description 批量添加保温箱
// @Tags 保温箱
// @Accept  application/json
// @Product application/json
// @Param data body dto.CoolerBoxInsertReq true "data"
// @Success 200 {string} string	"{"code": 200, "message": "添加成功"}"
// @Success 200 {string} string	"{"code": -1, "message": "添加失败"}"
// @Router /api/CoolerBox [post]
// @Security Bearer
func (e CoolerBoxController) BatchInsert(c *gin.Context) {
	s := service.CoolerBox{}
	req := dto.CoolerBoxBatchInsertReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	p := actions.GetPermissionFromContext(c)
	// 设置创建人
	req.SetCreateBy(user.GetUserId(c))
	req.SetDeptId(p.DeptId)
	err = s.BatchInsert(&req)
	if err != nil {
		e.Error(500, err, err.Error())
		return
	}
	e.OK(nil, "创建成功")
}