44 lines
703 B
Go
44 lines
703 B
Go
package models
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
)
|
|
|
|
type BaseModel struct {
|
|
Name string
|
|
Status int
|
|
Id string
|
|
}
|
|
|
|
const (
|
|
FAILED_PREPARING_ZIP_FILE = -2
|
|
FAILED_PREPARING = -1
|
|
|
|
PREPARING = 1
|
|
CONFIRM_PRE_TRAINING = 2
|
|
PREPARING_ZIP_FILE = 3
|
|
)
|
|
|
|
var ModelNotFoundError = errors.New("Model not found error")
|
|
|
|
func getBaseModel(db *sql.DB, id string) (base *BaseModel, err error) {
|
|
rows, err := db.Query("select name, status, id from models where id=$1;", id)
|
|
if err != nil {
|
|
return
|
|
}
|
|
defer rows.Close()
|
|
|
|
if !rows.Next() {
|
|
return nil, ModelNotFoundError
|
|
}
|
|
|
|
base = &BaseModel{}
|
|
err = rows.Scan(&base.Name, &base.Status, &base.Id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return
|
|
}
|