Each thread can use a thread local variable to keep the name of a job
which is currently running in the job.
The virThreadJobSetWorker API is supposed to be called once by any
thread which is used as a worker, i.e., it is waiting in a pool, woken
up to do a job, and returned back to the pool.
The virThreadJobSet/virThreadJobClear APIs are to be called at the
beginning/end of each job.
Signed-off-by: Jiri Denemark <jdenemar(a)redhat.com>
---
include/libvirt/virterror.h | 1 +
po/POTFILES.in | 1 +
src/Makefile.am | 2 +
src/libvirt_private.syms | 7 +++
src/util/virerror.c | 1 +
src/util/virthreadjob.c | 126 ++++++++++++++++++++++++++++++++++++++++++++
src/util/virthreadjob.h | 33 ++++++++++++
7 files changed, 171 insertions(+)
create mode 100644 src/util/virthreadjob.c
create mode 100644 src/util/virthreadjob.h
diff --git a/include/libvirt/virterror.h b/include/libvirt/virterror.h
index 5dc99dc..9c5b069 100644
--- a/include/libvirt/virterror.h
+++ b/include/libvirt/virterror.h
@@ -125,6 +125,7 @@ typedef enum {
VIR_FROM_FIREWALL = 59, /* Error from firewall */
VIR_FROM_POLKIT = 60, /* Error from polkit code */
+ VIR_FROM_THREAD = 61, /* Error from thread utils */
# ifdef VIR_ENUM_SENTINELS
VIR_ERR_DOMAIN_LAST
diff --git a/po/POTFILES.in b/po/POTFILES.in
index ab6ae3b..dd06ab3 100644
--- a/po/POTFILES.in
+++ b/po/POTFILES.in
@@ -220,6 +220,7 @@ src/util/virstorageencryption.c
src/util/virstoragefile.c
src/util/virstring.c
src/util/virsysinfo.c
+src/util/virthreadjob.c
src/util/virtime.c
src/util/virtpm.c
src/util/virtypedparam.c
diff --git a/src/Makefile.am b/src/Makefile.am
index 00b47e7..ec62ab4 100644
--- a/src/Makefile.am
+++ b/src/Makefile.am
@@ -160,6 +160,7 @@ UTIL_SOURCES = \
util/virsysinfo.c util/virsysinfo.h \
util/virsystemd.c util/virsystemd.h \
util/virthread.c util/virthread.h \
+ util/virthreadjob.c utils/virthreadjob.h \
util/virthreadpool.c util/virthreadpool.h \
util/virtime.h util/virtime.c \
util/virtpm.h util/virtpm.c \
@@ -2147,6 +2148,7 @@ libvirt_setuid_rpc_client_la_SOURCES = \
util/virstring.c \
util/virtime.c \
util/virthread.c \
+ util/virthreadjob.c \
util/virtypedparam.c \
util/viruri.c \
util/virutil.c \
diff --git a/src/libvirt_private.syms b/src/libvirt_private.syms
index 8141eba..f5756a7 100644
--- a/src/libvirt_private.syms
+++ b/src/libvirt_private.syms
@@ -2183,6 +2183,13 @@ virThreadSelf;
virThreadSelfID;
+# util/virthreadjob.h
+virThreadJobClear;
+virThreadJobGet;
+virThreadJobSet;
+virThreadJobSetWorker;
+
+
# util/virthreadpool.h
virThreadPoolFree;
virThreadPoolGetMaxWorkers;
diff --git a/src/util/virerror.c b/src/util/virerror.c
index 9635c82..c4e84e7 100644
--- a/src/util/virerror.c
+++ b/src/util/virerror.c
@@ -132,6 +132,7 @@ VIR_ENUM_IMPL(virErrorDomain, VIR_ERR_DOMAIN_LAST,
"Firewall",
"Polkit", /* 60 */
+ "Thread jobs",
)
diff --git a/src/util/virthreadjob.c b/src/util/virthreadjob.c
new file mode 100644
index 0000000..2f483f9
--- /dev/null
+++ b/src/util/virthreadjob.c
@@ -0,0 +1,126 @@
+/*
+ * virthreadjob.c: code for tracking job associated with current thread
+ *
+ * Copyright (C) 2013-2015 Red Hat, Inc.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library. If not, see
+ * <
http://www.gnu.org/licenses/>.
+ *
+ * Author: Jiri Denemark <jdenemar(a)redhat.com>
+ */
+
+#include <config.h>
+
+#include "internal.h"
+#include "virerror.h"
+#include "virlog.h"
+#include "virthread.h"
+#include "virthreadjob.h"
+
+#define VIR_FROM_THIS VIR_FROM_THREAD
+VIR_LOG_INIT("util.threadjob");
+
+virThreadLocal virThreadJobWorker;
+virThreadLocal virThreadJobName;
+
+
+static int
+virThreadJobOnceInit(void)
+{
+ if (virThreadLocalInit(&virThreadJobWorker, NULL) < 0 ||
+ virThreadLocalInit(&virThreadJobName, NULL) < 0)
+ return -1;
+ return 0;
+}
+
+VIR_ONCE_GLOBAL_INIT(virThreadJob)
+
+
+const char *
+virThreadJobGet(void)
+{
+ const char *job;
+
+ if (virThreadJobInitialize() < 0)
+ return NULL;
+
+ job = virThreadLocalGet(&virThreadJobName);
+ if (!job)
+ job = virThreadLocalGet(&virThreadJobWorker);
+
+ return job;
+}
+
+
+void
+virThreadJobSetWorker(const char *worker)
+{
+ if (!worker || virThreadJobInitialize() < 0)
+ return;
+
+ if (virThreadLocalSet(&virThreadJobWorker, (void *) worker) < 0)
+ virReportSystemError(errno,
+ _("cannot set worker name to %s"),
+ worker);
+
+ VIR_DEBUG("Thread %llu is running worker %s", virThreadSelfID(), worker);
+}
+
+
+void
+virThreadJobSet(const char *caller)
+{
+ const char *worker;
+
+ if (!caller || virThreadJobInitialize() < 0)
+ return;
+
+ if (virThreadLocalSet(&virThreadJobName, (void *) caller) < 0)
+ virReportSystemError(errno,
+ _("cannot set current job to %s"),
+ caller);
+
+ if ((worker = virThreadLocalGet(&virThreadJobWorker))) {
+ VIR_DEBUG("Thread %llu (%s) is now running job %s",
+ virThreadSelfID(), worker, caller);
+ } else {
+ VIR_DEBUG("Thread %llu is now running job %s",
+ virThreadSelfID(), caller);
+ }
+}
+
+
+void
+virThreadJobClear(int rv)
+{
+ const char *old;
+ const char *worker;
+
+ if (virThreadJobInitialize() < 0)
+ return;
+
+ if (!(old = virThreadLocalGet(&virThreadJobName)))
+ return;
+
+ if (virThreadLocalSet(&virThreadJobName, NULL) < 0)
+ virReportSystemError(errno, "%s", _("cannot reset current
job"));
+
+ if ((worker = virThreadLocalGet(&virThreadJobWorker))) {
+ VIR_DEBUG("Thread %llu (%s) finished job %s with ret=%d",
+ virThreadSelfID(), worker, old, rv);
+ } else {
+ VIR_DEBUG("Thread %llu finished job %s with ret=%d",
+ virThreadSelfID(), old, rv);
+ }
+}
diff --git a/src/util/virthreadjob.h b/src/util/virthreadjob.h
new file mode 100644
index 0000000..659f91d
--- /dev/null
+++ b/src/util/virthreadjob.h
@@ -0,0 +1,33 @@
+/*
+ * virthreadjob.h: APIs for tracking job associated with current thread
+ *
+ * Copyright (C) 2013-2015 Red Hat, Inc.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library. If not, see
+ * <
http://www.gnu.org/licenses/>.
+ *
+ * Author: Jiri Denemark <jdenemar(a)redhat.com>
+ */
+
+#ifndef __VIR_THREAD_JOB_H__
+# define __VIR_THREAD_JOB_H__
+
+
+const char *virThreadJobGet(void);
+
+void virThreadJobSetWorker(const char *caller);
+void virThreadJobSet(const char *caller);
+void virThreadJobClear(int rv);
+
+#endif
--
2.3.3