/*
 * @Author: yangzilong
 * @Date: 2021-12-15 17:59:50
 * @Last Modified by: yangzilong
 * @Email: yangzilong@objecteye.com
 * @Description:
 */
#include "stl_aiplatform.h"
#include "MultiSourceProcess.h"


int tsl_aiplatform_init(void **handle, tsl_aiplatform_param param)
{
	*handle = new CMultiSourceProcess();
	CMultiSourceProcess* tools = (CMultiSourceProcess*)*handle;
	return tools->InitAlgorthim(param);
}


#ifdef POST_USE_RABBITMQ
int add_mq_conn(void *handle, mq_type_t tstatus, rabbitmq_conn_params_t mq_conn_param)
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	int res = tools->AddMqConn(tstatus, mq_conn_param);
	return res;
}


int get_task_status(void *handle, char *task_id)
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	int res = tools->GetTaskStatus(task_id);
	return res;
}
#endif


int add_task(void *handle, task_param param)
{
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	int res = tools->AddTask(param);
	return res;
}


int pause_task(void *handle, char * task_id, const int max_timeout_ms)
{
	int error_code = FAILED;
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	error_code = tools->PauseTask(task_id);
	return error_code;
}


int restart_task(void *handle, char * task_id, const int max_timeout_ms)
{
	int error_code = FAILED;
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	error_code = tools->RestartTask(task_id);
	return error_code;
}

int finish_task(void *handle, char * task_id, const int max_timeout_ms)
{
	int error_code = FAILED;
	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
	error_code = tools->FinishTask(task_id);
	return error_code;
}


int tsl_aiplatform_release(void **handle)
{
	if (*handle)
	{
		CMultiSourceProcess* tools = (CMultiSourceProcess*)*handle;
		tools->CloseAllTask();

		delete tools;
		tools = NULL;
	}
	return SUCCESS;
}


const char* get_tsl_aiplatform_version()
{
	return "sy_stl_aiplatform_version:0.0.0.220803.gpu.x64";
}