Skip to content

Commit

Permalink
Make stress test more realistic (open-telemetry#1531)
Browse files Browse the repository at this point in the history
  • Loading branch information
cijothomas committed Feb 14, 2024
1 parent 054b0c8 commit f80d754
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 14 deletions.
12 changes: 1 addition & 11 deletions stress/src/logs.rs
Original file line number Diff line number Diff line change
@@ -1,9 +1,5 @@
use opentelemetry::KeyValue;
use opentelemetry_appender_tracing::layer;
use opentelemetry_sdk::{
logs::{Config, LogProcessor, LoggerProvider},
Resource,
};
use opentelemetry_sdk::logs::{LogProcessor, LoggerProvider};
use tracing::error;
use tracing_subscriber::{prelude::*, Layer};

Expand Down Expand Up @@ -65,12 +61,6 @@ impl LogProcessor for NoOpLogProcessor {
fn main() {
// LoggerProvider with a no-op processor.
let provider: LoggerProvider = LoggerProvider::builder()
.with_config(
Config::default().with_resource(Resource::new(vec![KeyValue::new(
"service.name",
"log-appender-tracing-example",
)])),
)
.with_log_processor(NoOpLogProcessor {})
.build();

Expand Down
31 changes: 28 additions & 3 deletions stress/src/traces.rs
Original file line number Diff line number Diff line change
@@ -1,19 +1,44 @@
use lazy_static::lazy_static;
use opentelemetry::{
trace::{Span, SpanBuilder, Tracer, TracerProvider as _},
KeyValue,
trace::{Span, SpanBuilder, TraceResult, Tracer, TracerProvider as _},
Context, KeyValue,
};
use opentelemetry_sdk::{
export::trace::SpanData,
trace::{self as sdktrace, SpanProcessor},
};
use opentelemetry_sdk::trace as sdktrace;

mod throughput;

lazy_static! {
static ref PROVIDER: sdktrace::TracerProvider = sdktrace::TracerProvider::builder()
.with_config(sdktrace::config().with_sampler(sdktrace::Sampler::AlwaysOn))
.with_span_processor(NoOpSpanProcessor {})
.build();
static ref TRACER: sdktrace::Tracer = PROVIDER.tracer("stress");
}

#[derive(Debug)]
pub struct NoOpSpanProcessor;

impl SpanProcessor for NoOpSpanProcessor {
fn on_start(&self, _span: &mut opentelemetry_sdk::trace::Span, _cx: &Context) {
// No-op
}

fn on_end(&self, _span: SpanData) {
// No-op
}

fn force_flush(&self) -> TraceResult<()> {
Ok(())
}

fn shutdown(&mut self) -> TraceResult<()> {
Ok(())
}
}

fn main() {
throughput::test_throughput(test_span);
}
Expand Down

0 comments on commit f80d754

Please sign in to comment.