Commit 713aafbc authored by Shuhei Matsumoto's avatar Shuhei Matsumoto Committed by Tomasz Zawadzki
Browse files

lib/thread: Add spdk_thread_set_cpumask() and enum SPDK_THREAD_OP_RESCHED 



Add an new API spdk_thread_set_cpumask() and enum SPDK_THREAD_OP_RESCHED.

spdk_thread_set_cpumask() can be called only from the current thread
and requires SPDK thread operation supports reschedule operation.

spdk_thread_set_cpumask() updates the cpumask of the current thread to
the specified value, and then invokes framework's reschedule operation
to the thread.

If spdk_thread_set_cpumask() calls multiple times in a single
spdk_thread_poll() context, the last value will be used in the
reschedule operation.

Signed-off-by: default avatarShuhei Matsumoto <shuhei.matsumoto.xt@hitachi.com>
Change-Id: I7808626b10269543c1e2cd86793a504daa4b6389
Reviewed-on: https://review.spdk.io/gerrit/c/spdk/spdk/+/499


Tested-by: default avatarSPDK CI Jenkins <sys_sgci@intel.com>
Reviewed-by: default avatarBen Walker <benjamin.walker@intel.com>
Reviewed-by: default avatarJim Harris <james.r.harris@intel.com>
parent 1368aec8

include/spdk/thread.h

+17 −0
Original line number Diff line number Diff line
@@ -76,6 +76,11 @@ enum spdk_thread_op { 
	 * should frequently call spdk_thread_poll() on the thread provided.

	 */

	SPDK_THREAD_OP_NEW,



	/* Called when SPDK thread needs to be rescheduled. (e.g., when cpumask of the

	 * SPDK thread is updated.

	 */

	SPDK_THREAD_OP_RESCHED,

};



/**
@@ -309,6 +314,18 @@ void *spdk_thread_get_ctx(struct spdk_thread *thread); 
 */

struct spdk_cpuset *spdk_thread_get_cpumask(struct spdk_thread *thread);



/**

 * Set the current thread's cpumask to the specified value. The thread may be

 * rescheduled to one of the CPUs specified in the cpumask.

 *

 * This API requires SPDK thread operation supports SPDK_THREAD_OP_RESCHED.

 *

 * \param cpumask The new cpumask for the thread.

 *

 * \return 0 on success, negated errno otherwise.

 */

int spdk_thread_set_cpumask(struct spdk_cpuset *cpumask);



/**

 * Return the thread object associated with the context handle previously

 * obtained by calling spdk_thread_get_ctx().

lib/thread/thread.c

+29 −0
Original line number Diff line number Diff line
@@ -479,6 +479,35 @@ spdk_thread_get_cpumask(struct spdk_thread *thread) 
	return &thread->cpumask;

}



int

spdk_thread_set_cpumask(struct spdk_cpuset *cpumask)

{

	struct spdk_thread *thread;



	if (!g_thread_op_supported_fn || !g_thread_op_supported_fn(SPDK_THREAD_OP_RESCHED)) {

		SPDK_ERRLOG("Framework does not support reschedule operation.\n");

		assert(false);

		return -ENOTSUP;

	}



	thread = spdk_get_thread();

	if (!thread) {

		SPDK_ERRLOG("Called from non-SPDK thread\n");

		assert(false);

		return -EINVAL;

	}



	spdk_cpuset_copy(&thread->cpumask, cpumask);



	/* Invoke framework's reschedule operation. If this function is called multiple times

	 * in a single spdk_thread_poll() context, the last cpumask will be used in the

	 * reschedule operation.

	 */

	g_thread_op_fn(thread, SPDK_THREAD_OP_RESCHED);



	return 0;

}



struct spdk_thread *

spdk_thread_get_from_ctx(void *ctx)

{