1
0
mirror of https://github.com/vlang/v.git synced 2023-08-10 21:13:21 +03:00

pg: cleanup implementation, make -cstrict pass with both gcc-11 and clang-12 (by tightening the C wrapper declarations)

This commit is contained in:
Delyan Angelov 2022-12-14 13:10:23 +02:00
parent 70bb989537
commit 173b1895d6
No known key found for this signature in database
GPG Key ID: 66886C0F12D595ED
2 changed files with 97 additions and 53 deletions

View File

@ -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, pg_stmt_binder(mut param_types, mut param_vals, mut param_lens, mut param_formats,
where) where)
res := C.PQexecParams(db.conn, query.str, param_vals.len, param_types.data, param_vals.data, res := C.PQexecParams(db.conn, &char(query.str), param_vals.len, param_types.data,
param_lens.data, param_formats.data, 0) 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') 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) { fn pg_stmt_match(mut types []u32, mut vals []&char, mut lens []int, mut formats []int, data orm.Primitive) {
d := data
match data { match data {
bool { bool {
types << u32(Oid.t_bool) types << u32(Oid.t_bool)
vals << &char(&(d as bool)) vals << &char(&data)
lens << int(sizeof(bool)) lens << int(sizeof(bool))
formats << 1 formats << 1
} }
u8 { u8 {
types << u32(Oid.t_char) types << u32(Oid.t_char)
vals << &char(&(d as u8)) vals << &char(&data)
lens << int(sizeof(u8)) lens << int(sizeof(u8))
formats << 1 formats << 1
} }
u16 { u16 {
types << u32(Oid.t_int2) types << u32(Oid.t_int2)
num := conv.htn16(&data) num := conv.htn16(data)
vals << &char(&num) vals << &char(&num)
lens << int(sizeof(u16)) lens << int(sizeof(u16))
formats << 1 formats << 1
} }
u32 { u32 {
types << u32(Oid.t_int4) types << u32(Oid.t_int4)
num := conv.htn32(&data) num := conv.htn32(data)
vals << &char(&num) vals << &char(&num)
lens << int(sizeof(u32)) lens << int(sizeof(u32))
formats << 1 formats << 1
} }
u64 { u64 {
types << u32(Oid.t_int8) types << u32(Oid.t_int8)
num := conv.htn64(&data) num := conv.htn64(data)
vals << &char(&num) vals << &char(&num)
lens << int(sizeof(u64)) lens << int(sizeof(u64))
formats << 1 formats << 1
} }
i8 { i8 {
types << u32(Oid.t_char) types << u32(Oid.t_char)
vals << &char(&(d as i8)) vals << &char(&data)
lens << int(sizeof(i8)) lens << int(sizeof(i8))
formats << 1 formats << 1
} }
i16 { i16 {
types << u32(Oid.t_int2) types << u32(Oid.t_int2)
num := conv.htn16(unsafe { &u16(&data) }) num := conv.htn16(u16(data))
vals << &char(&num) vals << &char(&num)
lens << int(sizeof(i16)) lens << int(sizeof(i16))
formats << 1 formats << 1
} }
int { int {
types << u32(Oid.t_int4) types << u32(Oid.t_int4)
num := conv.htn32(unsafe { &u32(&data) }) num := conv.htn32(u32(data))
vals << &char(&num) vals << &char(&num)
lens << int(sizeof(int)) lens << int(sizeof(int))
formats << 1 formats << 1
} }
i64 { i64 {
types << u32(Oid.t_int8) types << u32(Oid.t_int8)
num := conv.htn64(unsafe { &u64(&data) }) num := conv.htn64(u64(data))
vals << &char(&num) vals << &char(&num)
lens << int(sizeof(i64)) lens << int(sizeof(i64))
formats << 1 formats << 1
} }
f32 { f32 {
types << u32(Oid.t_float4) types << u32(Oid.t_float4)
vals << &char(unsafe { &f32(&(d as f32)) }) vals << &char(&data)
lens << int(sizeof(f32)) lens << int(sizeof(f32))
formats << 1 formats << 1
} }
f64 { f64 {
types << u32(Oid.t_float8) types << u32(Oid.t_float8)
vals << &char(unsafe { &f64(&(d as f64)) }) vals << &char(&data)
lens << int(sizeof(f64)) lens << int(sizeof(f64))
formats << 1 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, // 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 // the server infers a data type for the parameter symbol in the same way
// it would do for an untyped literal string. // it would do for an untyped literal string.
types << &u8(0) types << u32(0)
vals << data.str vals << &char(data.str)
lens << data.len lens << data.len
formats << 0 formats << 0
} }
time.Time { time.Time {
datetime := ((d as time.Time).format_ss() as string) datetime := data.format_ss()
types << &u8(0) types << u32(0)
vals << datetime.str vals << &char(datetime.str)
lens << datetime.len lens << datetime.len
formats << 0 formats << 0
} }

View File

@ -26,9 +26,6 @@ pub mut:
vals []string vals []string
} }
struct C.PGResult {
}
pub struct Config { pub struct Config {
pub: pub:
host string = 'localhost' host string = 'localhost'
@ -38,52 +35,101 @@ pub:
dbname string 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: // Params:
// const Oid *paramTypes // const Oid *paramTypes
// const char *const *paramValues // const char *const *paramValues
// const int *paramLengths // const int *paramLengths
// const int *paramFormats // 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 // connect makes a new connection to the database server using
// the parameters from the `Config` structure, returning // the parameters from the `Config` structure, returning
// a connection error when something goes wrong // a connection error when something goes wrong
pub fn connect(config Config) !DB { pub fn connect(config Config) !DB {
conninfo := 'host=${config.host} port=${config.port} user=${config.user} dbname=${config.dbname} password=${config.password}' 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 { if conn == 0 {
return error('libpq memory allocation error') return error('libpq memory allocation error')
} }
status := C.PQstatus(conn) status := unsafe { ConnStatusType(C.PQstatus(conn)) }
if status != C.CONNECTION_OK { if status != .ok {
// We force the construction of a new string as the // We force the construction of a new string as the
// error message will be freed by the next `PQfinish` // error message will be freed by the next `PQfinish`
// call // call
@ -165,7 +211,7 @@ pub fn (db DB) q_strings(query string) ![]Row {
// for the result, returning an error on failure and a // for the result, returning an error on failure and a
// row set on success // row set on success
pub fn (db DB) exec(query string) ![]Row { 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') 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 { 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() } e := unsafe { C.PQerrorMessage(db.conn).vstring() }
if e != '' { if e != '' {
return error('pg exec error: "${e}"') return error('pg exec error: "${e}"')
@ -191,11 +237,11 @@ pub fn (db DB) exec_param_many(query string, params []string) ![]Row {
unsafe { unsafe {
mut param_vals := []&char{len: params.len} mut param_vals := []&char{len: params.len}
for i in 0 .. 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, res := C.PQexecParams(db.conn, &char(query.str), params.len, 0, param_vals.data,
0) 0, 0, 0)
return db.handle_error_or_result(res, 'exec_param_many') 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 // copy_expert execute COPY commands
// https://www.postgresql.org/docs/9.5/libpq-copy.html // https://www.postgresql.org/docs/9.5/libpq-copy.html
pub fn (db DB) copy_expert(query string, mut file io.ReaderWriter) !int { pub fn (db DB) copy_expert(query string, mut file io.ReaderWriter) !int {
res := C.PQexec(db.conn, query.str) mut res := C.PQexec(db.conn, &char(query.str))
status := C.PQresultStatus(res) status := unsafe { ExecStatusType(C.PQresultStatus(res)) }
defer { defer {
C.PQclear(res) 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}') return error('pg copy error:\n${e}')
} }
if status == C.PGRES_COPY_IN { if status == .copy_in {
mut buf := []u8{len: 4 * 1024} mut buf := []u8{len: 4 * 1024}
for { for {
n := file.read(mut buf) or { n := file.read(mut buf) or {
msg := 'pg copy error: Failed to read from input' msg := 'pg copy error: Failed to read from input'
C.PQputCopyEnd(db.conn, msg.str) C.PQputCopyEnd(db.conn, &char(msg.str))
return err return err
} }
if n <= 0 { 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 { if code != 1 {
return error('pg copy error: Failed to finish copy command, code: ${code}') 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 { for {
address := &u8(0) address := &char(0)
n_bytes := C.PQgetCopyData(db.conn, &address, 0) n_bytes := C.PQgetCopyData(db.conn, &address, 0)
if n_bytes > 0 { if n_bytes > 0 {
mut local_buf := []u8{len: n_bytes} mut local_buf := []u8{len: n_bytes}