hcq/openGauss-connector-python-psycopg2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加注释和修改函数名称

Browse Source 
修改查询oid的值的sql语句
This commit is contained in:
zcl
2024-03-27 16:03:47 +08:00
parent 5235f6399c
commit f3a182e1bb
2 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
psycopg/connection_type.c
View File

@ -1332,7 +1332,7 @@ register_type_uint(connectionObject *self, PyThreadState **tstate)
_typecast_INTEGER_types = (long int*)malloc((size+1)*sizeof(long int));
for (int i=0; i< size; i++) {
unsigned int uint_val = pq_get_custom_type_oid(self, uint_arr[i], tstate);
unsigned int uint_val = pq_get_pg_catalog_custom_type_oid(self, uint_arr[i], tstate);
_typecast_INTEGER_types[i] = (long int)uint_val;
}
typecastObject_initlist _typecast_builtins[] = {

9
psycopg/pqpath.c
View File

@ -608,15 +608,16 @@ cleanup:
}
/* Get the oid of uint type created by the dolphin plugin, which is only in the pg_catalog by default */
unsigned int
pq_get_custom_type_oid(connectionObject *conn, const char *param, PyThreadState **tstate)
pq_get_pg_catalog_custom_type_oid(connectionObject *conn, const char *param, PyThreadState **tstate)
{
char query[256];
int size;
unsigned int rv = 0;
size = PyOS_snprintf(query, sizeof(query), "select oid from pg_type where typname= %s", param);
size = PyOS_snprintf(query, sizeof(query), "select oid from pg_type where typnamespace = 11 and typname= %s", param);
if (size < 0 || (size_t)size >= sizeof(query)) {
conn_set_error(conn, "query too large");
goto cleanup;
@ -631,7 +632,7 @@ pq_get_custom_type_oid(connectionObject *conn, const char *param, PyThreadState
}
if (!conn->pgres) {
Dprintf("pq_get_custom_type_oid: PQexec returned NULL");
Dprintf("pq_get_pg_catalog_custom_type_oid: PQexec returned NULL");
PyEval_RestoreThread(*tstate);
if (!PyErr_Occurred()) {
conn_set_error(conn, PQerrorMessage(conn->pgconn));
@ -640,7 +641,7 @@ pq_get_custom_type_oid(connectionObject *conn, const char *param, PyThreadState
goto cleanup;
}
if (PQresultStatus(conn->pgres) != PGRES_TUPLES_OK) {
Dprintf("pq_get_custom_type_oid: result was not TUPLES_OK (%s)",
Dprintf("pq_get_pg_catalog_custom_type_oid: result was not TUPLES_OK (%s)",
PQresStatus(PQresultStatus(conn->pgres)));
goto cleanup;
}