bpo-32436: Make PyContextVar_Get a little bit faster (python#5350)

1st1 · web-flow · commit 226e50049da4 · 2018-01-26T17:24:52.000-05:00
Since context.c is compiled with Py_BUILD_CORE, using a macro
will result in a slightly more optimal code.
diff --git a/Python/context.c b/Python/context.c
@@ -145,7 +145,8 @@ PyContextVar_Get(PyContextVar *var, PyObject *def, PyObject **val)
 {
     assert(PyContextVar_CheckExact(var));
 
-    PyThreadState *ts = PyThreadState_Get();
+    PyThreadState *ts = PyThreadState_GET();
+    assert(ts != NULL);
     if (ts->context == NULL) {
         goto not_found;
     }

Original file line number	Diff line number	Diff line change
`@@ -145,7 +145,8 @@ PyContextVar_Get(PyContextVar var, PyObject def, PyObject **val)`
`145`	`145`	`{`
`146`	`146`	`assert(PyContextVar_CheckExact(var));`
`147`	`147`
`148`		`- PyThreadState *ts = PyThreadState_Get();`
	`148`	`+ PyThreadState *ts = PyThreadState_GET();`
	`149`	`+ assert(ts != NULL);`
`149`	`150`	`if (ts->context == NULL) {`
`150`	`151`	`goto not_found;`
`151`	`152`	`}`