MESSAGE_DISPATCH.js 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179
  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. var axios = require('axios');
  6. const HttpClient = require('../src/HttpClient');
  7. const WebSocketClient = require('../src/WebSocketClient');
  8. const rabbitMq = require('../src/mq/rabbit-mq');
  9. console.log("运行 MESSAGE_DISPATCH")
  10. CMD.updateFilterConfig = async function(){
  11. const postData = {
  12. cmd:"db_filter_config",
  13. fun:"getConfig",
  14. data:{}
  15. };
  16. axios.post('http://127.0.0.1:9100/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.updateMainConfig = async function(){
  25. const postData = {
  26. cmd:"tg_main",
  27. fun:"getOpenMainList",
  28. data:{}
  29. };
  30. axios.post('http://127.0.0.1:9100/tg/back/api', postData)
  31. .then(response => {
  32. redis_help.setKeyValue("MainConfig",JSON.stringify(response.data.data))
  33. })
  34. .catch(error => {
  35. console.error('updateFilterConfig error!', error);
  36. });
  37. }
  38. CMD.updateAppConfig = async function(){
  39. const postData = {
  40. cmd:"tg_app",
  41. fun:"getAppList",
  42. data:{}
  43. };
  44. axios.post('http://127.0.0.1:9100/tg/back/api', postData)
  45. .then(response => {
  46. redis_help.setKeyValue("AppConfig",JSON.stringify(response.data.data))
  47. })
  48. .catch(error => {
  49. console.error('updateFilterConfig error!', error);
  50. });
  51. }
  52. CMD.updatePlatformConfig = async function(){
  53. const postData = {
  54. cmd:"tg_platform",
  55. fun:"getPlatformList",
  56. data:{}
  57. };
  58. axios.post('http://127.0.0.1:9100/tg/back/api', postData)
  59. .then(response => {
  60. redis_help.setKeyValue("PlatformConfig",JSON.stringify(response.data.data))
  61. })
  62. .catch(error => {
  63. console.error('updateFilterConfig error!', error);
  64. });
  65. }
  66. var ws_isOpen = false
  67. CMD.myListener = {
  68. onOpen: function (webSocket, response) {
  69. ws_isOpen = true
  70. //打开链接后,想服务器端发送一条消息
  71. sendMessage("status_task",{work:1})
  72. },
  73. onMessage: function (webSocket, msg) { //msg可能是字符串,也可能是byte数组,取决于服务器送的内容
  74. recvMessage(msg)
  75. },
  76. onClosing: function (webSocket, code, response) {
  77. ws_isOpen = false
  78. },
  79. onClosed: function (webSocket, code, response) {
  80. ws_isOpen = false
  81. },
  82. onFailure: function (webSocket, t, response) {
  83. }
  84. }
  85. function sendMessage(cmd,data){
  86. if(ws_isOpen==true){
  87. var json = {};
  88. json.cmd=cmd; //方法
  89. json.data= data; //参数
  90. var msg=JSON.stringify(json);
  91. if(CMD.wsClient!=undefined&&CMD.wsClient!=null){
  92. CMD.wsClient.sendMsg(msg);
  93. }
  94. }
  95. }
  96. async function sendMQMessage(message,routingKey = null) {
  97. try {
  98. if(routingKey!=null){
  99. await rabbitMq.producerDirectMsg( message,"exchange_system",routingKey);
  100. }else{
  101. await rabbitMq.producerDirectMsg( message,"exchange_system");
  102. }
  103. console.log('消息发送成功');
  104. } catch (error) {
  105. console.error('发送消息失败:', error);
  106. }
  107. }
  108. async function recvMessage(data){
  109. try {
  110. // 单个任务
  111. let json_msg = JSON.parse(data)
  112. console.log("recvMessage:",json_msg.cmd)
  113. switch (json_msg.cmd) {
  114. case "updateFilterConfig":
  115. CMD.updateFilterConfig()
  116. break;
  117. case "updateMainConfig":
  118. CMD.updateMainConfig()
  119. break;
  120. case "updateAppConfig":
  121. CMD.updateAppConfig()
  122. break;
  123. case "updatePlatformConfig":
  124. CMD.updatePlatformConfig()
  125. break;
  126. default:
  127. const postData = {
  128. cmd:"filter_task",
  129. fun:json_msg.cmd,
  130. data:json_msg
  131. };
  132. sendMQMessage(postData,json_msg.cmd)
  133. // const client = new HttpClient({
  134. // timeout: 5000,
  135. // // headers: {
  136. // // 'User-Agent': 'Custom User Agent'
  137. // // }
  138. // });
  139. // const response = await client.post('http://127.0.0.1:9111',postData);
  140. break;
  141. }
  142. }catch(e){
  143. console.error("recvMessage error1!:",e)
  144. }
  145. }
  146. CMD.init = function(){
  147. redis_help.connect((results)=>{
  148. if(results){
  149. CMD.wsClient = new WebSocketClient(config.ws_config.host, {
  150. reconnectDelay: 5000, // 5 seconds
  151. heartbeatInterval: 30000, // 30 seconds
  152. heartbeatTimeout: 5000 // 5 seconds
  153. },CMD.myListener);
  154. }
  155. })
  156. CMD.updateConfig()
  157. CMD.updateTimeConfig()
  158. }
  159. CMD.updateTimeConfig = function(){
  160. let create_heiyan_book_link = require('../src/api/hy/create_heiyan_book_link')
  161. create_heiyan_book_link.login((data)=>{
  162. console.log("hei_yan_token:",data['token'])
  163. redis_help.setKeyValue("hei_yan_token",data['token'])
  164. })
  165. }
  166. //定时更新配置
  167. CMD.updateConfig = function(){
  168. setInterval(()=>{
  169. CMD.updateTimeConfig()
  170. },time_count)
  171. }
  172. CMD.init()