Skip to content
Snippets Groups Projects
PythonClient.py 1.14 KiB
Newer Older
  • Learn to ignore specific revisions
  • #!/usr/bin/env python
    
    import sys
    
    sys.path.append('../../../target/gen-py')
    
    
    from packetstreamer import PacketStreamer
    from packetstreamer.ttypes import *
    
    from thrift import Thrift
    from thrift.transport import TSocket
    from thrift.transport import TTransport
    from thrift.protocol import TBinaryProtocol
    
    try:
    
        # Make socket
        transport = TSocket.TSocket('localhost', 9090)
    
        # Buffering is critical. Raw sockets are very slow
        transport = TTransport.TFramedTransport(transport)
    
        # Wrap in a protocol
        protocol = TBinaryProtocol.TBinaryProtocol(transport)
    
        # Create a client to use the protocol encoder
        client = PacketStreamer.Client(protocol)
    
        # Connect!
        transport.open()
    
        while 1:
            packets = client.getPackets("session1")
            print 'session1 packets num: %d' % (len(packets))
            count = 1
            for packet in packets:
                print "Packet %d: %s"% (count, packet)
                if "FilterTimeout" in packet:
                    sys.exit()
                count += 1 
    
        # Close!
        transport.close()
    
    except Thrift.TException, tx:
        print '%s' % (tx.message)
    
    except KeyboardInterrupt, e:
        print 'Bye-bye'