feat(db): added function to convert sqlite output to struct

Jef Roosens 2022-05-28 22:22:55 +02:00 committed by Chewing_Bever
parent 31e903ebeb
commit 4f32dec5b5
Signed by untrusted user: Jef Roosens
GPG Key ID: B75D4F293C7052DB
3 changed files with 29 additions and 9 deletions

View File

@ -1,6 +1,7 @@
module db module db
import sqlite import sqlite
import time
struct VieterDb { struct VieterDb {
conn sqlite.DB conn sqlite.DB
@ -66,3 +67,21 @@ pub fn init(db_path string) ?VieterDb {
conn: conn conn: conn
} }
} }
pub fn row_into<T>(row sqlite.Row) T {
mut i := 0
mut out := T{}
$for field in T.fields {
$if field.typ is string {
out.$(field.name) = row.vals[i]
} $else $if field.typ is int {
out.$(field.name) = row.vals[i].int()
} $else $if field.typ is time.Time {
out.$(field.name) = time.unix(row.vals[i].int())
}
i += 1
}
return out
}

View File

@ -19,18 +19,19 @@ pub fn (db &VieterDb) get_build_logs(filter BuildLogFilter) []BuildLog {
where_parts << 'start_time < $filter.after.unix_time()' where_parts << 'start_time < $filter.after.unix_time()'
} }
where_str := if where_parts.len > 0 { mut where_str := ''
' where ' + where_parts.map('($it)').join(' and ')
} else { if where_parts.len > 0 {
'' where_str = ' where ' + where_parts.map('($it)').join(' and ')
} }
query := 'select from BuildLog' + where_str query := 'select from BuildLog' + where_str
res := db.conn.exec(query) rows, _ := db.conn.exec(query)
res := rows.map(row_into<BuildLog>(it))
/* res := sql db.conn { */ // res := sql db.conn {
/* select from BuildLog where filter.repo == 0 || repo_id == filter.repo order by id */ // select from BuildLog where filter.repo == 0 || repo_id == filter.repo order by id
/* } */ // }
return res return res
} }

View File

@ -3,7 +3,7 @@ module models
import time import time
pub struct BuildLog { pub struct BuildLog {
pub: pub mut:
id int [primary; sql: serial] id int [primary; sql: serial]
repo_id int [nonull] repo_id int [nonull]
start_time time.Time [nonull] start_time time.Time [nonull]