/*
 *  COPYRIHT (C) 2014 - 2020, piy_xrh
 * 
 *	  Stpool is a portable and efficient tasks pool library, it can work on diferent 
 * platforms such as Windows, linux, unix and ARM.  
 *
 *    If you have any troubles or questions on usin the library, contact me.
 *
 * 	  (Email: piy_xrh@163.com  QQ: 1169732280)
 */

#include <stdlib.h>
#include <string.h>

#include "msglog.h"
#include "timer.h"
#include "cpool_factory.h"
#include "stpool_internal.h"
#include "stpool_group.h"

EXPORT void 
stpool_task_set_gid(struct sttask *ptask, int gid)
{
	assert (TASK_CAST_DOWN(ptask)->gid == gid ||
	        !TASK_CAST_DOWN(ptask)->f_stat);
	
	if (TASK_CAST_DOWN(ptask)->pool && 
		(TASK_CAST_DOWN(ptask)->f_stat || TASK_CAST_DOWN(ptask)->ref)) {
		ctask_t *ptask0 = TASK_CAST_DOWN(ptask);
		
		if (Invokable(group_create, ptask0->pool, pmex)) { 
			/**
			 * Try synchronize the env
			 */
			if (Invokable(task_wsync, ptask0->pool, tskm))
				Invoke(task_wsync, ptask0->pool, tskm, ptask0);
			
			if (ptask0->f_stat || ptask0->ref)
				MSG_log(M_POOL, LOG_WARN,
						"It is unsafe to chane the task's gid since it is not free now or\n"
						"its current reference is not zero. tsk(%s/%p) ref(%d) vmflas(%p) stat(%p).\n",
						ptask->task_name, ptask, ptask0->ref, ptask0->f_vmflags, ptask0->f_stat);
		}
	}
	TASK_CAST_DOWN(ptask)->gid = gid;
}

EXPORT int  
stpool_task_gid(struct sttask *ptask)
{
	return TASK_CAST_DOWN(ptask)->gid;
}

EXPORT const char *
stpool_task_name2(struct sttask *ptask, char *name_buffer, int len)
{
	stpool_t *p = TASK_CAST_DOWN(ptask)->pool;

	if (!p)
		return NULL;

	return stpool_group_name2(p, TASK_CAST_DOWN(ptask)->gid, name_buffer, len);
}

EXPORT int  
stpool_task_throttle_wait(struct sttask *ptask, long ms)
{
	stpool_t *pool = TASK_CAST_DOWN(ptask)->pool;

	if (!pool) {
		MSG_log(M_POOL, LOG_WARN,
				"tsk(%s/%p): Firstly, you should call @stpool_task_set_pool to specify its destination\n",
				ptask->task_name, ptask);
		return POOL_TASK_ERR_DESTINATION;
	}

	return stpool_group_throttle_wait(pool, TASK_CAST_DOWN(ptask)->gid, ms);
}

EXPORT int  
stpool_task_pgthrottle_wait(struct sttask *ptask, long ms)
{
	int e;
	uint64_t us_now = 0;
	long us_elapsed;
	stpool_t *pool = TASK_CAST_DOWN(ptask)->pool;
	
	if (!pool) {
		MSG_log(M_POOL, LOG_WARN,
				"tsk(%s/%p): Firstly, you should call @stpool_task_set_pool to specify its destination\n",
				ptask->task_name, ptask);
		return POOL_TASK_ERR_DESTINATION;
	}
	
	if (ms > 0)
		us_now = us_startr();
	
	if (!(e = stpool_throttle_wait(pool, ms))) {
		if (ms > 0) {
			us_elapsed = us_endr(us_now);
			
			if (us_elapsed >= ms * 1000)
				ms = 0;
			else
				ms -= us_elapsed / 1000;
		}
		e = stpool_group_throttle_wait(pool, TASK_CAST_DOWN(ptask)->gid, ms);
	}
	
	return e;
}

EXPORT int  
stpool_group_create(stpool_t *pool, const char *name, struct gscheduler_attr *attr, int pri_q_num, int suspend)
{
	int group_gid = -1;

	if (Invokable(group_create, pool, pmex)) {
		group_gid = Invoke(group_create, pool, pmex, name, pri_q_num, suspend);

		if (-1 != group_gid && attr)
			stpool_group_setattr(pool, group_gid, attr);
	}

	return group_gid;
}

