feat: 添加获取设备物模型统计信息的接口及相关逻辑

Author: hezhuozhuo

internal/api/device.go

@@ -206,6 +206,23 @@ func (*DeviceApi) HandleDeviceTemplateMenu(c *gin.Context) {
 	c.Set("data", data)
 }
 
+// HandleDeviceTemplateStats 获取设备物模型统计信息
+// @Router   /api/v1/device/template/stats [get]
+func (*DeviceApi) HandleDeviceTemplateStats(c *gin.Context) {
+	var req model.GetDeviceTemplateStatsReq
+	if !BindAndValidate(c, &req) {
+		return
+	}
+	userClaims := c.MustGet("claims").(*utils.UserClaims)
+	data, err := service.GroupApp.DeviceTemplate.GetDeviceTemplateStats(req, userClaims)
+	if err != nil {
+		c.Error(err)
+		return
+	}
+
+	c.Set("data", data)
+}
+
 // DeleteDeviceTemplate 删除设备模版
 // @Router   /api/v1/device/template/{id} [delete]
 func (*DeviceApi) DeleteDeviceTemplate(c *gin.Context) {

internal/dal/device_template.go

@@ -118,3 +118,59 @@ func GetDeviceTemplateMenu(req *model.GetDeviceTemplateMenuReq, claims *utils.Us
 	return data, err
 
 }
+
+// GetDeviceTemplateStats 获取设备物模型统计信息
+func GetDeviceTemplateStats(deviceTemplateID string, tenantID string) (*model.GetDeviceTemplateStatsRsp, error) {
+	ctx := context.Background()
+
+	// 查询物模型基本信息
+	dt := query.DeviceTemplate
+	template, err := dt.WithContext(ctx).
+		Where(dt.ID.Eq(deviceTemplateID), dt.TenantID.Eq(tenantID)).
+		First()
+	if err != nil {
+		logrus.Error("query device template error: ", err)
+		return nil, err
+	}
+
+	// 统计关联设备总数和在线设备数
+	// 通过 device_configs 表关联 devices 表
+	dc := query.DeviceConfig
+	d := query.Device
+
+	// 统计总设备数
+	totalDevices, err := d.WithContext(ctx).
+		Join(dc, dc.ID.EqCol(d.DeviceConfigID)).
+		Where(dc.DeviceTemplateID.Eq(deviceTemplateID), d.TenantID.Eq(tenantID)).
+		Count()
+	if err != nil {
+		logrus.Error("query total devices error: ", err)
+		return nil, err
+	}
+
+	// 统计在线设备数
+	onlineDevices, err := d.WithContext(ctx).
+		Join(dc, dc.ID.EqCol(d.DeviceConfigID)).
+		Where(dc.DeviceTemplateID.Eq(deviceTemplateID), d.TenantID.Eq(tenantID), d.IsOnline.Eq(1)).
+		Count()
+	if err != nil {
+		logrus.Error("query online devices error: ", err)
+		return nil, err
+	}
+
+	// 构造返回结果
+	label := ""
+	if template.Label != nil {
+		label = *template.Label
+	}
+
+	result := &model.GetDeviceTemplateStatsRsp{
+		DeviceTemplateID: template.ID,
+		Name:             template.Name,
+		Label:            label,
+		TotalDevices:     totalDevices,
+		OnlineDevices:    onlineDevices,
+	}
+
+	return result, nil
+}

internal/model/device_templates.http.go

@@ -50,3 +50,17 @@ type GetDeviceTemplateRsp struct {
 	CreatedAt    time.Time              `json:"created_at"`
 	UpdatedAt    time.Time              `json:"updated_at"`
 }
+
+// GetDeviceTemplateStatsReq 获取设备物模型统计请求
+type GetDeviceTemplateStatsReq struct {
+	DeviceTemplateID string `json:"device_template_id" form:"device_template_id" validate:"required,max=36"` // 物模型ID
+}
+
+// GetDeviceTemplateStatsRsp 获取设备物模型统计响应
+type GetDeviceTemplateStatsRsp struct {
+	DeviceTemplateID string `json:"device_template_id"` // 物模型ID
+	Name             string `json:"name"`               // 物模型名称
+	Label            string `json:"label"`              // 标签
+	TotalDevices     int64  `json:"total_devices"`      // 关联设备总数
+	OnlineDevices    int64  `json:"online_devices"`     // 在线设备数
+}

internal/service/device_template.go

@@ -194,3 +194,14 @@ func (*DeviceTemplate) GetDeviceTemplateMenu(req model.GetDeviceTemplateMenuReq,
 	}
 	return data, nil
 }
+
+// GetDeviceTemplateStats 获取设备物模型统计信息
+func (*DeviceTemplate) GetDeviceTemplateStats(req model.GetDeviceTemplateStatsReq, claims *utils.UserClaims) (*model.GetDeviceTemplateStatsRsp, error) {
+	data, err := dal.GetDeviceTemplateStats(req.DeviceTemplateID, claims.TenantID)
+	if err != nil {
+		return nil, errcode.WithData(errcode.CodeDBError, map[string]interface{}{
+			"sql_error": err.Error(),
+		})
+	}
+	return data, nil
+}

router/apps/device.go

@@ -108,6 +108,9 @@ func (*Device) InitDevice(Router *gin.RouterGroup) {
 		// 模板下拉菜单
 		deviceTemplateapi.GET("/menu", api.Controllers.DeviceApi.HandleDeviceTemplateMenu)
 
+		// 物模型统计信息
+		deviceTemplateapi.GET("/stats", api.Controllers.DeviceApi.HandleDeviceTemplateStats)
+
 		// 根据设备ID获取模板
 		deviceTemplateapi.GET("/chart", api.Controllers.DeviceApi.HandleDeviceTemplateByDeviceId)