Search in sources :

Example 1 with LogBufferEvent

use of io.cdap.cdap.logging.logbuffer.LogBufferEvent in project cdap by caskdata.

the class LogBufferProcessorPipelineTest method testSingleAppender.

@Test
public void testSingleAppender() throws Exception {
    LoggerContext loggerContext = LogPipelineTestUtil.createLoggerContext("WARN", ImmutableMap.of("test.logger", "INFO"), MockAppender.class.getName());
    final MockAppender appender = LogPipelineTestUtil.getAppender(loggerContext.getLogger(Logger.ROOT_LOGGER_NAME), "Test", MockAppender.class);
    MockCheckpointManager checkpointManager = new MockCheckpointManager();
    LogBufferPipelineConfig config = new LogBufferPipelineConfig(1024L, 300L, 500L, 4);
    loggerContext.start();
    LogBufferProcessorPipeline pipeline = new LogBufferProcessorPipeline(new LogProcessorPipelineContext(CConfiguration.create(), "test", loggerContext, NO_OP_METRICS_CONTEXT, 0), config, checkpointManager, 0);
    // start the pipeline
    pipeline.startAndWait();
    // start thread to write to incomingEventQueue
    List<ILoggingEvent> events = getLoggingEvents();
    AtomicInteger i = new AtomicInteger(0);
    List<LogBufferEvent> bufferEvents = events.stream().map(event -> {
        LogBufferEvent lbe = new LogBufferEvent(event, serializer.toBytes(event).length, new LogBufferFileOffset(0, i.get()));
        i.incrementAndGet();
        return lbe;
    }).collect(Collectors.toList());
    // start a thread to send log buffer events to pipeline
    ExecutorService executorService = Executors.newSingleThreadExecutor();
    executorService.execute(() -> {
        for (int count = 0; count < 40; count++) {
            pipeline.processLogEvents(bufferEvents.iterator());
            try {
                Thread.sleep(100);
            } catch (InterruptedException e) {
            // should not happen
            }
        }
    });
    // wait for pipeline to append all the logs to appender. The DEBUG message should get filtered out.
    Tasks.waitFor(200, () -> appender.getEvents().size(), 60, TimeUnit.SECONDS, 100, TimeUnit.MILLISECONDS);
    executorService.shutdown();
    pipeline.stopAndWait();
    loggerContext.stop();
}
Also used : ILoggingEvent(ch.qos.logback.classic.spi.ILoggingEvent) LogBufferFileOffset(io.cdap.cdap.logging.logbuffer.LogBufferFileOffset) LoggerContext(ch.qos.logback.classic.LoggerContext) CheckpointManager(io.cdap.cdap.logging.meta.CheckpointManager) MockAppender(io.cdap.cdap.logging.pipeline.MockAppender) LoggingEventSerializer(io.cdap.cdap.logging.serialize.LoggingEventSerializer) ImmutableList(com.google.common.collect.ImmutableList) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) LogProcessorPipelineContext(io.cdap.cdap.logging.pipeline.LogProcessorPipelineContext) Map(java.util.Map) ExecutorService(java.util.concurrent.ExecutorService) NoopMetricsContext(io.cdap.cdap.api.metrics.NoopMetricsContext) Tasks(io.cdap.cdap.common.utils.Tasks) ImmutableMap(com.google.common.collect.ImmutableMap) Checkpoint(io.cdap.cdap.logging.meta.Checkpoint) Set(java.util.Set) Test(org.junit.Test) IOException(java.io.IOException) Collectors(java.util.stream.Collectors) Executors(java.util.concurrent.Executors) TimeUnit(java.util.concurrent.TimeUnit) Level(ch.qos.logback.classic.Level) MetricsContext(io.cdap.cdap.api.metrics.MetricsContext) List(java.util.List) CConfiguration(io.cdap.cdap.common.conf.CConfiguration) Logger(ch.qos.logback.classic.Logger) LogBufferEvent(io.cdap.cdap.logging.logbuffer.LogBufferEvent) Collections(java.util.Collections) LogPipelineTestUtil(io.cdap.cdap.logging.pipeline.LogPipelineTestUtil) LogProcessorPipelineContext(io.cdap.cdap.logging.pipeline.LogProcessorPipelineContext) ILoggingEvent(ch.qos.logback.classic.spi.ILoggingEvent) LoggerContext(ch.qos.logback.classic.LoggerContext) Checkpoint(io.cdap.cdap.logging.meta.Checkpoint) MockAppender(io.cdap.cdap.logging.pipeline.MockAppender) LogBufferEvent(io.cdap.cdap.logging.logbuffer.LogBufferEvent) AtomicInteger(java.util.concurrent.atomic.AtomicInteger) ExecutorService(java.util.concurrent.ExecutorService) LogBufferFileOffset(io.cdap.cdap.logging.logbuffer.LogBufferFileOffset) Test(org.junit.Test)

