chore: started working on #38
This commit is contained in:
@@ -8,16 +8,21 @@ import (
|
||||
var FailedToGetIdAfterInsertError = errors.New("Failed to Get Id After Insert Error")
|
||||
|
||||
func AddDataPoint(db *sql.DB, class_id string, file_path string, mode DATA_POINT_MODE) (id string, err error) {
|
||||
id = ""
|
||||
result, err := db.Query("insert into model_data_point (class_id, file_path, model_mode) values ($1, $2, $3) returning id;", class_id, file_path, mode)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer result.Close()
|
||||
if !result.Next() {
|
||||
err = FailedToGetIdAfterInsertError
|
||||
return
|
||||
}
|
||||
err = result.Scan(&id)
|
||||
id = ""
|
||||
result, err := db.Query("insert into model_data_point (class_id, file_path, model_mode, status) values ($1, $2, $3, 1) returning id;", class_id, file_path, mode)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer result.Close()
|
||||
if !result.Next() {
|
||||
err = FailedToGetIdAfterInsertError
|
||||
return
|
||||
}
|
||||
err = result.Scan(&id)
|
||||
return
|
||||
}
|
||||
|
||||
func UpdateDataPointStatus(db *sql.DB, data_point_id string, status int, message *string) (err error) {
|
||||
_, err = db.Exec("update model_data_point set status=$1, status_message=$2 where id=$3", status, message, data_point_id)
|
||||
return
|
||||
}
|
||||
|
||||
@@ -71,3 +71,14 @@ func CreateClass(db *sql.DB, model_id string, order int, name string) (id string
|
||||
err = rows.Scan(&id)
|
||||
return
|
||||
}
|
||||
|
||||
func GetNumberOfWrongDataPoints(db *sql.DB, model_id string) (number int, err error) {
|
||||
number = 0
|
||||
rows, err := db.Query("select count(mdp.id) from model_data_point as mdp join model_classes as mc on mc.id = mdp.class_id where mc.model_id=$1 and mdp.status=-1;", model_id)
|
||||
if err != nil { return }
|
||||
defer rows.Close()
|
||||
// TODO not an error because if there is no result means that there is no need to count
|
||||
if !rows.Next() { return }
|
||||
err = rows.Scan(&number)
|
||||
return
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user