blackbeard420/kore
1
0
Fork 0
mirror of https://github.com/jorisvink/kore synced 2025-03-09 12:39:01 -04:00
Code Issues Projects Releases Wiki Activity

add lock.trylock() and make lock.release() sync.

Browse Source
This commit is contained in:
Joris Vink 2019-09-20 10:03:43 +02:00
parent 06fd5ca2f2
commit d99d0b2d77
2 changed files with 33 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
include/kore/python_methods.h
View File

@ -357,10 +357,13 @@ struct pylock {
static PyObject *pylock_aexit(struct pylock *, PyObject *); static PyObject *pylock_aexit(struct pylock *, PyObject *);
static PyObject *pylock_aenter(struct pylock *, PyObject *); static PyObject *pylock_aenter(struct pylock *, PyObject *);
static PyObject *pylock_release(struct pylock *, PyObject *);
static PyObject *pylock_trylock(struct pylock *, PyObject *);
static PyMethodDef pylock_methods[] = { static PyMethodDef pylock_methods[] = {
METHOD("aqcuire", pylock_aenter, METH_NOARGS), METHOD("aqcuire", pylock_aenter, METH_NOARGS),
METHOD("release", pylock_aexit, METH_NOARGS), METHOD("release", pylock_release, METH_NOARGS),
METHOD("trylock", pylock_trylock, METH_NOARGS),
METHOD("__aexit__", pylock_aexit, METH_VARARGS), METHOD("__aexit__", pylock_aexit, METH_VARARGS),
METHOD("__aenter__", pylock_aenter, METH_NOARGS), METHOD("__aenter__", pylock_aenter, METH_NOARGS),
METHOD(NULL, NULL, -1) METHOD(NULL, NULL, -1)

29
src/python.c
View File

@ -2833,6 +2833,35 @@ pylock_dealloc(struct pylock *lock)
PyObject_Del((PyObject *)lock); PyObject_Del((PyObject *)lock);
} }
static PyObject *
pylock_trylock(struct pylock *lock, PyObject *args)
{
if (lock->owner != NULL)
Py_RETURN_FALSE;
lock->owner = coro_running;
Py_RETURN_TRUE;
}
static PyObject *
pylock_release(struct pylock *lock, PyObject *args)
{
if (lock->owner == NULL) {
PyErr_SetString(PyExc_RuntimeError, "no lock owner set");
return (NULL);
}
if (lock->owner->id != coro_running->id) {
PyErr_SetString(PyExc_RuntimeError, "lock not owned by caller");
return (NULL);
}
pylock_do_release(lock);
Py_RETURN_NONE;
}
static PyObject * static PyObject *
pylock_aenter(struct pylock *lock, PyObject *args) pylock_aenter(struct pylock *lock, PyObject *args)
{ {