aboutsummaryrefslogtreecommitdiff
path: root/v_windows/v/old/vlib/mysql/orm.v
diff options
context:
space:
mode:
authorIndrajith K L2022-12-03 17:00:20 +0530
committerIndrajith K L2022-12-03 17:00:20 +0530
commitf5c4671bfbad96bf346bd7e9a21fc4317b4959df (patch)
tree2764fc62da58f2ba8da7ed341643fc359873142f /v_windows/v/old/vlib/mysql/orm.v
downloadcli-tools-windows-f5c4671bfbad96bf346bd7e9a21fc4317b4959df.tar.gz
cli-tools-windows-f5c4671bfbad96bf346bd7e9a21fc4317b4959df.tar.bz2
cli-tools-windows-f5c4671bfbad96bf346bd7e9a21fc4317b4959df.zip
Adds most of the toolsHEADmaster
Diffstat (limited to 'v_windows/v/old/vlib/mysql/orm.v')
-rw-r--r--v_windows/v/old/vlib/mysql/orm.v296
1 files changed, 296 insertions, 0 deletions
diff --git a/v_windows/v/old/vlib/mysql/orm.v b/v_windows/v/old/vlib/mysql/orm.v
new file mode 100644
index 0000000..c68be63
--- /dev/null
+++ b/v_windows/v/old/vlib/mysql/orm.v
@@ -0,0 +1,296 @@
+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 << byte(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 << byte(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_rowid();'
+ 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_byte(&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 = *(&byte(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 {
+ '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)
+ }
+ }
+}