初始

api/project/index_count.go

@@ -0,0 +1,162 @@
+package project
+
+import (
+	"context"
+	"encoding/json"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/glog"
+)
+
+// PvModuleStatsReq 用于查询统计信息的请求结构体
+type PvModuleStatsReq struct {
+	g.Meta       `path:"moduleStats" method:"get" tags:"项目首页相关" summary:"获取模块的统计信息"`
+	SubProjectId string `json:"subProjectId"`
+	FangzhenId   string `json:"fangzhenId"`
+}
+
+// PvModuleStatsRes 用于封装统计结果的响应结构体，调整为使用字符串描述类型
+type PvModuleStatsRes struct {
+	Data []struct {
+		TypeDesc string `json:"typeDesc"`
+		Total    int    `json:"total"`
+		Complete int    `json:"complete"`
+	} `json:"data"`
+}
+
+func (p Project) QueryPvModuleStats(ctx context.Context, req *PvModuleStatsReq) (res *PvModuleStatsRes, err error) {
+	model := g.Model("pv_module").Fields("type, COUNT(*) as total, SUM(CASE WHEN status='2' THEN 1 ELSE 0 END) as complete").Group("type")
+	if req.SubProjectId != "" {
+		model = model.Where("sub_projectid", req.SubProjectId)
+	}
+	if req.FangzhenId != "" {
+		model = model.Where("fangzhen_id", req.FangzhenId)
+	}
+
+	var stats []struct {
+		Type     int `json:"type"`
+		Total    int `json:"total"`
+		Complete int `json:"complete"`
+	}
+
+	err = model.Scan(&stats)
+	if err != nil {
+		glog.Error(ctx, "查询 pv_module 统计信息失败：", err)
+		return nil, err
+	}
+
+	var resultStats []struct {
+		TypeDesc string `json:"typeDesc"`
+		Total    int    `json:"total"`
+		Complete int    `json:"complete"`
+	}
+
+	for _, stat := range stats {
+		typeDesc, exists := ConstMap[stat.Type]
+		if !exists {
+			glog.Error(ctx, "未知的 type：", stat.Type)
+			continue // 或者处理未知类型
+		}
+		resultStats = append(resultStats, struct {
+			TypeDesc string `json:"typeDesc"`
+			Total    int    `json:"total"`
+			Complete int    `json:"complete"`
+		}{TypeDesc: typeDesc, Total: stat.Total, Complete: stat.Complete})
+	}
+
+	return &PvModuleStatsRes{Data: resultStats}, nil
+}
+
+// 修改请求结构体以接收类型切片
+type PvModuleQueryReq struct {
+	g.Meta       `path:"indexCount" method:"get" tags:"项目首页相关" summary:"根据子项目ID、方针ID和类型来获取数据"`
+	SubProjectId string `json:"subProjectId"`
+	FangzhenId   string `json:"fangzhenId"`
+	Types        []int  `json:"types"` // 新增字段，用于指定需要查询的类型
+}
+
+// 修改返回结果结构体，去除 total 和 complete
+type PvModuleQueryRes struct {
+	Data map[int][]PVModule `json:"data"`
+	View string             `json:"view" orm:"view"`
+}
+
+func (p Project) QueryPVModule(ctx context.Context, req *PvModuleQueryReq) (res *PvModuleQueryRes, err error) {
+	res = new(PvModuleQueryRes)
+	// 查询对应方针的 View 值
+	g.Model("qianqi_fangzhen").Where("id", req.FangzhenId).Scan(&res)
+	// 准备存储每种 type 对应的数据
+	data := make(map[int][]PVModule)
+
+	// 仅针对请求中指定的 type 进行查询
+	for _, t := range req.Types {
+		var modules []PVModule
+		queryModel := g.Model("pv_module").Where("type", t)
+		if req.SubProjectId != "" {
+			queryModel = queryModel.Where("sub_projectid", req.SubProjectId)
+		}
+		if req.FangzhenId != "" {
+			queryModel = queryModel.Where("fangzhen_id", req.FangzhenId)
+		}
+
+		err = queryModel.Scan(&modules)
+		if err != nil {
+			continue // 或处理错误
+		}
+
+		// 对 modules 进行处理，比如反序列化 Detail 字段、添加倾角和方位角等
+		var processedModules []PVModule
+		for _, m := range modules {
+			// 反序列化 Detail 字段
+			var detailMap map[string]interface{}
+			err = json.Unmarshal([]byte(m.Detail), &detailMap)
+			if err != nil {
+				continue
+			}
+
+			// 为其添加倾角和方位角
+			//detailMap["roll"] = m.Tilt
+			//detailMap["heading"] = m.Azimuth
+			// 重新序列化
+			serializedDetail, _ := json.Marshal(detailMap)
+
+			processedModules = append(processedModules, PVModule{
+				ID:           m.ID,
+				FangzhenID:   m.FangzhenID,
+				SubProjectID: m.SubProjectID,
+				WorkID:       m.WorkID,
+				Name:         m.Name,
+				Status:       m.Status,
+				DoneTime:     m.DoneTime,
+				Detail:       string(serializedDetail),
+				Type:         t,
+				DeviceID:     m.DeviceID,
+				EquipmentSn:  "",
+			})
+		}
+
+		// 遍历processedModules中那些DeviceID不为空的值，取出数据，然后查询equipment表，根据设备id查询equipment表，得到设备sn，赋值给EquipmentSn
+		for i, mod := range processedModules {
+			if mod.DeviceID != "" {
+				var equipment struct {
+					EquipmentSn string
+				}
+				// 使用g.DB模型查询equipment表中的SN字段，假定设备的序列号字段为SN
+				err := g.Model("equipment").Fields("equipmentSn").Where("id", mod.DeviceID).Scan(&equipment)
+				if err != nil {
+					// 处理查询错误，例如记录日志或跳过
+					continue
+				}
+				// 更新模块的EquipmentSn字段
+				processedModules[i].EquipmentSn = equipment.EquipmentSn
+			}
+		}
+
+		// 更新数据存储结构
+		data[t] = processedModules
+	}
+
+	res.Data = data
+
+	return res, nil
+}