v/vlib/mysql/orm.v

297 lines
5.3 KiB
V

module mysql
import orm
import time
type Prims = byte | f32 | f64 | i16 | i64 | i8 | int | string | u16 | u32 | u64
// sql expr
pub fn (db Connection) @select(config orm.SelectConfig, data orm.QueryData, where orm.QueryData) ?[][]orm.Primitive {
query := orm.orm_select_gen(config, '`', false, '?', 0, where)
mut ret := [][]orm.Primitive{}
mut stmt := db.init_stmt(query)
stmt.prepare() ?
mysql_stmt_binder(mut stmt, where) ?
mysql_stmt_binder(mut stmt, data) ?
if data.data.len > 0 || where.data.len > 0 {
stmt.bind_params() ?
}
mut status := stmt.execute() ?
num_fields := stmt.get_field_count()
metadata := stmt.gen_metadata()
fields := stmt.fetch_fields(metadata)
mut dataptr := []Prims{}
for i in 0 .. num_fields {
f := unsafe { fields[i] }
match FieldType(f.@type) {
.type_tiny {
dataptr << u8(0)
}
.type_short {
dataptr << u16(0)
}
.type_long {
dataptr << u32(0)
}
.type_longlong {
dataptr << u64(0)
}
.type_float {
dataptr << f32(0)
}
.type_double {
dataptr << f64(0)
}
.type_string {
dataptr << ''
}
else {
dataptr << u8(0)
}
}
}
mut vptr := []&char{}
for d in dataptr {
vptr << d.get_data_ptr()
}
unsafe { dataptr.free() }
lens := []u32{len: int(num_fields), init: 0}
stmt.bind_res(fields, vptr, lens, num_fields)
stmt.bind_result_buffer() ?
stmt.store_result() ?
mut row := 0
for {
status = stmt.fetch_stmt() ?
if status == 1 || status == 100 {
break
}
row++
data_list := buffer_to_primitive(vptr, config.types) ?
ret << data_list
}
stmt.close() ?
return ret
}
// sql stmt
pub fn (db Connection) insert(table string, data orm.QueryData) ? {
query := orm.orm_stmt_gen(table, '`', .insert, false, '?', 1, data, orm.QueryData{})
mysql_stmt_worker(db, query, data, orm.QueryData{}) ?
}
pub fn (db Connection) update(table string, data orm.QueryData, where orm.QueryData) ? {
query := orm.orm_stmt_gen(table, '`', .update, false, '?', 1, data, where)
mysql_stmt_worker(db, query, data, where) ?
}
pub fn (db Connection) delete(table string, where orm.QueryData) ? {
query := orm.orm_stmt_gen(table, '`', .delete, false, '?', 1, orm.QueryData{}, where)
mysql_stmt_worker(db, query, orm.QueryData{}, where) ?
}
pub fn (db Connection) last_id() orm.Primitive {
query := 'SELECT last_insert_id();'
id := db.query(query) or {
Result{
result: 0
}
}
return orm.Primitive(id.rows()[0].vals[0].int())
}
// table
pub fn (db Connection) create(table string, fields []orm.TableField) ? {
query := orm.orm_table_gen(table, '`', false, 0, fields, mysql_type_from_v, false) or {
return err
}
mysql_stmt_worker(db, query, orm.QueryData{}, orm.QueryData{}) ?
}
pub fn (db Connection) drop(table string) ? {
query := 'DROP TABLE `$table`;'
mysql_stmt_worker(db, query, orm.QueryData{}, orm.QueryData{}) ?
}
fn mysql_stmt_worker(db Connection, query string, data orm.QueryData, where orm.QueryData) ? {
mut stmt := db.init_stmt(query)
stmt.prepare() ?
mysql_stmt_binder(mut stmt, data) ?
mysql_stmt_binder(mut stmt, where) ?
if data.data.len > 0 || where.data.len > 0 {
stmt.bind_params() ?
}
stmt.execute() ?
stmt.close() ?
}
fn mysql_stmt_binder(mut stmt Stmt, d orm.QueryData) ? {
for data in d.data {
stmt_binder_match(mut stmt, data)
}
}
fn stmt_binder_match(mut stmt Stmt, data orm.Primitive) {
match data {
bool {
stmt.bind_bool(&data)
}
i8 {
stmt.bind_i8(&data)
}
i16 {
stmt.bind_i16(&data)
}
int {
stmt.bind_int(&data)
}
i64 {
stmt.bind_i64(&data)
}
byte {
stmt.bind_u8(&data)
}
u16 {
stmt.bind_u16(&data)
}
u32 {
stmt.bind_u32(&data)
}
u64 {
stmt.bind_u64(&data)
}
f32 {
stmt.bind_f32(unsafe { &f32(&data) })
}
f64 {
stmt.bind_f64(unsafe { &f64(&data) })
}
string {
stmt.bind_text(data)
}
time.Time {
stmt.bind_int(&int(data.unix))
}
orm.InfixType {
stmt_binder_match(mut stmt, data.right)
}
}
}
fn buffer_to_primitive(data_list []&char, types []int) ?[]orm.Primitive {
mut res := []orm.Primitive{}
for i, data in data_list {
mut primitive := orm.Primitive(0)
match types[i] {
5 {
primitive = *(&i8(data))
}
6 {
primitive = *(&i16(data))
}
7, -1 {
primitive = *(&int(data))
}
8 {
primitive = *(&i64(data))
}
9 {
primitive = *(&u8(data))
}
10 {
primitive = *(&u16(data))
}
11 {
primitive = *(&u32(data))
}
12 {
primitive = *(&u64(data))
}
13 {
primitive = *(&f32(data))
}
14 {
primitive = *(&f64(data))
}
15 {
primitive = *(&bool(data))
}
orm.string {
primitive = unsafe { cstring_to_vstring(&char(data)) }
}
orm.time {
timestamp := *(&int(data))
primitive = time.unix(timestamp)
}
else {
return error('Unknown type ${types[i]}')
}
}
res << primitive
}
return res
}
fn mysql_type_from_v(typ int) ?string {
str := match typ {
5, 9, 16 {
'TINYINT'
}
6, 10 {
'SMALLINT'
}
7, 11, orm.time {
'INT'
}
8, 12 {
'BIGINT'
}
13 {
'FLOAT'
}
14 {
'DOUBLE'
}
orm.string {
'TEXT'
}
-1 {
'SERIAL'
}
else {
''
}
}
if str == '' {
return error('Unknown type $typ')
}
return str
}
fn (p Prims) get_data_ptr() &char {
return match p {
string {
p.str
}
else {
&char(&p)
}
}
}