EXPORT int  
stpool_group_gid(stpool_t *pool, const char *name)
{
	if (!strcmp(name, stpool_desc(pool)))
		return 0;

	if (!Invokable(group_id, pool, pmex)) 
		return -1;
	
	return Invoke(group_id, pool, pmex, name);
}

EXPORT const char *
stpool_group_name2(stpool_t *pool, int gid, char *name_buffer, int len)
{
	if (!gid && !Invokable(group_create, pool, pmex)) {
		const char *desc = stpool_desc(pool);

		if (name_buffer) {
			strncat(name_buffer, desc, len);
			return name_buffer;
		}
		return desc;
	}
	
	TRY_Invoke_return_res(NULL, group_desc, pool, pmex, gid, name_buffer, len);
}

EXPORT struct sttask_group_stat *
stpool_group_stat(stpool_t *pool, int gid, struct sttask_group_stat *stat)
{
	struct ctask_group_stat gstat;

	if (!Invokable(group_create, pool, pmex)) {
		struct pool_stat pstat;

		if (gid)
			return NULL;
		stpool_stat(pool, &pstat);
		
		stat->gid = 0;
		stat->desc = (char *)pstat.desc;
		stat->desc_length = strlen(stat->desc);
		stat->attr.limit_paralle_tasks = pstat.maxthreads;
		stat->attr.receive_benifits = 1;
		stat->created = pstat.created;
		stat->waiters = pstat.waiters;
		stat->suspended = pstat.suspended;
		stat->throttle_enabled = pstat.throttle_enabled;
		stat->priq_num = pstat.priq_num;
		stat->npendings = pstat.curtasks_pending;
		stat->nrunnings = pstat.curthreads_active;
		stat->ndispatchings = pstat.curtasks_removing;
		
		return stat;
	}
	
	Invoke(group_stat, pool, pmex, gid, &gstat);

	assert (sizeof(*stat) == sizeof(gstat));
	memcpy(stat, &gstat, sizeof(*stat));

	return stat;
}

EXPORT int  
stpool_group_stat_all(stpool_t *pool, struct sttask_group_stat **stat)
{
	int n = 0;
	struct ctask_group_stat *gstat;

	if (!Invokable(group_create, pool, pmex)) {
		*stat = malloc(sizeof(struct sttask_group_stat));
		
		if (*stat) {
			stpool_group_stat(pool, 0, *stat);
			n = 1;
		}

	} else {
		n = Invoke(group_stat_all, pool, pmex, &gstat);
		*stat = (struct sttask_group_stat *)gstat;
	}
	
	return n;
}

EXPORT void 
stpool_group_setattr(stpool_t *pool, int gid, struct gscheduler_attr *attr)
{
	struct scheduler_attr attr0;
	
	if (!gid && !Invokable(group_create, pool, pmex)) {
		if (attr->limit_paralle_tasks <= 0)
			attr->limit_paralle_tasks = 1;

		stpool_adjust_abs(pool, attr->limit_paralle_tasks, 0);
		return;
	}

	if (!Invokable(group_setattr, pool, pmex))
		return;
	
	MSG_log(M_POOL, LOG_INFO,
		   "{\"%s\"/%p} Confiurin the group(%d)'s schedulin attributes(%d) ...\n",
		   pool->desc, pool, gid, attr->limit_paralle_tasks);
		
	memcpy(&attr0, attr, sizeof(*attr));
	Invoke(group_setattr, pool, pmex, gid, &attr0);
}

EXPORT struct gscheduler_attr *
stpool_group_getattr(stpool_t *pool, int gid, struct gscheduler_attr *attr)
{
	struct scheduler_attr attr0;
	
	if (!gid && !Invokable(group_create, pool, pmex)) {
		struct pool_stat pstat;

		stpool_stat(pool, &pstat);
		attr->limit_paralle_tasks = pstat.maxthreads;
		attr->receive_benifits = 1;
		return attr;
	}

	if (!Invokable(group_getattr, pool, pmex) || Invoke(group_getattr, pool, pmex, gid, &attr0)) 
		return NULL;

	memcpy(attr, &attr0, sizeof(*attr));
	return attr;
}

