workbench

app/device/controller/workbenchController.go

@@ -0,0 +1,28 @@
+package controller
+
+import (
+	"github.com/gin-gonic/gin"
+	"iot_manager_service/app/device/service"
+	"iot_manager_service/app/middleware"
+	"iot_manager_service/util"
+	"net/http"
+)
+
+// 工作台管理对象
+var Workbench = new(workbenchCtl)
+
+type workbenchCtl struct{}
+
+func (c *workbenchCtl) CountDevice(ctx *gin.Context) {
+	value, isExist := ctx.Get(middleware.Authorization)
+	if !isExist || value == nil {
+		ctx.JSON(http.StatusUnauthorized, util.NormalResponse(http.StatusUnauthorized, "", nil))
+	}
+
+	device, 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))
+}

app/device/dao/workbenchDao.go

@@ -0,0 +1,45 @@
+package dao
+
+type CountDevice struct {
+	CountNum   int    `json:"countNum"`
+	DeviceType string `json:"deviceType"`
+}
+
+func GetDeviceCount(tenantId string) ([]CountDevice, error) {
+	var result []CountDevice
+	var count CountDevice
+	Db.Debug().Model(&LampPole{}).Select(" count(*) count_num, 'lamppole' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&CameraDevice{}).Select(" count(*) count_num, 'camera' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&Gateway{}).Select(" count(*) count_num, 'gateway' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&LightControl{}).Select(" count(*) count_num, 'control' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&SwitchBox{}).Select(" count(*) count_num, 'box' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&InfoBoard{}).Select(" count(*) count_num, 'board' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&OptoSensor{}).Select(" count(*) count_num, 'sensor' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&Zigbee{}).Select(" count(*) count_num, 'zigbee' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&AlarmTerminal{}).Select(" count(*) count_num, 'terminal' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&Alarm{}).Select(" count(*) count_num, 'alarmserve' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	Db.Debug().Model(&IpBroadcast{}).Select(" count(*) count_num, 'ipcast' device_type ").
+		Where("tenant_id = ? and is_deleted = 0", tenantId).Scan(&count)
+	result = append(result, count)
+	return result, nil
+}

app/device/model/workbench.go

@@ -0,0 +1,7 @@
+package model
+
+import "iot_manager_service/app/device/dao"
+
+type RspCountDevice struct {
+	CountDevices []dao.CountDevice `json:"data"`
+}

app/device/service/workbenchService.go

@@ -0,0 +1,20 @@
+package service
+
+import (
+	"iot_manager_service/app/device/dao"
+	"iot_manager_service/util"
+)
+
+// 中间件管理服务
+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
+}

config/config.yaml

@@ -18,13 +18,12 @@ database:
   host: "127.0.0.1"
   user: "root"
   password: "root"
-  #password : "lczm*2019"
+  #password : "lczm@2019oS"
   port: "3306"
-  name: "ipole_iot"
+  name: "iot"
   prefix: "t_"
   timezone: "Asia/Shanghai"
 
 redis:
   host: "127.0.0.1:6379"
-  user: "root"
-  password: "root"
+  #password: "lczm*2019"

router/router.go

@@ -408,4 +408,10 @@ func InitRouter(engine *gin.Engine) {
 		menu.GET("top-menu", system.Menu.TopMenu)
 		menu.GET("authRoutes", system.Menu.Routes)
 	}
+
+	//工作台
+	workbench := engine.Group("/api/longchi/report/workbench/")
+	{
+		workbench.POST("countdevice", controller.Workbench.CountDevice)
+	}
 }