vieter-v
/
v
2
0
Fork 0
Code Issues Pull Requests Releases Activity

mysql: add error handling

pull/2963/head
Don Alfons Nisnoni 2019-12-03 16:18:20 +08:00 committed by Alexander Medvednikov
parent 33793a869f
commit f806e0f9e5
1 changed files with 14 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
vlib/mysql/mysql.v
View File

@ -23,31 +23,29 @@ fn C.mysql_init(mysql &C.MYSQL) &C.MYSQL
fn C.mysql_real_connect(mysql &C.MYSQL, host byteptr, user byteptr, passwd byteptr, db byteptr, port u32, unix_socket byteptr, clientflag u64) &C.MYSQL
fn C.mysql_query(mysql &C.MYSQL, q byteptr) int
fn C.mysql_error(mysql &C.MYSQL) byteptr
fn C.mysql_errno(mysql &C.MYSQL) int
fn C.mysql_num_fields(res &C.MYSQL_RES) int
fn C.mysql_store_result(mysql &C.MYSQL) &C.MYSQL_RES
fn C.mysql_fetch_row(res &C.MYSQL_RES) &byteptr
fn C.mysql_free_result(res &C.MYSQL_RES)
fn C.mysql_close(sock &C.MYSQL)
pub fn connect(server, user, passwd, dbname string) DB {
pub fn connect(server, user, passwd, dbname string) ?DB {
conn := C.mysql_init(0)
if isnil(conn) {
eprintln('mysql_init failed')
exit(1)
return error_with_code(get_error_msg(conn), get_errno(conn))
}
conn2 := C.mysql_real_connect(conn, server.str, user.str, passwd.str, dbname.str, 0, 0, 0)
if isnil(conn2) {
eprintln('mysql_real_connect failed')
exit(1)
return error_with_code(get_error_msg(conn), get_errno(conn))
}
return DB {conn: conn2}
}
pub fn (db DB) query(q string) Result {
pub fn (db DB) query(q string) ?Result {
ret := C.mysql_query(db.conn, q.str)
if ret != 0 {
C.fprintf(stderr, '%s\n', mysql_error(db.conn))
exit(1)
return error_with_code(get_error_msg(db.conn), get_errno(db.conn))
}
res := C.mysql_store_result(db.conn)
return Result {result: res}
@ -85,3 +83,11 @@ pub fn (r Result) rows() []Row {
pub fn (r Result) free() {
C.mysql_free_result(r.result)
}
fn get_error_msg(conn &C.MYSQL) string {
return string(C.mysql_error(conn))
}
fn get_errno(conn &C.MYSQL) int {
return C.mysql_errno(conn)
}