v/vlib/mysql/mysql.v

170 lines
3.9 KiB
V
Raw Normal View History

2019-08-18 16:24:43 +02:00
module mysql
#flag -lmysqlclient
#flag linux -I/usr/include/mysql
2019-08-18 16:24:43 +02:00
#include <mysql.h>
pub struct Connection {
host string
port u32
username string
password string
dbname string
flag int
mut:
conn &MYSQL
2019-08-18 16:24:43 +02:00
}
pub fn new_connection(host, username, password, dbname string) ?Connection {
instance := mysql_init(0)
if isnil(instance) {
return error_with_code(get_error_msg(instance), get_errno(instance))
}
return Connection{ host, 0, username, password, dbname, 0, instance }
2019-08-18 16:24:43 +02:00
}
pub fn (conn mut Connection) connect() ?bool {
mut instance := mysql_init(0)
if !isnil(conn.conn) {
instance = conn.conn
}
if isnil(instance) {
return error_with_code(get_error_msg(instance), get_errno(instance))
}
conn.conn = C.mysql_real_connect(
instance,
conn.host.str,
conn.username.str,
conn.password.str,
conn.dbname.str,
conn.port,
0,
conn.flag
)
if isnil(conn.conn) {
return error_with_code(get_error_msg(instance), get_errno(instance))
}
return true
2019-08-18 16:24:43 +02:00
}
pub fn (conn Connection) query(q string) ?Result {
if mysql_query(conn.conn, q.str) != 0 {
return error_with_code(get_error_msg(conn.conn), get_errno(conn.conn))
}
res := mysql_store_result(conn.conn)
return Result{res}
}
pub fn (conn Connection) select_db(dbname string) ?bool {
if mysql_select_db(conn.conn, dbname.str) != 0 {
return error_with_code(get_error_msg(conn.conn), get_errno(conn.conn))
}
return true
}
2019-08-18 16:24:43 +02:00
pub fn (conn Connection) change_user(username, password, dbname string) ?bool {
mut ret := true
if (dbname != '') {
ret = mysql_change_user(conn.conn, username.str, password.str, dbname.str)
} else {
ret = mysql_change_user(conn.conn, username.str, password.str, 0)
2019-08-18 16:24:43 +02:00
}
if !ret {
return error_with_code(get_error_msg(conn.conn), get_errno(conn.conn))
2019-08-18 16:24:43 +02:00
}
return ret
2019-08-18 16:24:43 +02:00
}
pub fn (conn Connection) affected_rows() u64 {
return mysql_affected_rows(conn.conn)
}
pub fn (conn Connection) autocommit(mode bool) {
mysql_autocommit(conn.conn, mode)
2019-08-18 16:24:43 +02:00
}
pub fn (conn Connection) escape_string(s string) string {
2019-12-13 17:12:42 +01:00
len := strlen(s.str)
to := malloc(2 * len + 1)
quote := byte(39) // single quote
mysql_real_escape_string_quote(conn.conn, to, s.str, len, quote)
2019-12-13 17:12:42 +01:00
return string(to)
}
pub fn (conn Connection) set_option(option_type int, val voidptr) {
mysql_options(conn.conn, option_type, val)
}
pub fn (conn Connection) get_option(option_type int) ?voidptr {
ret := voidptr(0)
if mysql_get_option(conn.conn, option_type, &ret) != 0 {
return error_with_code(get_error_msg(conn.conn), get_errno(conn.conn))
}
return ret
}
pub fn (conn Connection) refresh(options u32) ?bool {
if mysql_refresh(conn.conn, options) != 0 {
return error_with_code(get_error_msg(conn.conn), get_errno(conn.conn))
2020-01-11 13:25:59 +01:00
}
return true
}
pub fn (conn Connection) reset_connection() ?bool {
if mysql_reset_connection(conn.conn) != 0 {
return error_with_code(get_error_msg(conn.conn), get_errno(conn.conn))
}
return true
2019-08-18 16:24:43 +02:00
}
pub fn (conn Connection) ping() ?bool {
if mysql_ping(conn.conn) != 0 {
return error_with_code(get_error_msg(conn.conn), get_errno(conn.conn))
}
return true
2019-08-18 16:24:43 +02:00
}
pub fn (conn Connection) close() {
mysql_close(conn.conn)
2019-08-18 16:24:43 +02:00
}
/* MYSQL INFO & VERSION */
pub fn (conn Connection) info() string {
return string(mysql_info(conn.conn))
}
pub fn (conn Connection) get_host_info() string {
return string(mysql_get_host_info(conn.conn))
}
pub fn (conn Connection) get_server_info() string {
return string(mysql_get_server_info(conn.conn))
}
pub fn (conn Connection) get_server_version() u64 {
return mysql_get_server_version(conn.conn)
2019-08-18 16:24:43 +02:00
}
pub fn get_client_version() u64 {
return mysql_get_client_version()
2019-08-18 16:24:43 +02:00
}
2019-12-03 09:18:20 +01:00
pub fn get_client_info() string {
return string(mysql_get_client_info())
}
/* MYSQL DEBUG */
pub fn (conn Connection) dump_debug_info() ?bool {
if mysql_dump_debug_info(conn.conn) != 0 {
return error_with_code(get_error_msg(conn.conn), get_errno(conn.conn))
}
return true
2019-12-03 09:18:20 +01:00
}
pub fn debug(debug string) {
mysql_debug(debug.str)
2019-12-03 09:18:20 +01:00
}