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:
parent
70bb989537
commit
173b1895d6
@ -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
|
||||||
}
|
}
|
||||||
|
113
vlib/pg/pg.v
113
vlib/pg/pg.v
@ -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}
|
||||||
|
Loading…
Reference in New Issue
Block a user