valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-08 17:33:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Allow passing in of header

Browse Source
This commit is contained in:
Thomas Jackson 2015-03-07 08:02:28 -08:00
parent 0b035341a5
commit 2336b338c5
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
salt/transport/tcp.py
View File

@ -49,9 +49,13 @@ log = logging.getLogger(__name__)
import msgpack
# TODO: put in some lib?
def frame_msg(msg):
header = msgpack.dumps({'msgLen': len(msg)})
return '{0} {1}{2}'.format(len(header), header, msg)
def frame_msg(msg, header=None):
if header is None:
header = {}
header['msgLen'] = len(msg)
header_packed = msgpack.dumps(header)
return '{0} {1}{2}'.format(len(header_packed), header_packed, msg)
def socket_frame_recv(s, recv_size=4096):