EXPORT int
stpool_group_suspend(stpool_t *pool, int gid, long ms)
{
	int e;

	MSG_log(M_POOL, LOG_INFO,
		"{\"%s\"/%p} suspend group(%d) ...(%ld ms)\n",
		pool->desc, pool, gid, ms);
	
	if (!gid && !Invokable(group_create, pool, pmex))
		e = stpool_suspend(pool, ms);
	else
		TRY_Invoke_err(e, group_suspend, pool, pmex, gid, ms);
	
	return e;
}

EXPORT int
stpool_group_suspend_all(stpool_t *pool, long ms)
{
	int e;

	MSG_log(M_POOL, LOG_INFO,
		"{\"%s\"/%p} suspend all groups ...(%d ms)\n",
		pool->desc, pool, ms);
	
	if (!Invokable(group_create, pool, pmex))
		return stpool_suspend(pool, ms);

	TRY_Invoke_err(e, group_suspend_all, pool, pmex, ms);
	
	return e;
}

EXPORT void 
stpool_group_resume(stpool_t *pool, int gid)
{
	MSG_log(M_POOL, LOG_INFO,
		"{\"%s\"/%p} resume group(%d) ...\n",
		pool->desc, pool, gid);
	
	if (!gid && !Invokable(group_create, pool, pmex))
		stpool_resume(pool);
	else 
		TRY_Invoke(group_resume, pool, pmex, gid);
}

EXPORT void 
stpool_group_resume_all(stpool_t *pool)
{
	MSG_log(M_POOL, LOG_INFO,
		"{\"%s\"/%p} resume all groups ...\n",
		pool->desc, pool);
	
	if (!Invokable(group_create, pool, pmex))
		stpool_resume(pool);
	else 
		TRY_Invoke(group_resume_all, pool, pmex);
}

EXPORT int
stpool_group_set_overload_attr(stpool_t *pool, int gid, struct oaattr *attr)
{
	int res = 0;
	assert (sizeof(struct oaattr) == sizeof(struct cpool_oaattr));

	if (!gid && !Invokable(group_create, pool, pmex)) {	
		if (!Invokable(set_oaattr, pool, pm))
			res = POOL_ERR_NSUPPORT;
		else
			Invoke(set_oaattr, pool, pm, (struct cpool_oaattr *)attr);
	} else
		TRY_Invoke_err(res, group_set_oaattr, pool, pmex, gid, (struct cpool_oaattr *)attr);
	
	return res;
}

EXPORT struct oaattr *
stpool_group_get_overload_attr(stpool_t *pool, int gid, struct oaattr *attr)
{
	int res = 0;
	assert (sizeof(struct oaattr) == sizeof(struct cpool_oaattr));
	
	if (!gid && !Invokable(group_create, pool, pmex)) {
		if (!Invokable(get_oaattr, pool, pm))
			res = POOL_ERR_NSUPPORT;
		else
			Invoke(get_oaattr, pool, pm, (struct cpool_oaattr *)attr);
	} else
		TRY_Invoke_err(res, group_get_oaattr, pool, pmex, gid, (struct cpool_oaattr *)attr);
	
	return res ? NULL : attr;
}

EXPORT int  
stpool_group_add_routine(stpool_t *pool, int gid, const char *name,
					void (*task_run)(struct sttask *ptask),
					void (*task_err_handler)(struct sttask *ptask, long reasons),
					void *task_ar, struct schattr *attr)
{	
	int e;
	ctask_t *ptask;
	
	if (!Invokable(group_create, pool, pmex)) {
		if (gid)
			return POOL_ERR_GROUP_NOT_FOUND;	
	
		return stpool_add_routine(pool, name, task_run, task_err_handler, task_ar, attr);
	}
	
	/**
	 * We try to et a task object from the cache, and
	 * @__stpool_cache_put(x) is should be called later 
	 * to recycle it
	 */
	ptask = __stpool_cache_get(pool);
	if (!ptask)
		return POOL_ERR_NOMEM;
	
	__stpool_task_INIT(ptask, name, task_run, task_err_handler, task_ar);
	ptask->gid = gid;
	ptask->f_vmflags |= eTASK_VM_F_LOCAL_CACHE;
	
	if (attr)
		stpool_task_setschattr(TASK_CAST_UP(ptask), attr);
	
	/**
	 * Pay the task object back to cache if we fail 
	 * to add it into the pool's pendin queue 
	 */
	if ((e = stpool_task_queue(TASK_CAST_UP(ptask)))) 
		__stpool_cache_put(pool, ptask);

	return e;
}

