fyp/logic/models/types.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
}