python/mysqlclient
1
0
Fork 0
mirror of https://github.com/PyMySQL/mysqlclient.git synced 2025-08-15 11:10:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Discard results without converting them into Python objects. (#601)

Browse Source 
Fixes #560.
This commit is contained in:
Inada Naoki
2023-05-17 02:08:52 +09:00
committed by GitHub
parent 0220f427a9
commit a2e970698f
3 changed files with 126 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

69
src/MySQLdb/_mysql.c
View File

@ -1484,6 +1484,26 @@ _mysql_ResultObject_fetch_row(
return NULL;
}
static const char _mysql_ResultObject_discard__doc__[] =
"discard() -- Discard remaining rows in the resultset.";
static PyObject *
_mysql_ResultObject_discard(
_mysql_ResultObject *self,
PyObject *noargs)
{
check_result_connection(self);
MYSQL_ROW row;
while (NULL != (row = mysql_fetch_row(self->result))) {
// do nothing
}
if (mysql_errno(self->conn)) {
return _mysql_Exception(self->conn);
}
Py_RETURN_NONE;
}
static char _mysql_ConnectionObject_change_user__doc__[] =
"Changes the user and causes the database specified by db to\n\
become the default (current) database on the connection\n\
@ -2081,6 +2101,43 @@ _mysql_ConnectionObject_use_result(
return result;
}
static const char _mysql_ConnectionObject_discard_result__doc__[] =
"Discard current result set.\n\n"
"This function can be called instead of use_result() or store_result(). Non-standard.";
static PyObject *
_mysql_ConnectionObject_discard_result(
_mysql_ConnectionObject *self,
PyObject *noargs)
{
check_connection(self);
MYSQL *conn = &(self->connection);
Py_BEGIN_ALLOW_THREADS;
MYSQL_RES *res = mysql_use_result(conn);
if (res == NULL) {
Py_BLOCK_THREADS;
if (mysql_errno(conn) != 0) {
// fprintf(stderr, "mysql_use_result failed: %s\n", mysql_error(conn));
return _mysql_Exception(self);
}
Py_RETURN_NONE;
}
MYSQL_ROW row;
while (NULL != (row = mysql_fetch_row(res))) {
// do nothing.
}
mysql_free_result(res);
Py_END_ALLOW_THREADS;
if (mysql_errno(conn)) {
// fprintf(stderr, "mysql_free_result failed: %s\n", mysql_error(conn));
return _mysql_Exception(self);
}
Py_RETURN_NONE;
}
static void
_mysql_ConnectionObject_dealloc(
_mysql_ConnectionObject *self)
@ -2376,6 +2433,12 @@ static PyMethodDef _mysql_ConnectionObject_methods[] = {
METH_NOARGS,
_mysql_ConnectionObject_use_result__doc__
},
{
"discard_result",
(PyCFunction)_mysql_ConnectionObject_discard_result,
METH_NOARGS,
_mysql_ConnectionObject_discard_result__doc__
},
{NULL, NULL} /* sentinel */
};
@ -2437,6 +2500,12 @@ static PyMethodDef _mysql_ResultObject_methods[] = {
METH_VARARGS | METH_KEYWORDS,
_mysql_ResultObject_fetch_row__doc__
},
{
"discard",
(PyCFunction)_mysql_ResultObject_discard,
METH_NOARGS,
_mysql_ResultObject_discard__doc__
},
{
"field_flags",
(PyCFunction)_mysql_ResultObject_field_flags,