Hu Chunming / src

Blame view

tsl_aiplatform/ai_platform/stl_aiplatform.cpp 1.93 KB
Edit Raw Normal View History
85cc8cb9   Hu Chunming   原版代码 
1
2
3
4
5
6
7
8
 
  /*
   * @Author: yangzilong
   * @Date: 2021-12-15 17:59:50
   * @Last Modified by: yangzilong
   * @Email: yangzilong@objecteye.com
   * @Description:
   */
  #include "stl_aiplatform.h"
31a1e592   Hu Chunming   完成接口匹配 
9
 
  #include "MultiSourceProcess.h"
85cc8cb9   Hu Chunming   原版代码 
10
11
12
13
 
  
  
  int tsl_aiplatform_init(void **handle, tsl_aiplatform_param param)
  {
31a1e592   Hu Chunming   完成接口匹配 
14
15
 
  	*handle = new CMultiSourceProcess();
  	CMultiSourceProcess* tools = (CMultiSourceProcess*)*handle;
85cc8cb9   Hu Chunming   原版代码 
16
17
18
19
20
21
22
 
  	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)
  {
31a1e592   Hu Chunming   完成接口匹配 
23
 
  	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
85cc8cb9   Hu Chunming   原版代码 
24
25
26
 
  	int res = tools->AddMqConn(tstatus, mq_conn_param);
  	return res;
  }
85cc8cb9   Hu Chunming   原版代码 
27
28
29
30
 
  
  
  int get_task_status(void *handle, char *task_id)
  {
31a1e592   Hu Chunming   完成接口匹配 
31
 
  	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
85cc8cb9   Hu Chunming   原版代码 
32
33
34
 
  	int res = tools->GetTaskStatus(task_id);
  	return res;
  }
450989b5   Hu Chunming   跑通算法 
35
 
  #endif
85cc8cb9   Hu Chunming   原版代码 
36
37
38
39
 
  
  
  int add_task(void *handle, task_param param)
  {
31a1e592   Hu Chunming   完成接口匹配 
40
41
 
  	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
  	int res = tools->AddTask(param);
85cc8cb9   Hu Chunming   原版代码 
42
43
44
45
46
47
48
 
  	return res;
  }
  
  
  int pause_task(void *handle, char * task_id, const int max_timeout_ms)
  {
  	int error_code = FAILED;
31a1e592   Hu Chunming   完成接口匹配 
49
50
 
  	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
  	error_code = tools->PauseTask(task_id);
85cc8cb9   Hu Chunming   原版代码 
51
52
53
54
55
56
57
 
  	return error_code;
  }
  
  
  int restart_task(void *handle, char * task_id, const int max_timeout_ms)
  {
  	int error_code = FAILED;
31a1e592   Hu Chunming   完成接口匹配 
58
59
 
  	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
  	error_code = tools->RestartTask(task_id);
85cc8cb9   Hu Chunming   原版代码 
60
61
62
63
64
65
 
  	return error_code;
  }
  
  int finish_task(void *handle, char * task_id, const int max_timeout_ms)
  {
  	int error_code = FAILED;
31a1e592   Hu Chunming   完成接口匹配 
66
67
 
  	CMultiSourceProcess* tools = (CMultiSourceProcess*)handle;
  	error_code = tools->FinishTask(task_id);
85cc8cb9   Hu Chunming   原版代码 
68
69
70
71
72
73
74
75
 
  	return error_code;
  }
  
  
  int tsl_aiplatform_release(void **handle)
  {
  	if (*handle)
  	{
31a1e592   Hu Chunming   完成接口匹配 
76
77
 
  		CMultiSourceProcess* tools = (CMultiSourceProcess*)*handle;
  		tools->CloseAllTask();
85cc8cb9   Hu Chunming   原版代码 
78
79
80
81
82
83
84
85
86
87
88
89
 
  
  		delete tools;
  		tools = NULL;
  	}
  	return SUCCESS;
  }
  
  
  const char* get_tsl_aiplatform_version()
  {
  	return "sy_stl_aiplatform_version:0.0.0.220803.gpu.x64";
  }