telppa / ahkzmq Goto Github PK

#Noenv
#persistent

SetBatchLines -1

global hwnd_main := 0
global WM_NULL := 0x0000
global lp_process_ui_callback := RegisterCallback("process_ui_callback")
global lp_recevie_message := RegisterCallback("recevie_message")
global hThread := 0
global msg
global exit_flag := 0

;global CriSec := CriticalSection()
global message_queue := []

recevie_message() {
Critical 
  zmq := new ZeroMQ

  ; keypair := zmq.curve_keypair()
  ; MsgBox, % keypair.1 "`n" keypair.2

  context := zmq.context()
  socket := context.socket(zmq.SUB)

  socket.setsockopt_string(zmq.CURVE_SERVERKEY, "HIm0q5eoJ}Ur7-&prX?y{%wiI3L)I8>ge%sE}l/k", "utf-8")
  socket.setsockopt_string(zmq.CURVE_PUBLICKEY, "YHtx0zuc(ZkWoXaozP*JP+Kg<!#Dt.2S>oNUXn?Y", "utf-8")
  socket.setsockopt_string(zmq.CURVE_SECRETKEY, "{r]PaKZ%WD.q$VAk=E5jMeFkMkSep.Lk]voT/db+", "utf-8")

  socket.connect("tcp://192.168.79.129:3002")

  filter := "10001"
  socket.setsockopt_string(zmq.SUBSCRIBE, filter, "utf-8")

  poller := zmq.poller([[socket, zmq.POLLIN]])

  loop
  {

    if (exit_flag == 1)
      break

    socks := poller.poll()

    if (socks[1]) {

      msg := socket.recv_string(zmq.DONTWAIT, "utf-8", false)
      if (msg is integer && msg < 0) {
        if (socket.errno() == zmq.EAGAIN)
          continue
        if (socket.errno() == zmq.EINTR)
          continue
      }

      ;ToolTip % "weather: " msg[1] msg[2]
      ; 这里不应该直接给 UI 线程赋值，应该 postmessage
      ;GuiControl, Text, TestText, % msg.1 msg.2

      ;EnterCriticalSection(CriSec)
      message_queue.Push(msg)
      ;LeaveCriticalSection(CriSec)

      PostMessage, WM_NULL, 0, lp_process_ui_callback, , ahk_id %hwnd_main%

    }
  }

}

process_ui_callback()
{
   ;EnterCriticalSection(CriSec)
   if (message_queue.Length() > 0) {
      msg := message_queue.RemoveAt(1) 
   }   
   ;LeaveCriticalSection(CriSec)

  GuiControl, Text, TestText, % msg.1 " " msg.2 " " msg.3
}

Gui, +AlwaysOnTop +hWndhwnd_main
Gui, Add, Text, w1200 vTestText, 等待回调UI
Gui, Add, Button, gStart, Start
Gui, Add, Button, gGoButton, Exit
Gui, Show, w1200 h100,接收zeromq消息

OnMessage(WM_NULL, "ON_WM_NULL")

return

ON_WM_NULL(wParam, lParam)
{
  if (lParam != 0) {
    DllCall(lParam)
  }


 

  ; Func := Func("process_ui_callback")
  ; Func.Call(msg)
}

Start:
  if (!hThread)
    hThread := DllCall("CreateThread", Ptr, 0, Ptr, 0, Ptr, lp_recevie_message, Ptr, 0, UInt, 0, Ptr, 0)
return

GoButton:
  exit_flag := 1
  DllCall("WaitForSingleObject", "PTR", hThread, "UInt", 0xFFFFFFFF)
  DllCall("CloseHandle", "PTR", hThread)
ExitApp
return

#Include %A_LineFile%\..\..\ZeroMQ\ZeroMQ.ahk