80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
// Package database define database operation functions
|
|
package database
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"strings"
|
|
|
|
"modelRT/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// ParseAttrToken define return the attribute model interface based on the input attribute token
|
|
func ParseAttrToken(ctx context.Context, tx *gorm.DB, attrToken string) (model.AttrModelInterface, error) {
|
|
attrSlice := strings.Split(attrToken, ".")
|
|
attrLen := len(attrSlice)
|
|
if attrLen == 4 {
|
|
short := &model.ShortAttrInfo{
|
|
AttrGroupName: attrSlice[2],
|
|
AttrKey: attrSlice[3],
|
|
// TODO use redis query
|
|
// AttrValue: attrSlice[2],
|
|
}
|
|
err := FillingShortAttrModel(ctx, tx, attrSlice, short)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return short, nil
|
|
} else if attrLen == 7 {
|
|
long := &model.LongAttrInfo{
|
|
AttrGroupName: attrSlice[5],
|
|
AttrKey: attrSlice[6],
|
|
// TODO use redis query
|
|
// AttrValue: attrSlice[5],
|
|
}
|
|
err := FillingLongAttrModel(ctx, tx, attrSlice, long)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return long, nil
|
|
}
|
|
return nil, errors.New("invalid attribute token format")
|
|
}
|
|
|
|
// FillingShortAttrModel define filling short attribute model info
|
|
func FillingShortAttrModel(ctx context.Context, tx *gorm.DB, attrItems []string, attrModel *model.ShortAttrInfo) error {
|
|
component, err := QueryComponentByNsPath(ctx, tx, attrItems[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
attrModel.ComponentInfo = &component
|
|
return nil
|
|
}
|
|
|
|
// FillingLongAttrModel define filling long attribute model info
|
|
func FillingLongAttrModel(ctx context.Context, tx *gorm.DB, attrItems []string, attrModel *model.LongAttrInfo) error {
|
|
grid, err := QueryGridByName(ctx, tx, attrItems[0])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
attrModel.GridInfo = &grid
|
|
zone, err := QueryZoneByName(ctx, tx, attrItems[1])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
attrModel.ZoneInfo = &zone
|
|
station, err := QueryStationByName(ctx, tx, attrItems[2])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
attrModel.StationInfo = &station
|
|
component, err := QueryComponentByNsPath(ctx, tx, attrItems[3])
|
|
if err != nil {
|
|
return err
|
|
}
|
|
attrModel.ComponentInfo = &component
|
|
return nil
|
|
}
|