示例#1
0
def run(columnFamily):
    """
    Loop accepting input from stdin and writing it to the log server
    """

    scribbleWriter = scribble_lib.scribble_writer()
    try:
        while True:
            logMessage = sys.stdin.readline().rstrip()
            if logMessage:
                write_to_server(scribbleWriter, logMessage, columnFamily)
    except KeyboardInterrupt:
        return
示例#2
0
    def test_basic_buffering(self):
        """Test that basic buffer works"""

        baseLogMessage = "This is log record {0}"

        # Override the max buffer size so that we know when it should flush
        self.server.maxLogBufferSize = 5

        scribbleWriter = scribble_lib.scribble_writer()

        for idx in range(0, 4):
            scribble_client.write_to_server(scribbleWriter, baseLogMessage.format(idx),
                                            "Users")

            self.server.do_work()

        # There should be no buffered output yet
        self.assertEqual(0, len(self.bufferedOutput),
                            "Premature dumping of log buffer")

        scribble_client.write_to_server(scribbleWriter, baseLogMessage.format(4), "Users")

        # Manually 'run' the server for a bit
        [self.server.do_work() for i in range(15)]

        # Shutdown the server
        self.server.shutdown()

        rowCount = 0

        # Count the rows
        for colFam in self.bufferedOutput:
            for col in self.bufferedOutput[colFam].values():
                rowCount += len(col.values())

        # There should be buffered output now
        self.assertTrue(rowCount > 0,
                        "Buffered log data not dumped as expected")

        self.assertEqual(rowCount, 5,
                         "The expected amount of log data was not dumped")