kpt-pasture/model/data_waring.go

package model

import pasturePb "gitee.com/xuyiping_admin/go_proto/proto/go/backend/cow"

const DefaultUserId = 0

type DataWarning struct {
	Id                int64  `json:"id"`
	UserId            int64  `json:"userId"`
	Kind              string `json:"kind"`
	Name              string `json:"name"`
	Description       string `json:"description"`
	DataValue         string `json:"dataValue"`
	DataUpdateAt      int64  `json:"dataUpdateAt"`
	ConditionUpdateAt int64  `json:"conditionUpdateAt"`
	CreatedAt         int64  `json:"createdAt"`
	UpdatedAt         int64  `json:"updatedAt"`
}

func (d *DataWarning) TableName() string {
	return "data_warning"
}

func NewDataWarningList(userId int64, req []*pasturePb.WarningDataSet, warningMap map[string]*DataWarning) []*DataWarning {
	res := make([]*DataWarning, 0)
	for _, v := range req {
		defaultDataWarning := warningMap[v.Kind]
		res = append(res, NewDataWarning(userId, v.Kind, defaultDataWarning))
	}
	return res
}

func NewDataWarning(userId int64, Kind string, defaultDataWarning *DataWarning) *DataWarning {
	return &DataWarning{
		UserId:      userId,
		Kind:        Kind,
		Name:        defaultDataWarning.Name,
		Description: defaultDataWarning.Description,
	}
}