Skip to content

Commit ba7a1df

Browse files
committed
Grouping config options
1 parent 31116f4 commit ba7a1df

2 files changed

Lines changed: 12 additions & 1 deletion

File tree

receptor/config.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,6 +144,16 @@ def __init__(self, args=None):
144144
value_type='int',
145145
hint='If specified, the node will ping all other known nodes in the mesh every N seconds. The default is -1, meaning no pings are sent.',
146146
)
147+
self.add_config_option(
148+
section='node',
149+
key='groups',
150+
short_option='-g',
151+
long_option='--group',
152+
default_value=[],
153+
value_type='list',
154+
listof='str',
155+
hint='Define membership in one or more groups to aid in message routing',
156+
)
147157
# Controller options
148158
self.add_config_option(
149159
section='controller',

receptor/protocol.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,8 @@ def send_handshake(self):
109109
"cmd": "HI",
110110
"id": self.receptor.node_id,
111111
"expire_time": time.time() + 10,
112-
"capabilities": self.receptor.work_manager.get_capabilities(),
112+
"meta": dict(capabilities=self.receptor.work_manager.get_capabilities(),
113+
groups=self.receptor.config.node_groups),
113114
}).encode("utf-8")
114115
self.transport.write(msg + DELIM)
115116

0 commit comments

Comments
 (0)