29 lines
658 B
Go
29 lines
658 B
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"modelRT/orm"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// QueryBayDevColumnNames returns the bay table columns exposed as token7
|
|
// candidates under token6=bay.
|
|
func QueryBayDevColumnNames(ctx context.Context, db *gorm.DB) ([]string, error) {
|
|
columnTypes, err := db.WithContext(ctx).Migrator().ColumnTypes((&orm.Bay{}).TableName())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
columnNames := make([]string, 0, len(columnTypes))
|
|
for _, columnType := range columnTypes {
|
|
columnName := columnType.Name()
|
|
if strings.HasPrefix(columnName, "dev_") {
|
|
columnNames = append(columnNames, columnName)
|
|
}
|
|
}
|
|
return columnNames, nil
|
|
}
|