File tree 1 file changed +11
-5
lines changed
1 file changed +11
-5
lines changed Original file line number Diff line number Diff line change 5
5
6
6
if __name__ == "__main__" :
7
7
if len (sys .argv ) < 2 :
8
- print ("You must specify a command (freeze/unfreeze)" )
8
+ print ("You must specify a command (freeze/unfreeze)" , file = sys . stderr )
9
9
sys .exit (1 )
10
10
11
11
broker_address = "tcp://127.0.0.1:9658"
12
12
13
+ if sys .argv [1 ] == "freeze" :
14
+ command = b"freeze"
15
+ elif sys .argv [1 ] == "unfreeze" :
16
+ command = b"unfreeze"
17
+ else :
18
+ print ("Unknown command" , file = sys .stderr )
19
+ sys .exit (1 )
20
+
13
21
context = zmq .Context ()
14
22
socket = context .socket (zmq .DEALER )
23
+ socket .setsockopt (zmq .LINGER , - 1 )
15
24
socket .connect (broker_address )
16
25
17
- if sys .argv [1 ] == "freeze" :
18
- socket .send_multipart ([b"freeze" ])
19
- elif sys .argv [1 ] == "unfreeze" :
20
- socket .send_multipart ([b"unfreeze" ])
26
+ socket .send_multipart ([command ])
21
27
22
28
socket .disconnect (broker_address )
23
29
You can’t perform that action at this time.
0 commit comments