workbench count device、alarm

app/device/controller/workbenchController.go

@@ -19,10 +19,38 @@ func (c *workbenchCtl) CountDevice(ctx *gin.Context) {
 		ctx.JSON(http.StatusUnauthorized, util.NormalResponse(http.StatusUnauthorized, "", nil))
 	}
 
-	device, err := service.WorkbenchService.CountDevice(value.(*middleware.JwtToken).TenantId)
+	devices, err := service.WorkbenchService.CountDevice(value.(*middleware.JwtToken).TenantId)
 	if err != nil {
 		ctx.JSON(http.StatusOK, err)
 		return
 	}
-	ctx.JSON(http.StatusOK, util.SuccessResponse(util.Succeeded, device))
+	ctx.JSON(http.StatusOK, util.SuccessResponse(util.Succeeded, devices))
+}
+
+func (c *workbenchCtl) CountAlarm(ctx *gin.Context) {
+	value, isExist := ctx.Get(middleware.Authorization)
+	if !isExist || value == nil {
+		ctx.JSON(http.StatusUnauthorized, util.NormalResponse(http.StatusUnauthorized, "", nil))
+	}
+
+	alarms, err := service.WorkbenchService.CountAlarm(value.(*middleware.JwtToken).TenantId)
+	if err != nil {
+		ctx.JSON(http.StatusOK, err)
+		return
+	}
+	ctx.JSON(http.StatusOK, util.SuccessResponse(util.Succeeded, alarms))
+}
+
+func (c *workbenchCtl) Aqi(ctx *gin.Context) {
+	value, isExist := ctx.Get(middleware.Authorization)
+	if !isExist || value == nil {
+		ctx.JSON(http.StatusUnauthorized, util.NormalResponse(http.StatusUnauthorized, "", nil))
+	}
+
+	alarms, err := service.WorkbenchService.CountAlarm(value.(*middleware.JwtToken).TenantId)
+	if err != nil {
+		ctx.JSON(http.StatusOK, err)
+		return
+	}
+	ctx.JSON(http.StatusOK, util.SuccessResponse(util.Succeeded, alarms))
 }

app/device/dao/workbenchDao.go

@@ -43,3 +43,14 @@ func GetDeviceCount(tenantId string) ([]CountDevice, error) {
 	result = append(result, count)
 	return result, nil
 }
+
+type CountAlarm struct {
+	LastYearAlarm  int `json:"lastYearAlarm"`
+	LastMonthAlarm int `json:"lastMonthAlarm"`
+	Backlog        int `json:"backlog"`
+}
+
+func GetAlarmCount(tenantId string) (*CountAlarm, error) {
+	var alarm CountAlarm
+	return &alarm, nil
+}

app/device/service/optoSensoService.go

@@ -54,6 +54,7 @@ func (s *optoSensorService) CreateOrUpdate(req dao.OptoSensor) *util.Errors {
 	//todo operation record
 	return util.SuccessResponse(util.Succeeded, nil)
 }
+
 func (s *optoSensorService) List(searchValue string, current, size int) ([]dao.OptoSensor, *util.Errors) {
 	device := dao.OptoSensor{}
 	if searchValue != "" {
@@ -69,6 +70,7 @@ func (s *optoSensorService) List(searchValue string, current, size int) ([]dao.O
 	}
 	return devices, nil
 }
+
 func (s *optoSensorService) Remove(id int) *util.Errors {
 	// 创建查询实例
 	device := &dao.OptoSensor{

app/device/service/workbenchService.go

@@ -11,10 +11,17 @@ var WorkbenchService = new(workbenchService)
 type workbenchService struct{}
 
 func (s *workbenchService) CountDevice(tenantId string) ([]dao.CountDevice, *util.Errors) {
-	// 创建查询实例
 	counts, err := dao.GetDeviceCount(tenantId)
 	if err != nil {
 		return nil, util.FailResponse(err.Error(), nil)
 	}
 	return counts, nil
 }
+
+func (s *workbenchService) CountAlarm(tenantId string) (*dao.CountAlarm, *util.Errors) {
+	count, err := dao.GetAlarmCount(tenantId)
+	if err != nil {
+		return nil, util.FailResponse(err.Error(), nil)
+	}
+	return count, nil
+}

router/router.go

@@ -413,5 +413,7 @@ func InitRouter(engine *gin.Engine) {
 	workbench := engine.Group("/api/longchi/report/workbench/")
 	{
 		workbench.POST("countdevice", controller.Workbench.CountDevice)
+		workbench.POST("countalarm", controller.Workbench.CountAlarm)
+		workbench.POST("aqi", controller.Workbench.Aqi)
 	}
 }