Skip to content

Commit db79b24

Browse files
committed
set an indefinite linger period for brokerctl
1 parent cc749e7 commit db79b24

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

brokerctl/brokerctl.py

+11-5
Original file line numberDiff line numberDiff line change
@@ -5,19 +5,25 @@
55

66
if __name__ == "__main__":
77
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)
99
sys.exit(1)
1010

1111
broker_address = "tcp://127.0.0.1:9658"
1212

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+
1321
context = zmq.Context()
1422
socket = context.socket(zmq.DEALER)
23+
socket.setsockopt(zmq.LINGER, -1)
1524
socket.connect(broker_address)
1625

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])
2127

2228
socket.disconnect(broker_address)
2329

0 commit comments

Comments
 (0)