Thanks to visit codestin.com
Credit goes to github.com

Skip to content

gh-76785: Simplify Channels XID Types #116318

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ericsnowcurrently merged 6 commits into from
Mar 4, 2024
Merged

gh-76785: Simplify Channels XID Types #116318

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
6cb18f4
add_new_type() -> add_channelid_type()
ericsnowcurrently Feb 15, 2024
f5d0db8
Add clear_xid_types().
ericsnowcurrently Feb 15, 2024
a3ef45a
Do not use the XID type registry in set_channelend_types().
ericsnowcurrently Feb 15, 2024
db9d92c
Drop the XID type registry.
ericsnowcurrently Feb 15, 2024
1acc34f
Merge branch 'main' into simplify-channels-xid-types
ericsnowcurrently Mar 4, 2024
970d88b
Clear the right class.
ericsnowcurrently Mar 4, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
add_new_type() -> add_channelid_type()
  • Loading branch information
@ericsnowcurrently
ericsnowcurrently committed Mar 4, 2024
commit 6cb18f40e406ce32e847accd0a743d0ced070504
51 changes: 23 additions & 28 deletions Modules/_xxinterpchannelsmodule.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -223,28 +223,6 @@ add_new_exception(PyObject *mod, const char *name, PyObject *base)
#define ADD_NEW_EXCEPTION(MOD, NAME, BASE) \
add_new_exception(MOD, MODULE_NAME_STR "." Py_STRINGIFY(NAME), BASE)

static PyTypeObject *
add_new_type(PyObject *mod, PyType_Spec *spec, crossinterpdatafunc shared,
struct xid_class_registry *classes)
{
PyTypeObject *cls = (PyTypeObject *)PyType_FromModuleAndSpec(
mod, spec, NULL);
if (cls == NULL) {
return NULL;
}
if (PyModule_AddType(mod, cls) < 0) {
Py_DECREF(cls);
return NULL;
}
if (shared != NULL) {
if (register_xid_class(cls, shared, classes)) {
Py_DECREF(cls);
return NULL;
}
}
return cls;
}

static int
wait_for_lock(PyThread_type_lock mutex, PY_TIMEOUT_T timeout)
{
Expand Down Expand Up @@ -2614,6 +2592,25 @@ static PyType_Spec channelid_typespec = {
.slots = channelid_typeslots,
};

static PyTypeObject *
add_channelid_type(PyObject *mod)
{
PyTypeObject *cls = (PyTypeObject *)PyType_FromModuleAndSpec(
mod, &channelid_typespec, NULL);
if (cls == NULL) {
return NULL;
}
if (PyModule_AddType(mod, cls) < 0) {
Py_DECREF(cls);
return NULL;
}
if (ensure_xid_class(cls, _channelid_shared) < 0) {
Py_DECREF(cls);
return NULL;
}
return cls;
}


/* SendChannel and RecvChannel classes */

Expand Down Expand Up @@ -3294,13 +3291,11 @@ module_exec(PyObject *mod)
if (_globals_init() != 0) {
return -1;
}
struct xid_class_registry *xid_classes = NULL;

module_state *state = get_module_state(mod);
if (state == NULL) {
goto error;
}
xid_classes = &state->xid_classes;

/* Add exception types */
if (exceptions_init(mod) != 0) {
Expand All @@ -3319,8 +3314,7 @@ module_exec(PyObject *mod)
}

// ChannelID
state->ChannelIDType = add_new_type(
mod, &channelid_typespec, _channelid_shared, xid_classes);
state->ChannelIDType = add_channelid_type(mod);
if (state->ChannelIDType == NULL) {
goto error;
}
Expand All @@ -3332,8 +3326,9 @@ module_exec(PyObject *mod)
return 0;

error:
if (xid_classes != NULL) {
clear_xid_class_registry(xid_classes);
if (state->ChannelInfoType != NULL) {
(void)_PyCrossInterpreterData_UnregisterClass(state->ChannelInfoType);
Py_CLEAR(state->ChannelInfoType);
}
_globals_fini();
return -1;
Expand Down