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, "创建成功") }