MAIN_SERVE_FACTORY.js 8.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248
  1. const CMD = {}
  2. const redis_help = require('../src/use_redis');
  3. const rabbitMq = require('../src/mq/rabbit-mq');
  4. const config = require('../etc/config.json')
  5. const axios = require('axios');
  6. const mysql = require('mysql2/promise');
  7. const http = require('http');
  8. const tools = require('../tools');
  9. const dbConfig = config.isDebug?config.debug_task_mysql:config.release_task_mysql
  10. async function processTask(tg_platform_id, main_info) {
  11. try {
  12. console.log(`处理第 ${cur_page} 页任务`);
  13. const response = await CMD.syncMain(tg_platform_id, main_info);
  14. if (response !== null) {
  15. cur_page++;
  16. total = response.total;
  17. task_que.push(response);
  18. // 使用 Promise 延迟执行下一页
  19. await new Promise(resolve => setTimeout(resolve, 1000));
  20. return processTask(tg_platform_id, main_info);
  21. } else {
  22. console.log("完成所有创建:", main_info, cur_page);
  23. cur_page = -1;
  24. isStartTask = false;
  25. await redis_help.setKeyValue("isPauseTask", "false");
  26. }
  27. } catch (e) {
  28. console.error("processTask 错误:", e);
  29. isStartTask = false;
  30. await redis_help.setKeyValue("isPauseTask", "false");
  31. // if (cur_page !== -1) {
  32. // console.log("5秒后重试当前页...");
  33. // await new Promise(resolve => setTimeout(resolve, 5000));
  34. // return processTask(tg_platform_id, main_info);
  35. // }
  36. }
  37. }
  38. async function processSyncMainTask(){
  39. try{
  40. if(task_que.length>0){
  41. let sync_main_task_item = task_que.shift()
  42. let product_list = sync_main_task_item.data;
  43. let main_info = sync_main_task_item.main_info;
  44. let main_id = main_info.id
  45. const connection = await mysql.createConnection({
  46. ...dbConfig,
  47. multipleStatements: true
  48. });
  49. const values = product_list.map(item => [
  50. item.product_name,
  51. item.product_id,
  52. item.book_platform,
  53. main_id,
  54. main_info.app_id,
  55. 0,
  56. 0
  57. ]);
  58. // let table_name = "video_applet_product"
  59. // const insertSQL = `
  60. // INSERT INTO ${table_name}
  61. // (product_name, product_id, book_platform, main_id, dy_small_applet_app_id,status,
  62. // wait_status)
  63. // VALUES ?
  64. // `;
  65. const insertSQL = `
  66. INSERT INTO video_applet_product
  67. (product_name, product_id, book_platform, main_id, dy_small_applet_app_id, status, wait_status)
  68. VALUES ?
  69. ON DUPLICATE KEY UPDATE
  70. product_name = VALUES(product_name),
  71. dy_small_applet_app_id = VALUES(dy_small_applet_app_id)
  72. `;
  73. await connection.query(insertSQL, [values]);
  74. await connection.end();
  75. }
  76. }catch(e){
  77. console.error("processSyncMainTask:",e)
  78. }finally{
  79. setTimeout(processSyncMainTask, 1000);
  80. }
  81. }
  82. const messageHandler = async (msg) => {
  83. // CMD.start_task(msg['data']['data'])
  84. if(isStartTask){
  85. return
  86. }
  87. cur_page = 1;
  88. total = 0;
  89. isStartTask = true
  90. let main_info = msg['data']['data'].main_info
  91. console.log("main_info:",main_info)
  92. if(main_info.parent_platform_id!=0){
  93. tg_platform_id = main_info.parent_platform_id
  94. }else{
  95. tg_platform_id = main_info.tg_platform_id
  96. }
  97. await redis_help.setKeyValue("isPauseTask","true")
  98. await processTask(tg_platform_id,main_info)
  99. };
  100. // // 启动消费者
  101. // async function startConsumer() {
  102. // try {
  103. // await rabbitMq.consumerDirectMsg(messageHandler,"exchange_system","addMain");
  104. // } catch (error) {
  105. // console.error('启动消费者失败:', error);
  106. // }
  107. // }
  108. // 启动消费者
  109. async function startConsumer() {
  110. try {
  111. await rabbitMq.consumerDirectMsg(messageHandler,"exchange_system","syncMain");
  112. } catch (error) {
  113. console.error('启动消费者失败:', error);
  114. }
  115. processSyncMainTask()
  116. }
  117. CMD.init = async function(){
  118. redis_help.connect(()=>{
  119. })
  120. await startConsumer();
  121. }
  122. let task_que = []
  123. let cur_page = 1;
  124. let total = 0;
  125. let isStartTask = false
  126. CMD.syncMain = async function(tg_platform_id, main_info) {
  127. console.log(`====== 开始第 ${cur_page} 页请求 =====`);
  128. await new Promise(resolve => setTimeout(resolve, 500));
  129. try {
  130. let postData = {
  131. cmd: "video_product",
  132. fun: "search_book_data",
  133. data: {
  134. "product_name": "",
  135. "product_id": "",
  136. "tg_platform_id": tg_platform_id,
  137. "oce_material_id": "",
  138. "page_size": 500,
  139. "page_number": cur_page,
  140. "is_auto": "",
  141. "is_store": "",
  142. "genre": main_info.genre,
  143. "alias_name": ""
  144. }
  145. };
  146. console.log("发送请求数据:", postData);
  147. let client = tools.getOneNewClinet()
  148. let response = await client.post('http://127.0.0.1:9100/tg/back/api', postData, {
  149. headers: {
  150. 'Content-Type': 'application/json',
  151. 'X-Request-Type': 'API', // 标记 API 请求
  152. }
  153. });
  154. response = response.data
  155. // 检查响应数据结构
  156. if (!response || !response.data || !Array.isArray(response.data)) {
  157. console.log("响应数据无效或格式错误");
  158. return null;
  159. }
  160. // 检查是否有数据
  161. if (response.data.length <= 0) {
  162. console.log("没有更多数据");
  163. return null;
  164. }
  165. // 返回正确的数据结构
  166. return {
  167. data: response.data,
  168. main_info: main_info,
  169. total: response.total || 0
  170. };
  171. } catch (e) {
  172. console.error("请求错误:", {
  173. message: e.message,
  174. code: e.code,
  175. stack: e.stack
  176. });
  177. return null;
  178. }
  179. };
  180. // CMD.start_task = async function(msgBody){
  181. // let PlatformConfig = JSON.parse(await redis_help.getKeyValue("PlatformConfig"))
  182. // let main_info = msgBody.main_info
  183. // let list = msgBody.list
  184. // let PlatformInfo = null
  185. // if(main_info.running_status==0){
  186. // console.log("创建了一个关闭的主体:",main_info)
  187. // return
  188. // }
  189. // console.log("main_info:",main_info)
  190. // console.log("list:",list)
  191. // for (let index = 0; index < PlatformConfig.length; index++) {
  192. // const element = PlatformConfig[index];
  193. // if(element.tg_platform_id == main_info.tg_platform_id){
  194. // PlatformInfo = element
  195. // break
  196. // }
  197. // }
  198. // if(PlatformInfo==null){
  199. // console.log("平台配置错误:",msgBody)
  200. // }else{
  201. // redis_help.setKeyValue("isPauseTask","true")
  202. // console.log("list::",list)
  203. // for (let index = 0; index < list.length; index++) {
  204. // const element = list[index];
  205. // let n_data = {book_id:element.product_id,
  206. // book_name:element.product_name,
  207. // tg_platform_id:element.book_platform,
  208. // app_id:element.dy_small_applet_app_id,
  209. // main_id:main_info.id
  210. // }
  211. // const result = await video_applet_product_controllers.createAppletProductData({
  212. // book_platform:n_data.tg_platform_id,
  213. // product_name:n_data.book_name,
  214. // product_id:n_data.book_id,
  215. // dy_small_applet_app_id:n_data.app_id,
  216. // status:0,
  217. // main_id:n_data.main_id,
  218. // promotion_id:'',
  219. // dy_small_program_start:'',
  220. // dy_small_program_start_data:'',
  221. // wait_status:0,
  222. // })
  223. // }
  224. // redis_help.setKeyValue("isPauseTask","false")
  225. // }
  226. // }
  227. CMD.init()