40 lines
1.1 KiB
Go
40 lines
1.1 KiB
Go
package db
|
|
|
|
import (
|
|
schsdk "gitlink.org.cn/cloudream/common/sdks/scheduler"
|
|
schmod "gitlink.org.cn/cloudream/scheduler/common/models"
|
|
)
|
|
|
|
type ModelsDB struct {
|
|
*DB
|
|
}
|
|
|
|
func (db *DB) Models() *ModelsDB {
|
|
return &ModelsDB{DB: db}
|
|
}
|
|
|
|
//func (*ModelsDB) GetAll(ctx SQLContext) ([]schmod.Models, error) {
|
|
// var ret []schmod.Models
|
|
// err := sqlx.Select(ctx, &ret, "select * from Models")
|
|
//
|
|
// return ret, err
|
|
//}
|
|
|
|
func (*ModelsDB) GetAll(ctx SQLContext) ([]schmod.Models, error) {
|
|
var ret []schmod.Models
|
|
err := ctx.Table("Models").Find(&ret).Error
|
|
return ret, err
|
|
}
|
|
|
|
//func (*ModelsDB) GetModelByID(ctx SQLContext, modelID schsdk.ModelID, OjbStgID int64) (schmod.ModelResource, error) {
|
|
// var ret schmod.ModelResource
|
|
// err := sqlx.Get(ctx, &ret, "select * from ModelResource where modelID = ? and OjbStgID = ?", modelID, OjbStgID)
|
|
// return ret, err
|
|
//}
|
|
|
|
func (*ModelsDB) GetModelByID(ctx SQLContext, modelID schsdk.ModelID, OjbStgID int64) (schmod.ModelResource, error) {
|
|
var ret schmod.ModelResource
|
|
err := ctx.Table("ModelResource").Where("modelID = ? AND OjbStgID = ?", modelID, OjbStgID).First(&ret).Error
|
|
return ret, err
|
|
}
|