|
|
@@ -2,6 +2,8 @@ package backend
|
|
|
|
|
|
import (
|
|
|
"context"
|
|
|
+ "errors"
|
|
|
+ "fmt"
|
|
|
"kpt-pasture/model"
|
|
|
"kpt-pasture/util"
|
|
|
"net/http"
|
|
|
@@ -9,6 +11,8 @@ import (
|
|
|
"strings"
|
|
|
"time"
|
|
|
|
|
|
+ "gorm.io/gorm"
|
|
|
+
|
|
|
"gitee.com/xuyiping_admin/pkg/logger/zaplog"
|
|
|
"go.uber.org/zap"
|
|
|
|
|
|
@@ -179,8 +183,169 @@ func (s *StoreEntry) FocusIndicatorsSet(ctx context.Context, req *pasturePb.Inde
|
|
|
}
|
|
|
|
|
|
userFocusIndicators := strings.Join(req.IndicatorsKind, ",")
|
|
|
- if err = s.DB.Model(new(model.SystemUser)).Where("id = ?", userModel.SystemUser.Id).Update("indicators_kinds", userFocusIndicators).Error; err != nil {
|
|
|
+ if err = s.DB.Model(new(model.SystemUser)).
|
|
|
+ Where("id = ?", userModel.SystemUser.Id).
|
|
|
+ Update("indicators_kinds", userFocusIndicators).Error; err != nil {
|
|
|
return xerr.WithStack(err)
|
|
|
}
|
|
|
return nil
|
|
|
}
|
|
|
+
|
|
|
+func (s *StoreEntry) DataWarningSet(ctx context.Context, req *pasturePb.IndexDataWarningSetRequest) error {
|
|
|
+ userModel, err := s.GetUserModel(ctx)
|
|
|
+ if err != nil {
|
|
|
+ return xerr.WithStack(err)
|
|
|
+ }
|
|
|
+ if len(req.WarningDataSet) <= 0 {
|
|
|
+ return nil
|
|
|
+ }
|
|
|
+ currentUserDataWarningList, err := s.FindDataWarning(ctx, userModel.SystemUser.Id)
|
|
|
+ fmt.Println(currentUserDataWarningList)
|
|
|
+ if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
|
+ // 如果不存在则新增
|
|
|
+ defaultDataWarningMap, err := s.defaultDataWarningMap(ctx)
|
|
|
+ if err != nil {
|
|
|
+ return xerr.WithStack(err)
|
|
|
+ }
|
|
|
+ dataWarningList := model.NewDataWarningList(userModel.SystemUser.Id, req.WarningDataSet, defaultDataWarningMap)
|
|
|
+ fmt.Println(dataWarningList)
|
|
|
+ } else {
|
|
|
+ return xerr.WithStack(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ return nil
|
|
|
+}
|
|
|
+
|
|
|
+func (s *StoreEntry) DataWarningList(ctx context.Context) (*pasturePb.IndexDataWarningResponse, error) {
|
|
|
+ userModel, err := s.GetUserModel(ctx)
|
|
|
+ if err != nil {
|
|
|
+ return nil, xerr.WithStack(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ defaultUserDataWarningList, err := s.FindDataWarning(ctx, model.DefaultUserId)
|
|
|
+ if err != nil {
|
|
|
+ return nil, xerr.Custom("默认预警数据有误，请联系管理员！")
|
|
|
+ }
|
|
|
+
|
|
|
+ var isExist bool // 判断是否存在自己的设置的数据
|
|
|
+ currentUserDataWarningList, err := s.FindDataWarning(ctx, userModel.SystemUser.Id)
|
|
|
+ if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
|
+ // 如果用户没有配置自己的预警数据，则使用默认数据
|
|
|
+ isExist = true
|
|
|
+ currentUserDataWarningList = defaultUserDataWarningList
|
|
|
+ } else {
|
|
|
+ return nil, xerr.WithStack(err)
|
|
|
+ }
|
|
|
+
|
|
|
+ newTime := time.Now().Unix()
|
|
|
+ needUpdateWarningIds := make([]int64, 0)
|
|
|
+ warningMap := make(map[int64]*model.DataWarning)
|
|
|
+ for _, warningData := range currentUserDataWarningList {
|
|
|
+
|
|
|
+ warningMap[warningData.Id] = warningData
|
|
|
+ // 如果预警数据更新时间大于预警条件更新时间，并且更新时间距离当前时间小于2小时，则跳过
|
|
|
+ if warningData.DataUpdateAt > warningData.ConditionUpdateAt && newTime-warningData.DataUpdateAt < int64(2*time.Hour) {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ needUpdateWarningIds = append(needUpdateWarningIds, warningData.Id)
|
|
|
+ }
|
|
|
+
|
|
|
+ // 需要重新计算更新的warningId
|
|
|
+ if len(needUpdateWarningIds) > 0 {
|
|
|
+ s.UpdateWarningData(ctx, needUpdateWarningIds)
|
|
|
+ }
|
|
|
+ if isExist {
|
|
|
+ currentUserDataWarningList, _ = s.FindDataWarning(ctx, model.DefaultUserId)
|
|
|
+ } else {
|
|
|
+ currentUserDataWarningList, _ = s.FindDataWarning(ctx, userModel.SystemUser.Id)
|
|
|
+ }
|
|
|
+
|
|
|
+ dataSet := make([]*pasturePb.WarningDataSet, 0)
|
|
|
+ dataShow := make([]*pasturePb.WarningDataShow, 0)
|
|
|
+ for _, warningData := range currentUserDataWarningList {
|
|
|
+ dataUpdateTimeFormat := ""
|
|
|
+ if warningData.DataUpdateAt > 0 {
|
|
|
+ dataUpdateTimeFormat = time.Unix(warningData.DataUpdateAt, 0).Format(model.LayoutTime)
|
|
|
+ }
|
|
|
+ dataShow = append(dataShow, &pasturePb.WarningDataShow{
|
|
|
+ Name: warningData.Name,
|
|
|
+ Number: warningData.DataValue,
|
|
|
+ Describe: warningData.Description,
|
|
|
+ DataUpdateTimeFormat: dataUpdateTimeFormat,
|
|
|
+ })
|
|
|
+ }
|
|
|
+
|
|
|
+ defaultDataWarningItems, err := s.FindDataWarningItems(ctx, model.DefaultUserId)
|
|
|
+
|
|
|
+ fmt.Println(defaultDataWarningItems)
|
|
|
+
|
|
|
+ return &pasturePb.IndexDataWarningResponse{
|
|
|
+ Code: http.StatusOK,
|
|
|
+ Msg: "ok",
|
|
|
+ Data: &pasturePb.DataWarning{
|
|
|
+ DataSet: dataSet,
|
|
|
+ DataShow: dataShow,
|
|
|
+ },
|
|
|
+ }, nil
|
|
|
+}
|
|
|
+
|
|
|
+func (s *StoreEntry) FindDataWarning(ctx context.Context, userId int64) ([]*model.DataWarning, error) {
|
|
|
+ defaultUserDataWarningList := make([]*model.DataWarning, 0)
|
|
|
+ if err := s.DB.Model(new(model.DataWarning)).
|
|
|
+ Where("user_id = ?", userId).
|
|
|
+ Find(&defaultUserDataWarningList).Error; err != nil {
|
|
|
+ return nil, xerr.WithStack(err)
|
|
|
+ }
|
|
|
+ return defaultUserDataWarningList, nil
|
|
|
+}
|
|
|
+
|
|
|
+func (s *StoreEntry) FindDataWarningItems(ctx context.Context, userId int64) ([]*model.DataWarningItems, error) {
|
|
|
+ defaultUserDataWarningItemsList := make([]*model.DataWarningItems, 0)
|
|
|
+ if err := s.DB.Model(new(model.DataWarningItems)).
|
|
|
+ Where("user_id = ?", userId).
|
|
|
+ Find(&defaultUserDataWarningItemsList).Error; err != nil {
|
|
|
+ return nil, xerr.WithStack(err)
|
|
|
+ }
|
|
|
+ return defaultUserDataWarningItemsList, nil
|
|
|
+}
|
|
|
+
|
|
|
+func (s *StoreEntry) defaultDataWarningMap(ctx context.Context) (map[string]*model.DataWarning, error) {
|
|
|
+ defaultDataWarning, err := s.FindDataWarning(ctx, model.DefaultUserId)
|
|
|
+ if err != nil {
|
|
|
+ return nil, xerr.Custom("默认预警数据有误，请联系管理员！")
|
|
|
+ }
|
|
|
+ defaultDataWarningMap := make(map[string]*model.DataWarning)
|
|
|
+ for _, v := range defaultDataWarning {
|
|
|
+ defaultDataWarningMap[v.Kind] = v
|
|
|
+ }
|
|
|
+ return defaultDataWarningMap, nil
|
|
|
+}
|
|
|
+
|
|
|
+// UpdateWarningData 更新计算数据
|
|
|
+func (s *StoreEntry) UpdateWarningData(ctx context.Context, needUpdateWarningIds []int64) {
|
|
|
+ if len(needUpdateWarningIds) <= 0 {
|
|
|
+ return
|
|
|
+ }
|
|
|
+ for _, warningId := range needUpdateWarningIds {
|
|
|
+ query, params, err := s.BuildQuery(warningId)
|
|
|
+ if err != nil {
|
|
|
+ zaplog.Error("UpdateWarningData", zap.Any("BuildQuery", err), zap.Any("warningId", warningId))
|
|
|
+ }
|
|
|
+ if len(query) == 0 || len(params) == 0 {
|
|
|
+ continue
|
|
|
+ }
|
|
|
+ var count int64
|
|
|
+ if err = s.DB.Model(new(model.Cow)).Where(query, params...).Count(&count).Error; err != nil {
|
|
|
+ zaplog.Error("UpdateWarningData", zap.Any("err", err), zap.Any("query", query), zap.Any("params", params))
|
|
|
+ }
|
|
|
+
|
|
|
+ if err = s.DB.Model(new(model.DataWarning)).
|
|
|
+ Where("id = ?", warningId).
|
|
|
+ Updates(map[string]interface{}{
|
|
|
+ "data_value": count,
|
|
|
+ "data_update_at": time.Now().Unix(),
|
|
|
+ }).Error; err != nil {
|
|
|
+ zaplog.Error("UpdateWarningData", zap.Any("update", err))
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
Yi