diff --git a/vlib/pg/orm.v b/vlib/pg/orm.v index 9c8f3ac353..b866cad194 100644 --- a/vlib/pg/orm.v +++ b/vlib/pg/orm.v @@ -77,8 +77,8 @@ fn pg_stmt_worker(db DB, query string, data orm.QueryData, where orm.QueryData) pg_stmt_binder(mut param_types, mut param_vals, mut param_lens, mut param_formats, where) - res := C.PQexecParams(db.conn, query.str, param_vals.len, param_types.data, param_vals.data, - param_lens.data, param_formats.data, 0) + res := C.PQexecParams(db.conn, &char(query.str), param_vals.len, param_types.data, + param_vals.data, param_lens.data, param_formats.data, 0) // here, the last 0 means require text results, 1 - binary results return db.handle_error_or_result(res, 'orm_stmt_worker') } @@ -89,77 +89,76 @@ fn pg_stmt_binder(mut types []u32, mut vals []&char, mut lens []int, mut formats } fn pg_stmt_match(mut types []u32, mut vals []&char, mut lens []int, mut formats []int, data orm.Primitive) { - d := data match data { bool { types << u32(Oid.t_bool) - vals << &char(&(d as bool)) + vals << &char(&data) lens << int(sizeof(bool)) formats << 1 } u8 { types << u32(Oid.t_char) - vals << &char(&(d as u8)) + vals << &char(&data) lens << int(sizeof(u8)) formats << 1 } u16 { types << u32(Oid.t_int2) - num := conv.htn16(&data) + num := conv.htn16(data) vals << &char(&num) lens << int(sizeof(u16)) formats << 1 } u32 { types << u32(Oid.t_int4) - num := conv.htn32(&data) + num := conv.htn32(data) vals << &char(&num) lens << int(sizeof(u32)) formats << 1 } u64 { types << u32(Oid.t_int8) - num := conv.htn64(&data) + num := conv.htn64(data) vals << &char(&num) lens << int(sizeof(u64)) formats << 1 } i8 { types << u32(Oid.t_char) - vals << &char(&(d as i8)) + vals << &char(&data) lens << int(sizeof(i8)) formats << 1 } i16 { types << u32(Oid.t_int2) - num := conv.htn16(unsafe { &u16(&data) }) + num := conv.htn16(u16(data)) vals << &char(&num) lens << int(sizeof(i16)) formats << 1 } int { types << u32(Oid.t_int4) - num := conv.htn32(unsafe { &u32(&data) }) + num := conv.htn32(u32(data)) vals << &char(&num) lens << int(sizeof(int)) formats << 1 } i64 { types << u32(Oid.t_int8) - num := conv.htn64(unsafe { &u64(&data) }) + num := conv.htn64(u64(data)) vals << &char(&num) lens << int(sizeof(i64)) formats << 1 } f32 { types << u32(Oid.t_float4) - vals << &char(unsafe { &f32(&(d as f32)) }) + vals << &char(&data) lens << int(sizeof(f32)) formats << 1 } f64 { types << u32(Oid.t_float8) - vals << &char(unsafe { &f64(&(d as f64)) }) + vals << &char(&data) lens << int(sizeof(f64)) formats << 1 } @@ -167,15 +166,15 @@ fn pg_stmt_match(mut types []u32, mut vals []&char, mut lens []int, mut formats // If paramTypes is NULL, or any particular element in the array is zero, // the server infers a data type for the parameter symbol in the same way // it would do for an untyped literal string. - types << &u8(0) - vals << data.str + types << u32(0) + vals << &char(data.str) lens << data.len formats << 0 } time.Time { - datetime := ((d as time.Time).format_ss() as string) - types << &u8(0) - vals << datetime.str + datetime := data.format_ss() + types << u32(0) + vals << &char(datetime.str) lens << datetime.len formats << 0 } diff --git a/vlib/pg/pg.v b/vlib/pg/pg.v index 24d49ef246..0ec540dd23 100644 --- a/vlib/pg/pg.v +++ b/vlib/pg/pg.v @@ -26,9 +26,6 @@ pub mut: vals []string } -struct C.PGResult { -} - pub struct Config { pub: host string = 'localhost' @@ -38,52 +35,101 @@ pub: dbname string } -fn C.PQconnectdb(a &u8) &C.PGconn +// -fn C.PQerrorMessage(voidptr) &u8 +struct C.pg_result {} -fn C.PQgetvalue(&C.PGResult, int, int) &u8 +struct C.pg_conn {} -fn C.PQstatus(voidptr) int +[typedef] +pub struct C.PGresult {} -fn C.PQresultStatus(voidptr) int +[typedef] +pub struct C.PGconn {} -fn C.PQntuples(&C.PGResult) int +pub enum ConnStatusType { + ok = C.CONNECTION_OK + bad = C.CONNECTION_BAD + // Non-blocking mode only below here + // The existence of these should never be relied upon - they should only be used for user feedback or similar purposes. + started = C.CONNECTION_STARTED // Waiting for connection to be made. + made = C.CONNECTION_MADE // Connection OK; waiting to send. + awaiting_response = C.CONNECTION_AWAITING_RESPONSE // Waiting for a response from the postmaster. + auth_ok = C.CONNECTION_AUTH_OK // Received authentication; waiting for backend startup. + setenv = C.CONNECTION_SETENV // Negotiating environment. + ssl_startup = C.CONNECTION_SSL_STARTUP // Negotiating SSL. + needed = C.CONNECTION_NEEDED // Internal state: connect() needed + check_writable = C.CONNECTION_CHECK_WRITABLE // Check if we could make a writable connection. + consume = C.CONNECTION_CONSUME // Wait for any pending message and consume them. + gss_startup = C.CONNECTION_GSS_STARTUP // Negotiating GSSAPI. +} -fn C.PQnfields(&C.PGResult) int +[typedef] +pub enum ExecStatusType { + empty_query = C.PGRES_EMPTY_QUERY // empty query string was executed + command_ok = C.PGRES_COMMAND_OK // a query command that doesn't return anything was executed properly by the backend + tuples_ok = C.PGRES_TUPLES_OK // a query command that returns tuples was executed properly by the backend, PGresult contains the result tuples + copy_out = C.PGRES_COPY_OUT // Copy Out data transfer in progress + copy_in = C.PGRES_COPY_IN // Copy In data transfer in progress + bad_response = C.PGRES_BAD_RESPONSE // an unexpected response was recv'd from the backend + nonfatal_error = C.PGRES_NONFATAL_ERROR // notice or warning message + fatal_error = C.PGRES_FATAL_ERROR // query failed + copy_both = C.PGRES_COPY_BOTH // Copy In/Out data transfer in progress + single_tuple = C.PGRES_SINGLE_TUPLE // single tuple from larger resultset +} -fn C.PQexec(voidptr, &u8) &C.PGResult +// + +fn C.PQconnectdb(const_conninfo &char) &C.PGconn + +fn C.PQstatus(const_conn &C.PGconn) int + +fn C.PQerrorMessage(const_conn &C.PGconn) &char + +fn C.PQexec(res &C.PGconn, const_query &char) &C.PGresult + +// + +fn C.PQgetvalue(const_res &C.PGresult, int, int) &char + +fn C.PQresultStatus(const_res &C.PGresult) int + +fn C.PQntuples(const_res &C.PGresult) int + +fn C.PQnfields(const_res &C.PGresult) int // Params: // const Oid *paramTypes // const char *const *paramValues // const int *paramLengths // const int *paramFormats -fn C.PQexecParams(conn voidptr, command &u8, nParams int, paramTypes int, paramValues &u8, paramLengths int, paramFormats int, resultFormat int) &C.PGResult +fn C.PQexecParams(conn &C.PGconn, const_command &char, nParams int, const_paramTypes &int, const_paramValues &char, const_paramLengths &int, const_paramFormats &int, resultFormat int) &C.PGresult -fn C.PQputCopyData(conn voidptr, buffer &u8, nbytes int) int +fn C.PQputCopyData(conn &C.PGconn, const_buffer &char, nbytes int) int -fn C.PQputCopyEnd(voidptr, &u8) int +fn C.PQputCopyEnd(conn &C.PGconn, const_errmsg &char) int -fn C.PQgetCopyData(conn voidptr, buffer &&u8, async int) int +fn C.PQgetCopyData(conn &C.PGconn, buffer &&char, async int) int -fn C.PQclear(&C.PGResult) voidptr +// cleanup -fn C.PQfreemem(voidptr) +fn C.PQclear(res &C.PGresult) -fn C.PQfinish(voidptr) +fn C.PQfreemem(ptr voidptr) + +fn C.PQfinish(conn &C.PGconn) // connect makes a new connection to the database server using // the parameters from the `Config` structure, returning // a connection error when something goes wrong pub fn connect(config Config) !DB { conninfo := 'host=${config.host} port=${config.port} user=${config.user} dbname=${config.dbname} password=${config.password}' - conn := C.PQconnectdb(conninfo.str) + conn := C.PQconnectdb(&char(conninfo.str)) if conn == 0 { return error('libpq memory allocation error') } - status := C.PQstatus(conn) - if status != C.CONNECTION_OK { + status := unsafe { ConnStatusType(C.PQstatus(conn)) } + if status != .ok { // We force the construction of a new string as the // error message will be freed by the next `PQfinish` // call @@ -165,7 +211,7 @@ pub fn (db DB) q_strings(query string) ![]Row { // for the result, returning an error on failure and a // row set on success pub fn (db DB) exec(query string) ![]Row { - res := C.PQexec(db.conn, query.str) + res := C.PQexec(db.conn, &char(query.str)) return db.handle_error_or_result(res, 'exec') } @@ -177,7 +223,7 @@ fn rows_first_or_empty(rows []Row) !Row { } pub fn (db DB) exec_one(query string) !Row { - res := C.PQexec(db.conn, query.str) + res := C.PQexec(db.conn, &char(query.str)) e := unsafe { C.PQerrorMessage(db.conn).vstring() } if e != '' { return error('pg exec error: "${e}"') @@ -191,11 +237,11 @@ pub fn (db DB) exec_param_many(query string, params []string) ![]Row { unsafe { mut param_vals := []&char{len: params.len} for i in 0 .. params.len { - param_vals[i] = params[i].str + param_vals[i] = &char(params[i].str) } - res := C.PQexecParams(db.conn, query.str, params.len, 0, param_vals.data, 0, 0, - 0) + res := C.PQexecParams(db.conn, &char(query.str), params.len, 0, param_vals.data, + 0, 0, 0) return db.handle_error_or_result(res, 'exec_param_many') } } @@ -220,9 +266,8 @@ fn (db DB) handle_error_or_result(res voidptr, elabel string) ![]Row { // copy_expert execute COPY commands // https://www.postgresql.org/docs/9.5/libpq-copy.html pub fn (db DB) copy_expert(query string, mut file io.ReaderWriter) !int { - res := C.PQexec(db.conn, query.str) - status := C.PQresultStatus(res) - + mut res := C.PQexec(db.conn, &char(query.str)) + status := unsafe { ExecStatusType(C.PQresultStatus(res)) } defer { C.PQclear(res) } @@ -232,12 +277,12 @@ pub fn (db DB) copy_expert(query string, mut file io.ReaderWriter) !int { return error('pg copy error:\n${e}') } - if status == C.PGRES_COPY_IN { + if status == .copy_in { mut buf := []u8{len: 4 * 1024} for { n := file.read(mut buf) or { msg := 'pg copy error: Failed to read from input' - C.PQputCopyEnd(db.conn, msg.str) + C.PQputCopyEnd(db.conn, &char(msg.str)) return err } if n <= 0 { @@ -250,14 +295,14 @@ pub fn (db DB) copy_expert(query string, mut file io.ReaderWriter) !int { } } - code := C.PQputCopyEnd(db.conn, 0) + code := C.PQputCopyEnd(db.conn, &char(0)) if code != 1 { return error('pg copy error: Failed to finish copy command, code: ${code}') } - } else if status == C.PGRES_COPY_OUT { + } else if status == .copy_out { for { - address := &u8(0) + address := &char(0) n_bytes := C.PQgetCopyData(db.conn, &address, 0) if n_bytes > 0 { mut local_buf := []u8{len: n_bytes}