Java InMemoryStorage类代码示例

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Test
public void acceptsTrace() {
  storage = new InMemoryStorage();
  StorageConsumer storageConsumer = new StorageConsumer() {
    @Override Logger log() {
      return logger;
    }

    @Override protected StorageComponent tryCompute() {
      return storage;
    }
  };

  storageConsumer.accept(TestObjects.TRACE);
  assertThat(logger.lines())
      .extracting("level", "text")
      .containsExactly(tuple(LogLevel.DebugLevel, "Wrote 3 spans"));

  assertThat(storage.spanStore().getRawTrace(
      TestObjects.TRACE.get(0).traceIdHigh,
      TestObjects.TRACE.get(0).traceId
  )).isEqualTo(TestObjects.TRACE);
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Test(timeout = 1000L)
public void get_memoizes() throws InterruptedException {
  AtomicInteger provisionCount = new AtomicInteger();

  StorageConsumer storageConsumer = new StorageConsumer() {
    @Override protected StorageComponent tryCompute() {
      provisionCount.incrementAndGet();
      return new InMemoryStorage();
    }
  };

  int getCount = 1000;
  CountDownLatch latch = new CountDownLatch(getCount);
  ExecutorService exec = Executors.newFixedThreadPool(10);
  for (int i = 0; i < getCount; i++) {
    exec.execute(() -> {
      storageConsumer.get();
      latch.countDown();
    });
  }
  latch.await();
  exec.shutdown();
  exec.awaitTermination(1, TimeUnit.SECONDS);

  assertThat(provisionCount.get()).isEqualTo(1);
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Before
public void setup() {
  store = new InMemoryStorage();
  metrics = new InMemoryCollectorMetrics();

  collector = new SQSCollector.Builder()
      .queueUrl(sqsRule.queueUrl())
      .parallelism(2)
      .waitTimeSeconds(1) // using short wait time to make test teardown faster
      .endpointConfiguration(new EndpointConfiguration(sqsRule.queueUrl(), "us-east-1"))
      .credentialsProvider(new AWSStaticCredentialsProvider(new BasicAWSCredentials("x", "x")))
      .metrics(metrics)
      .sampler(CollectorSampler.ALWAYS_SAMPLE)
      .storage(store)
      .build()
      .start();
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
TestLazyRegisterEventProcessorFactoryWithHost() {
  super(EventHubCollector.newBuilder()
      .connectionString(
          "endpoint=sb://someurl.net;SharedAccessKeyName=dumbo;SharedAccessKey=uius7y8ewychsih")
      .storageConnectionString("UseDevelopmentStorage=true")
      .storage(new InMemoryStorage()));
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Test
public void check_failsWhenNotStarted() {
  try (ScribeCollector scribe =
           ScribeCollector.builder().storage(new InMemoryStorage()).port(12345).build()) {

    CheckResult result = scribe.check();
    assertThat(result.ok).isFalse();
    assertThat(result.exception)
        .isInstanceOf(IllegalStateException.class);

    scribe.start();
    assertThat(scribe.check().ok).isTrue();
  }
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Test
public void start_failsWhenCantBindPort() {
  thrown.expect(ChannelException.class);
  thrown.expectMessage("Failed to bind to: 0.0.0.0/0.0.0.0:12345");

  ScribeCollector.Builder builder =
      ScribeCollector.builder().storage(new InMemoryStorage()).port(12345);

  try (ScribeCollector first = builder.build().start()) {
    try (ScribeCollector samePort = builder.build().start()) {
    }
  }
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
/**
 * The current implementation does not include dependency aggregation. It includes retrieval of
 * pre-aggregated links.
 *
 * <p>This uses {@link InMemorySpanStore} to prepare links and {@link CassandraDependenciesWriter}
 * to store them.
 *
 * <p>Note: The zipkin-dependencies-spark doesn't use any of these classes: it reads and writes to
 * the keyspace directly.
 */
@Override
public void processDependencies(List<Span> spans) {
  InMemoryStorage mem = new InMemoryStorage();
  mem.spanConsumer().accept(spans);
  List<DependencyLink> links = mem.spanStore().getDependencies(TODAY + DAY, null);

  // This gets or derives a timestamp from the spans
  long midnight = midnightUTC(MergeById.apply(spans).get(0).timestamp / 1000);
  new CassandraDependenciesWriter(storage.session.get()).write(links, midnight);
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
/**
 * The current implementation does not include dependency aggregation. It includes retrieval of
 * pre-aggregated links.
 *
 * <p>This uses {@link InMemorySpanStore} to prepare links and {@link #writeDependencyLinks(List,
 * long)}} to store them.
 */
@Override
public void processDependencies(List<Span> spans) {
  InMemoryStorage mem = new InMemoryStorage();
  mem.spanConsumer().accept(spans);
  List<DependencyLink> links = mem.spanStore().getDependencies(TODAY + DAY, null);

  // This gets or derives a timestamp from the spans
  long midnight = midnightUTC(MergeById.apply(spans).get(0).timestamp / 1000);
  writeDependencyLinks(links, midnight);
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Test
public void unsampledSpansArentStored() {
  collector = Collector.builder(Collector.class)
      .sampler(CollectorSampler.create(0f))
      .storage(new InMemoryStorage()).build();

  collector.accept(asList(span(Long.MIN_VALUE)), NOOP);

  assertThat(collector.storage.spanStore().getServiceNames()).isEmpty();
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Bean
StorageComponent storage() {
    return new InMemoryStorage();
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
private EventHubCollector.Builder getBuilder(){
    return EventHubCollector.builder()
            .storage(new InMemoryStorage());

}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Bean
StorageComponent storage() {
  return new InMemoryStorage();
}
 

import zipkin.storage.InMemoryStorage; //导入依赖的package包/类
@Bean StorageComponent storage() {
  return new InMemoryStorage();
}