MESSAGE_DISPATCH.js 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405
  1. const CMD = {}
  2. const time_count = 60000*5 //60秒更新一次配置
  3. const config = require('../etc/config.json')
  4. const redis_help = require('../src/use_redis');
  5. const WebSocketClient = require('../src/WebSocketClient');
  6. const rabbitMq = require('../src/mq/rabbit-mq');
  7. const tools = require('../tools');
  8. console.log("运行 MESSAGE_DISPATCH")
  9. CMD.updateFilterConfig = async function(){
  10. const postData = {
  11. cmd:"db_filter_config",
  12. fun:"getConfig",
  13. data:{}
  14. };
  15. let client = tools.getOneNewClinet()
  16. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  17. .then(response => {
  18. redis_help.setKeyValue("FilterConfig",JSON.stringify(response.data.data))
  19. })
  20. .catch(error => {
  21. console.error('updateFilterConfig error!', error);
  22. });
  23. }
  24. CMD.updateCloseMainConfig = async function(){
  25. const postData = {
  26. cmd:"tg_main",
  27. fun:"getCloseCreateLinkMainList",
  28. data:{}
  29. };
  30. let client = tools.getOneNewClinet()
  31. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  32. .then(response => {
  33. redis_help.setKeyValue("CloseMainConfig",JSON.stringify(response.data.data))
  34. })
  35. .catch(error => {
  36. console.error('updateCloseMainConfig error!', error);
  37. });
  38. }
  39. CMD.updateMainConfig = async function(){
  40. const postData = {
  41. cmd:"tg_main",
  42. fun:"getOpenCreateLinkMainList",
  43. data:{}
  44. };
  45. let client = tools.getOneNewClinet()
  46. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  47. .then(response => {
  48. CMD.updateCloseMainConfig()
  49. redis_help.setKeyValue("MainConfig",JSON.stringify(response.data.data))
  50. })
  51. .catch(error => {
  52. console.error('updateFilterConfig error!', error);
  53. });
  54. }
  55. CMD.updateAppConfig = async function(){
  56. const postData = {
  57. cmd:"tg_app",
  58. fun:"getAppList",
  59. data:{}
  60. };
  61. let client = tools.getOneNewClinet()
  62. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  63. .then(response => {
  64. redis_help.setKeyValue("AppConfig",JSON.stringify(response.data.data))
  65. })
  66. .catch(error => {
  67. console.error('updateFilterConfig error!', error);
  68. });
  69. }
  70. CMD.updatePlatformConfig = async function(){
  71. const postData = {
  72. cmd:"tg_platform",
  73. fun:"getPlatformList",
  74. data:{}
  75. };
  76. let client = tools.getOneNewClinet()
  77. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  78. .then(response => {
  79. CMD.updatePlatformShared()
  80. redis_help.setKeyValue("PlatformConfig",JSON.stringify(response.data.data))
  81. })
  82. .catch(error => {
  83. console.error('updateFilterConfig error!', error);
  84. });
  85. }
  86. CMD.updatePlatformShared = async function(){
  87. const postData = {
  88. cmd:"platform_shared",
  89. fun:"gePlatformSharedList",
  90. data:{}
  91. };
  92. let client = tools.getOneNewClinet()
  93. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  94. .then(response => {
  95. redis_help.setKeyValue("PlatformShared",JSON.stringify(response.data.data))
  96. })
  97. .catch(error => {
  98. console.error('PlatformShared error!', error);
  99. });
  100. }
  101. CMD.updateFqMfKeyList = async function(){
  102. const postData = {
  103. cmd:"fq_mf_book",
  104. fun:"get_all_fq_mf_key",
  105. data:{}
  106. };
  107. let client = tools.getOneNewClinet()
  108. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  109. .then(response => {
  110. redis_help.setKeyValue("all_fq_mf_key",JSON.stringify(response.data.data))
  111. })
  112. .catch(error => {
  113. console.error('all_fq_mf_key error!', error);
  114. });
  115. }
  116. CMD.updateYwKeyList = async function(){
  117. const postData = {
  118. cmd:"yw_book",
  119. fun:"get_all_yw_key",
  120. data:{}
  121. };
  122. let client = tools.getOneNewClinet()
  123. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  124. .then(response => {
  125. console.log("updateYwKeyList:",response.data.data)
  126. redis_help.setKeyValue("all_yw_key",JSON.stringify(response.data.data))
  127. })
  128. .catch(error => {
  129. console.error('all_yw_key error!', error);
  130. });
  131. }
  132. CMD.updateFqKeyList = async function(){
  133. const postData = {
  134. cmd:"fq_book",
  135. fun:"get_all_fq_key",
  136. data:{}
  137. };
  138. let client = tools.getOneNewClinet()
  139. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  140. .then(response => {
  141. redis_help.setKeyValue("all_fq_key",JSON.stringify(response.data.data))
  142. })
  143. .catch(error => {
  144. console.error('all_fq_key error!', error);
  145. });
  146. }
  147. CMD.updateBlackBooks = async function(){
  148. const postData = {
  149. cmd:"book_black_list",
  150. fun:"get_all_black_books",
  151. data:{}
  152. };
  153. let client = tools.getOneNewClinet()
  154. client.post('http://127.0.0.1:9109/tg/back/api', postData)
  155. .then(response => {
  156. redis_help.setKeyValue("all_black_books",JSON.stringify(response.data.data))
  157. })
  158. .catch(error => {
  159. console.error('all_black_books error!', error);
  160. });
  161. }
  162. CMD.getPullDataConfig = async function() {
  163. const postData = {
  164. cmd:"pull_data_config",
  165. fun:"get_all_pull_data_list",
  166. data:{}
  167. };
  168. let client = tools.getOneNewClinet()
  169. return await client.post('http://127.0.0.1:9109/tg/back/api', postData)
  170. .then(response => {
  171. redis_help.setKeyValue("PullDataConfig",JSON.stringify(response.data.data))
  172. })
  173. .catch(error => {
  174. console.error('getPullDataConfig error!', error);
  175. });
  176. }
  177. CMD.getPullDataConfig2 = async function() {
  178. const postData = {
  179. cmd:"pull_data_config2",
  180. fun:"get_all_pull_data_list",
  181. data:{}
  182. };
  183. let client = tools.getOneNewClinet()
  184. return await client.post('http://127.0.0.1:9109/tg/back/api', postData)
  185. .then(response => {
  186. redis_help.setKeyValue("PullDataConfig2",JSON.stringify(response.data.data))
  187. })
  188. .catch(error => {
  189. console.error('getPullDataConfig2 error!', error);
  190. });
  191. }
  192. var ws_isOpen = false
  193. CMD.myListener = {
  194. onOpen: function (webSocket, response) {
  195. ws_isOpen = true
  196. //打开链接后,想服务器端发送一条消息
  197. sendMessage("status_task",{work:1})
  198. },
  199. onMessage: function (webSocket, msg) { //msg可能是字符串,也可能是byte数组,取决于服务器送的内容
  200. recvMessage(msg)
  201. },
  202. onClosing: function (webSocket, code, response) {
  203. ws_isOpen = false
  204. },
  205. onClosed: function (webSocket, code, response) {
  206. ws_isOpen = false
  207. },
  208. onFailure: function (webSocket, t, response) {
  209. }
  210. }
  211. function sendMessage(cmd,data){
  212. if(ws_isOpen==true){
  213. var json = {};
  214. json.cmd=cmd; //方法
  215. json.data= data; //参数
  216. var msg=JSON.stringify(json);
  217. if(CMD.wsClient!=undefined&&CMD.wsClient!=null){
  218. CMD.wsClient.sendMsg(msg);
  219. }
  220. }
  221. }
  222. async function sendMQMessage(message,routingKey = null) {
  223. try {
  224. if(routingKey!=null){
  225. await rabbitMq.producerDirectMsg( message,"exchange_system",routingKey);
  226. }else{
  227. await rabbitMq.producerDirectMsg( message,"exchange_system");
  228. }
  229. console.log('消息发送成功');
  230. } catch (error) {
  231. console.error('发送消息失败:', error);
  232. }
  233. }
  234. async function sendPullMQMessage(message,routingKey = null) {
  235. try {
  236. if(routingKey!=null){
  237. await rabbitMq.producerDirectMsg( message,"exchange_update_pull_config",routingKey);
  238. }else{
  239. await rabbitMq.producerDirectMsg( message,"exchange_update_pull_config");
  240. }
  241. console.log('消息发送成功');
  242. } catch (error) {
  243. console.error('发送消息失败:', error);
  244. }
  245. }
  246. async function sendPull2MQMessage(message,routingKey = null) {
  247. try {
  248. if(routingKey!=null){
  249. await rabbitMq.producerDirectMsg( message,"exchange_update_pull_config2",routingKey);
  250. }else{
  251. await rabbitMq.producerDirectMsg( message,"exchange_update_pull_config2");
  252. }
  253. console.log('消息发送成功');
  254. } catch (error) {
  255. console.error('发送消息失败:', error);
  256. }
  257. }
  258. async function recvMessage(data){
  259. try {
  260. // 单个任务
  261. let json_msg = JSON.parse(data)
  262. console.log("recvMessage:",json_msg.cmd)
  263. switch (json_msg.cmd) {
  264. case "updateFilterConfig":
  265. CMD.updateFilterConfig()
  266. break;
  267. case "updateMainConfig":
  268. CMD.updateMainConfig()
  269. break;
  270. case "updateAppConfig":
  271. CMD.updateAppConfig()
  272. break;
  273. case "updatePlatformConfig":
  274. CMD.updatePlatformConfig()
  275. break;
  276. case "updateFqKeyList":
  277. CMD.updateFqKeyList()
  278. break;
  279. case "updateFqMfKeyList":
  280. CMD.updateFqMfKeyList()
  281. break;
  282. case "updateYwKeyList":
  283. await CMD.updateYwKeyList()
  284. await tools.getYwOPENSESSID()
  285. break;
  286. case "updateBlackBooks":
  287. CMD.updateBlackBooks()
  288. break;
  289. // case "addMain":
  290. case "syncMain":
  291. {
  292. const postData = {
  293. cmd:"filter_task",
  294. fun:json_msg.cmd,
  295. data:json_msg
  296. };
  297. sendMQMessage(postData,json_msg.cmd)
  298. }
  299. break;
  300. case "getPullDataConfig": //获取拉取数据类配置
  301. {
  302. CMD.getPullDataConfig()
  303. CMD.getPullDataConfig2()
  304. }
  305. break;
  306. case "updatePullConig": //同步拉取配置
  307. {
  308. await CMD.getPullDataConfig()
  309. await CMD.getPullDataConfig2()
  310. await new Promise(resolve=>setTimeout(resolve,100))
  311. const postData = {
  312. cmd:"filter_task",
  313. fun:json_msg.cmd,
  314. data:json_msg
  315. };
  316. sendPullMQMessage(postData,json_msg.cmd)
  317. }
  318. break;
  319. case "updatePullConig2": //同步拉取配置
  320. {
  321. await CMD.getPullDataConfig2()
  322. await new Promise(resolve=>setTimeout(resolve,100))
  323. const postData = {
  324. cmd:"filter_task",
  325. fun:json_msg.cmd,
  326. data:json_msg
  327. };
  328. sendPull2MQMessage(postData,json_msg.cmd)
  329. }
  330. break;
  331. default:
  332. // const client = new HttpClient({
  333. // timeout: 5000,
  334. // // headers: {
  335. // // 'User-Agent': 'Custom User Agent'
  336. // // }
  337. // });
  338. // const response = await client.post('http://127.0.0.1:9111',postData);
  339. break;
  340. }
  341. }catch(e){
  342. console.error("recvMessage error1!:",e)
  343. }
  344. }
  345. CMD.init = function(){
  346. redis_help.connect((results)=>{
  347. if(results){
  348. CMD.wsClient = new WebSocketClient(config.ws_config.host, {
  349. reconnectDelay: 5000, // 5 seconds
  350. heartbeatInterval: 30000, // 30 seconds
  351. heartbeatTimeout: 5000 // 5 seconds
  352. },CMD.myListener);
  353. }
  354. })
  355. CMD.updateConfig()
  356. CMD.updateTimeConfig()
  357. // setInterval(async () => {
  358. // await require('../src/api/yw/check_login').update_cookies()
  359. // }, 3000*20);
  360. }
  361. CMD.updateTimeConfig = function(){
  362. // let create_heiyan_book_link = require('../src/api/hy/create_heiyan_book_link')
  363. // create_heiyan_book_link.login((data)=>{
  364. // console.log("hei_yan_token:",data['token'])
  365. // redis_help.setKeyValue("hei_yan_token",data['token'])
  366. // })
  367. // let qm_login = require('../src/api/qm/qm_login')
  368. // qm_login.login((data,headers)=>{
  369. // console.log("qi_mao_token:",headers['n-token'])
  370. // redis_help.setKeyValue("qi_mao_token",headers['n-token'])
  371. // })
  372. require('../src/api/dz/dz_login').login()
  373. require('../src/api/dz/dz_mf_login').login()
  374. }
  375. //定时更新配置
  376. CMD.updateConfig = function(){
  377. setInterval(()=>{
  378. CMD.updateTimeConfig()
  379. },time_count)
  380. }
  381. CMD.init()