MESSAGE_DISPATCH.js 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382
  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: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.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:9100/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:9100/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:9100/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:9100/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:9100/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:9100/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. let response = await client.post('http://127.0.0.1:9100/tg/back/api', postData)
  124. console.log("updateYwKeyList:",response.data.data)
  125. redis_help.setKeyValue("all_yw_key",JSON.stringify(response.data.data))
  126. }
  127. CMD.updateFqKeyList = async function(){
  128. const postData = {
  129. cmd:"fq_book",
  130. fun:"get_all_fq_key",
  131. data:{}
  132. };
  133. let client = tools.getOneNewClinet()
  134. client.post('http://127.0.0.1:9100/tg/back/api', postData)
  135. .then(response => {
  136. redis_help.setKeyValue("all_fq_key",JSON.stringify(response.data.data))
  137. })
  138. .catch(error => {
  139. console.error('all_fq_key error!', error);
  140. });
  141. }
  142. CMD.updateBlackBooks = async function(){
  143. const postData = {
  144. cmd:"book_black_list",
  145. fun:"get_all_black_books",
  146. data:{}
  147. };
  148. let client = tools.getOneNewClinet()
  149. client.post('http://127.0.0.1:9100/tg/back/api', postData)
  150. .then(response => {
  151. redis_help.setKeyValue("all_black_books",JSON.stringify(response.data.data))
  152. })
  153. .catch(error => {
  154. console.error('all_black_books error!', error);
  155. });
  156. }
  157. CMD.getPullDataConfig = async function() {
  158. const postData = {
  159. cmd:"pull_data_config",
  160. fun:"get_all_pull_data_list",
  161. data:{}
  162. };
  163. let client = tools.getOneNewClinet()
  164. return await client.post('http://127.0.0.1:9100/tg/back/api', postData)
  165. .then(response => {
  166. redis_help.setKeyValue("PullDataConfig",JSON.stringify(response.data.data))
  167. })
  168. .catch(error => {
  169. console.error('getPullDataConfig error!', error);
  170. });
  171. }
  172. var ws_isOpen = false
  173. CMD.myListener = {
  174. onOpen: function (webSocket, response) {
  175. ws_isOpen = true
  176. //打开链接后,想服务器端发送一条消息
  177. sendMessage("status_task",{work:1})
  178. },
  179. onMessage: function (webSocket, msg) { //msg可能是字符串,也可能是byte数组,取决于服务器送的内容
  180. recvMessage(msg)
  181. },
  182. onClosing: function (webSocket, code, response) {
  183. ws_isOpen = false
  184. },
  185. onClosed: function (webSocket, code, response) {
  186. ws_isOpen = false
  187. },
  188. onFailure: function (webSocket, t, response) {
  189. }
  190. }
  191. function sendMessage(cmd,data){
  192. if(ws_isOpen==true){
  193. var json = {};
  194. json.cmd=cmd; //方法
  195. json.data= data; //参数
  196. var msg=JSON.stringify(json);
  197. if(CMD.wsClient!=undefined&&CMD.wsClient!=null){
  198. CMD.wsClient.sendMsg(msg);
  199. }
  200. }
  201. }
  202. async function sendMQMessage(message,routingKey = null) {
  203. try {
  204. if(routingKey!=null){
  205. await rabbitMq.producerDirectMsg( message,"exchange_system",routingKey);
  206. }else{
  207. await rabbitMq.producerDirectMsg( message,"exchange_system");
  208. }
  209. console.log('消息发送成功');
  210. } catch (error) {
  211. console.error('发送消息失败:', error);
  212. }
  213. }
  214. async function sendPullMQMessage(message,routingKey = null) {
  215. try {
  216. if(routingKey!=null){
  217. await rabbitMq.producerDirectMsg( message,"exchange_update_pull_config",routingKey);
  218. }else{
  219. await rabbitMq.producerDirectMsg( message,"exchange_update_pull_config");
  220. }
  221. console.log('消息发送成功');
  222. } catch (error) {
  223. console.error('发送消息失败:', error);
  224. }
  225. }
  226. async function recvMessage(data){
  227. try {
  228. // 单个任务
  229. let json_msg = JSON.parse(data)
  230. console.log("recvMessage:",json_msg.cmd)
  231. switch (json_msg.cmd) {
  232. case "updateFilterConfig":
  233. CMD.updateFilterConfig()
  234. break;
  235. case "updateMainConfig":
  236. CMD.updateMainConfig()
  237. break;
  238. case "updateAppConfig":
  239. CMD.updateAppConfig()
  240. break;
  241. case "updatePlatformConfig":
  242. CMD.updatePlatformConfig()
  243. break;
  244. case "updateFqKeyList":
  245. CMD.updateFqKeyList()
  246. break;
  247. case "updateFqMfKeyList":
  248. CMD.updateFqMfKeyList()
  249. break;
  250. case "updateYwKeyList":
  251. await CMD.updateYwKeyList()
  252. await tools.getYwOPENSESSID()
  253. break;
  254. case "updateBlackBooks":
  255. CMD.updateBlackBooks()
  256. break;
  257. // case "addMain":
  258. case "syncMain":
  259. {
  260. const postData = {
  261. cmd:"filter_task",
  262. fun:json_msg.cmd,
  263. data:json_msg
  264. };
  265. sendMQMessage(postData,json_msg.cmd)
  266. }
  267. break;
  268. case "getPullDataConfig": //获取拉取数据类配置
  269. {
  270. CMD.getPullDataConfig()
  271. }
  272. break;
  273. case "updatePullConig": //同步拉取配置
  274. {
  275. await CMD.getPullDataConfig()
  276. await new Promise(resolve=>setTimeout(resolve,100))
  277. const postData = {
  278. cmd:"filter_task",
  279. fun:json_msg.cmd,
  280. data:json_msg
  281. };
  282. sendPullMQMessage(postData,json_msg.cmd)
  283. }
  284. break;
  285. default:
  286. // const client = new HttpClient({
  287. // timeout: 5000,
  288. // // headers: {
  289. // // 'User-Agent': 'Custom User Agent'
  290. // // }
  291. // });
  292. // const response = await client.post('http://127.0.0.1:9111',postData);
  293. break;
  294. }
  295. }catch(e){
  296. console.error("recvMessage error1!:",e)
  297. }
  298. }
  299. CMD.init = function(){
  300. redis_help.connect((results)=>{
  301. if(results){
  302. CMD.wsClient = new WebSocketClient(config.ws_config.host, {
  303. reconnectDelay: 5000, // 5 seconds
  304. heartbeatInterval: 30000, // 30 seconds
  305. heartbeatTimeout: 5000 // 5 seconds
  306. },CMD.myListener);
  307. }
  308. })
  309. CMD.updateConfig()
  310. CMD.updateTimeConfig()
  311. // setInterval(async () => {
  312. // await require('../src/api/yw/check_login').update_cookies()
  313. // }, 3000*20);
  314. }
  315. CMD.updateTimeConfig = async function(){
  316. let create_heiyan_book_link = require('../src/api/hy/create_heiyan_book_link')
  317. create_heiyan_book_link.login((data)=>{
  318. console.log("hei_yan_token:",data['token'])
  319. redis_help.setKeyValue("hei_yan_token",data['token'])
  320. })
  321. let qm_login = require('../src/api/qm/qm_login')
  322. qm_login.login((data,headers)=>{
  323. console.log("qi_mao_token:",headers['n-token'])
  324. redis_help.setKeyValue("qi_mao_token",headers['n-token'])
  325. })
  326. let qm_mf_login = require('../src/api/qm_mf/qm_mf_login')
  327. qm_mf_login.login((data,headers)=>{
  328. console.log("qi_mao_mf_token:",headers['n-token'])
  329. redis_help.setKeyValue("qi_mao_mf_token",headers['n-token'])
  330. })
  331. require('../src/api/dz/dz_login').login()
  332. let zy_token = await require('../src/api/zy/zy_login').zy_login()
  333. if(zy_token!=null){
  334. console.log("zy_token:",zy_token)
  335. await redis_help.setKeyValue("zy_token",zy_token)
  336. }
  337. let zy_mf_token = await require('../src/api/zy_mf/zy_mf_login').zy_login()
  338. if(zy_mf_token!=null){
  339. console.log("zy_token:",zy_mf_token)
  340. await redis_help.setKeyValue("zy_mf_token",zy_mf_token)
  341. }
  342. }
  343. //定时更新配置
  344. CMD.updateConfig = function(){
  345. setInterval(()=>{
  346. CMD.updateTimeConfig()
  347. },time_count)
  348. }
  349. CMD.init()