45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
import uuid
|
|
import struct
|
|
from libp2p.pubsub.pb import rpc_pb2
|
|
|
|
|
|
def message_id_generator(start_val):
|
|
"""
|
|
Generate a unique message id
|
|
:param start_val: value to start generating messages at
|
|
:return: message id
|
|
"""
|
|
val = start_val
|
|
def generator():
|
|
# Allow manipulation of val within closure
|
|
nonlocal val
|
|
|
|
# Increment id
|
|
val += 1
|
|
|
|
# Convert val to big endian
|
|
return struct.pack('>I', val)
|
|
|
|
return generator
|
|
|
|
def generate_RPC_packet(origin_id, topics, msg_content, msg_id):
|
|
"""
|
|
Generate RPC packet to send over wire
|
|
:param origin_id: peer id of the message origin
|
|
:param topics: list of topics
|
|
:param msg_content: string of content in data
|
|
:param msg_id: seqno for the message
|
|
"""
|
|
packet = rpc_pb2.RPC()
|
|
message = rpc_pb2.Message(
|
|
from_id=origin_id.encode('utf-8'),
|
|
seqno=msg_id,
|
|
data=msg_content.encode('utf-8'),
|
|
)
|
|
|
|
for topic in topics:
|
|
message.topicIDs.extend([topic.encode('utf-8')])
|
|
|
|
packet.publish.extend([message])
|
|
return packet
|