v/vlib/orm/orm_mut_connection_test.v

// vtest retry: 3
import orm

struct Database {
mut:
	last_query string
}

struct User {
	id   int @[primary; sql: serial]
	name string
}

fn (mut db Database) query(sql_stmt string) ! {
	db.last_query = sql_stmt
}

// select is used internally by V's ORM for processing `SELECT` queries
fn (mut db Database) select(config orm.SelectConfig, data orm.QueryData, where orm.QueryData) ![][]orm.Primitive {
	// 1. Create query and bind necessary data
	query := orm.orm_select_gen(config, '', true, ':', 1, where)
	mut ret := [][]orm.Primitive{}
	return ret
}

// insert is used internally by V's ORM for processing `INSERT` queries
fn (mut db Database) insert(table orm.Table, data orm.QueryData) ! {
	query, _ := orm.orm_stmt_gen(.sqlite, table, '', .insert, false, '?', 1, data, orm.QueryData{})

	db.query(query)!
}

// update is used internally by V's ORM for processing `UPDATE` queries
fn (mut db Database) update(table orm.Table, data orm.QueryData, where orm.QueryData) ! {
	mut query, _ := orm.orm_stmt_gen(.sqlite, table, '', .update, true, ':', 1, data,
		where)

	db.query(query)!
}

// delete is used internally by V's ORM for processing `DELETE ` queries
fn (mut db Database) delete(table orm.Table, where orm.QueryData) ! {
	query, converted := orm.orm_stmt_gen(.sqlite, table, '', .delete, true, ':', 1, orm.QueryData{},
		where)

	db.query(query)!
}

// last_id is used internally by V's ORM for post-processing `INSERT` queries
fn (mut db Database) last_id() int {
	return 0
}

// DDL (table creation/destroying etc)
fn sqlite_type_from_v(typ int) !string {
	return if typ in orm.nums || typ in orm.num64 || typ in [orm.serial, orm.time_, orm.enum_] {
		'INTEGER'
	} else if typ in orm.float {
		'REAL'
	} else if typ == orm.type_string {
		'TEXT'
	} else {
		error('Unknown type ${typ}')
	}
}

// create is used internally by V's ORM for processing table creation queries (DDL)
fn (mut db Database) create(table orm.Table, fields []orm.TableField) ! {
	mut query := orm.orm_table_gen(.sqlite, table, '', true, 0, fields, sqlite_type_from_v,
		false) or { return err }
	db.query(query)!
}

// drop is used internally by V's ORM for processing table destroying queries (DDL)
fn (mut db Database) drop(table orm.Table) ! {
	query := 'DROP TABLE ${table.name};'
	$if trace_orm ? {
		eprintln('> vsql drop: ${query}')
	}
	db.query(query)!
}

fn test_orm_mut_connection() {
	mut db := Database{}

	sql db {
		create table User
	}!

	first_user := User{
		name: 'first'
	}
	second_user := User{
		name: 'second'
	}

	sql db {
		insert first_user into User
		insert second_user into User
	}!

	sql db {
		drop table User
	}!

	assert db.last_query == 'DROP TABLE User;'
}