Example 2 with LogBufferEvent

use of io.cdap.cdap.logging.logbuffer.LogBufferEvent in project cdap by caskdata.

the class LogBufferReaderTest method verifyEvents.

private void verifyEvents(List<LogBufferEvent> logBufferEvents, LogBufferReader reader, Iterator<LogBufferEvent> iterator) throws IOException {
    while (reader.readEvents(logBufferEvents) > 0) {
        for (LogBufferEvent event : logBufferEvents) {
            LogBufferEvent next = iterator.next();
            Assert.assertEquals(next.getOffset(), event.getOffset());
            Assert.assertEquals(next.getEventSize(), event.getEventSize());
            Assert.assertEquals(next.getLogEvent().getMessage(), event.getLogEvent().getMessage());
        }
        logBufferEvents.clear();
    }
    Assert.assertFalse(iterator.hasNext());
}
Also used : LogBufferEvent(io.cdap.cdap.logging.logbuffer.LogBufferEvent)

Example 3 with LogBufferEvent

use of io.cdap.cdap.logging.logbuffer.LogBufferEvent in project cdap by caskdata.

the class LogBufferReaderTest method testLogReader.

@Test
public void testLogReader() throws Exception {
    String absolutePath = TMP_FOLDER.newFolder().getAbsolutePath();
    LogBufferWriter writer = new LogBufferWriter(absolutePath, 250, () -> {
    });
    ImmutableList<byte[]> events = getLoggingEvents();
    Iterable<LogBufferEvent> writtenEvents = writer.write(events.iterator());
    writer.close();
    List<LogBufferEvent> logBufferEvents = new LinkedList<>();
    // read from start positions, tests case where no checkpoints are persisted
    LogBufferReader reader = new LogBufferReader(absolutePath, 2, 3, -1, -1);
    Iterator<LogBufferEvent> iterator = writtenEvents.iterator();
    verifyEvents(logBufferEvents, reader, iterator);
    reader.close();
    // this should skip first and second event, this is because log buffer offsets are offset for event that is
    // already stored. so in this case, skip first event and skip second event as second event is the last stored event
    reader = new LogBufferReader(absolutePath, 2, 3, 0, 145);
    iterator = writtenEvents.iterator();
    iterator.next();
    iterator.next();
    verifyEvents(logBufferEvents, reader, iterator);
    reader.close();
}
Also used : LogBufferReader(io.cdap.cdap.logging.logbuffer.recover.LogBufferReader) LogBufferEvent(io.cdap.cdap.logging.logbuffer.LogBufferEvent) LogBufferWriter(io.cdap.cdap.logging.logbuffer.LogBufferWriter) LinkedList(java.util.LinkedList) Test(org.junit.Test)

Aggregations

LogBufferEvent (io.cdap.cdap.logging.logbuffer.LogBufferEvent)3 Test (org.junit.Test)2 Level (ch.qos.logback.classic.Level)1 Logger (ch.qos.logback.classic.Logger)1 LoggerContext (ch.qos.logback.classic.LoggerContext)1 ILoggingEvent (ch.qos.logback.classic.spi.ILoggingEvent)1 ImmutableList (com.google.common.collect.ImmutableList)1 ImmutableMap (com.google.common.collect.ImmutableMap)1 MetricsContext (io.cdap.cdap.api.metrics.MetricsContext)1 NoopMetricsContext (io.cdap.cdap.api.metrics.NoopMetricsContext)1 CConfiguration (io.cdap.cdap.common.conf.CConfiguration)1 Tasks (io.cdap.cdap.common.utils.Tasks)1 LogBufferFileOffset (io.cdap.cdap.logging.logbuffer.LogBufferFileOffset)1 LogBufferWriter (io.cdap.cdap.logging.logbuffer.LogBufferWriter)1 LogBufferReader (io.cdap.cdap.logging.logbuffer.recover.LogBufferReader)1 Checkpoint (io.cdap.cdap.logging.meta.Checkpoint)1 CheckpointManager (io.cdap.cdap.logging.meta.CheckpointManager)1 LogPipelineTestUtil (io.cdap.cdap.logging.pipeline.LogPipelineTestUtil)1 LogProcessorPipelineContext (io.cdap.cdap.logging.pipeline.LogProcessorPipelineContext)1 MockAppender (io.cdap.cdap.logging.pipeline.MockAppender)1