EXPORT int  
stpool_group_remove_all(stpool_t *pool, int gid, int dispatched_by_pool)
{
	long lflas = dispatched_by_pool ? eTASK_VM_F_REMOVE_BYPOOL : eTASK_VM_F_REMOVE;
	
	if (!gid && !Invokable(group_create, pool, pmex))
		return stpool_remove_all(pool, dispatched_by_pool);

	if (Invokable(group_remove_all, pool, pmex)) {
		MSG_log(M_POOL, LOG_INFO,
			"{\"%s\"/%p} Markin all group(%d)'s tasks (%d)...\n",
			pool->desc, pool, gid, dispatched_by_pool);

		return Invoke(group_remove_all, pool, pmex, gid, dispatched_by_pool);
	
	} else if (Invokable(group_mark_all, pool, pmex))
		return Invoke(group_mark_all, pool, pmex, gid, lflas);
	else
		return 0;
}

EXPORT int  
stpool_group_mark_all(stpool_t *pool, int gid, long lflas)
{
	MSG_log(M_POOL, LOG_INFO,
			"{\"%s\"/%p} Markin all group(%d)'s tasks with %p ...\n",
			pool->desc, pool, gid, lflas);
	
	if (!gid && !Invokable(group_create, pool, pmex))
		return stpool_mark_all(pool, lflas);

	if (!Invokable(group_mark_all, pool, pmex))
		return POOL_ERR_NSUPPORT;
	
	return Invoke(group_mark_all, pool, pmex, gid, lflas);
}

EXPORT int  
stpool_group_mark_cb(stpool_t *pool, int gid, Walk_cb wcb, void *wcb_ar)
{
	if (!gid && !Invokable(group_create, pool, pmex))
		return stpool_mark_cb(pool, wcb, wcb_ar);

	if (!Invokable(group_mark_cb, pool, pmex))
		return POOL_ERR_NSUPPORT;
	
	return Invoke(group_mark_cb, pool, pmex, gid, (Visit_cb)wcb, wcb_ar);
}

EXPORT int  
stpool_group_wait_all(stpool_t *pool, int gid, long ms)
{
	MSG_log(M_POOL, LOG_INFO,
			"{\"%s\"/%p} Start waitin for all group(%d) tasks's bein done ... (%ld ms)\n",
			pool->desc, pool, gid, ms);
	
	if (!gid && !Invokable(group_create, pool, pmex))
		return stpool_wait_all(pool, ms);
	
	TRY_Invoke_return_res(POOL_ERR_NSUPPORT, group_wait_all, pool, pmex, gid, ms); 
}

EXPORT int  
stpool_group_wait_cb(stpool_t *pool, int gid, Walk_cb wcb, void *wcb_ar, long ms)
{
	if (!gid && !Invokable(group_create, pool, pmex))
		return stpool_wait_cb(pool, wcb, wcb_ar, ms);

	if (!Invokable(group_wait_cb, pool, pmex)) 
		return POOL_ERR_NSUPPORT;
	
	return Invoke(group_wait_cb, pool, pmex, gid, (Visit_cb)wcb, wcb_ar, ms);
}

EXPORT int  
stpool_group_wait_any(stpool_t *pool, int gid, long ms)
{
	int e;

	if (!gid && !Invokable(group_create, pool, pmex))
		e = stpool_wait_any(pool, ms);
	else
		TRY_Invoke_err(e, group_wait_any, pool, pmex, gid, ms);
	
	return e;
}

EXPORT void
stpool_group_throttle_enable(stpool_t *pool, int gid, int enable)
{
	MSG_log(M_POOL, LOG_INFO,
			"{\"%s\"/%p} %s the group(%d)'s throttle ...\n",
			pool->desc, pool, enable ? "ENABLIN" : "DISABLIN", gid);
	
	if (!gid && !Invokable(group_create, pool, pmex))
		stpool_throttle_enable(pool, enable);
	else
		TRY_Invoke(group_throttle_enable, pool, pmex, gid, enable);
}

EXPORT int 
stpool_group_throttle_wait(stpool_t *pool, int gid, long ms)
{
	if (!gid && !Invokable(group_create, pool, pmex))
		return stpool_throttle_wait(pool, ms);
		
	TRY_Invoke_return_res(0, group_throttle_wait, pool, pmex, gid, ms);
}

EXPORT void 
stpool_group_delete(stpool_t *pool, int gid)
{
	TRY_Invoke(group_delete, pool, pmex, gid);
}