Skip to content

Commit f5eaf69

Browse files
committed
Various cleanup
1 parent 54df6c4 commit f5eaf69

File tree

3 files changed

+45
-54
lines changed

3 files changed

+45
-54
lines changed

crates/sdk-core-c-bridge/src/runtime.rs

Lines changed: 31 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,6 @@ impl Runtime {
203203
// Build telemetry options
204204
let mut log_forwarder = None;
205205
let telemetry_options = if let Some(v) = unsafe { options.telemetry.as_ref() } {
206-
// Prepare values
207206
let (attach_service_name, metric_prefix) =
208207
if let Some(v) = unsafe { v.metrics.as_ref() } {
209208
(v.attach_service_name, v.metric_prefix.to_option_string())
@@ -229,7 +228,6 @@ impl Runtime {
229228
}
230229
});
231230

232-
// Build with all values
233231
CoreTelemetryOptions::builder()
234232
.attach_service_name(attach_service_name)
235233
.maybe_metric_prefix(metric_prefix)
@@ -239,17 +237,15 @@ impl Runtime {
239237
CoreTelemetryOptions::default()
240238
};
241239

242-
let heartbeat_interval = if options.worker_heartbeat_interval_millis == 0 {
243-
None
244-
} else {
245-
Some(Duration::from_millis(
246-
options.worker_heartbeat_interval_millis,
247-
))
248-
};
249-
250240
let core_runtime_options = CoreRuntimeOptions::builder()
251241
.telemetry_options(telemetry_options)
252-
.maybe_heartbeat_interval(heartbeat_interval)
242+
.maybe_heartbeat_interval(if options.worker_heartbeat_interval_millis == 0 {
243+
None
244+
} else {
245+
Some(Duration::from_millis(
246+
options.worker_heartbeat_interval_millis,
247+
))
248+
})
253249
.build()
254250
.map_err(|e| anyhow::anyhow!(e))?;
255251

@@ -392,29 +388,30 @@ fn create_meter(
392388
));
393389
}
394390
// Build OTel exporter
395-
let metric_periodicity = if otel_options.metric_periodicity_millis > 0 {
396-
Some(Duration::from_millis(
397-
otel_options.metric_periodicity_millis.into(),
398-
))
399-
} else {
400-
None
401-
};
402-
let build = OtelCollectorOptions::builder()
403-
.url(Url::parse(otel_options.url.to_str())?)
404-
.headers(otel_options.headers.to_string_map_on_newlines())
405-
.metric_temporality(match otel_options.metric_temporality {
406-
OpenTelemetryMetricTemporality::Cumulative => MetricTemporality::Cumulative,
407-
OpenTelemetryMetricTemporality::Delta => MetricTemporality::Delta,
408-
})
409-
.global_tags(options.global_tags.to_string_map_on_newlines())
410-
.use_seconds_for_durations(otel_options.durations_as_seconds)
411-
.histogram_bucket_overrides(HistogramBucketOverrides {
412-
overrides: parse_histogram_bucket_overrides(
413-
&otel_options.histogram_bucket_overrides,
414-
)?,
415-
})
416-
.maybe_metric_periodicity(metric_periodicity);
417-
Ok(Arc::new(build_otlp_metric_exporter(build.build())?))
391+
Ok(Arc::new(build_otlp_metric_exporter(
392+
OtelCollectorOptions::builder()
393+
.url(Url::parse(otel_options.url.to_str())?)
394+
.headers(otel_options.headers.to_string_map_on_newlines())
395+
.metric_temporality(match otel_options.metric_temporality {
396+
OpenTelemetryMetricTemporality::Cumulative => MetricTemporality::Cumulative,
397+
OpenTelemetryMetricTemporality::Delta => MetricTemporality::Delta,
398+
})
399+
.global_tags(options.global_tags.to_string_map_on_newlines())
400+
.use_seconds_for_durations(otel_options.durations_as_seconds)
401+
.histogram_bucket_overrides(HistogramBucketOverrides {
402+
overrides: parse_histogram_bucket_overrides(
403+
&otel_options.histogram_bucket_overrides,
404+
)?,
405+
})
406+
.maybe_metric_periodicity(if otel_options.metric_periodicity_millis > 0 {
407+
Some(Duration::from_millis(
408+
otel_options.metric_periodicity_millis.into(),
409+
))
410+
} else {
411+
None
412+
})
413+
.build(),
414+
)?))
418415
} else if let Some(prom_options) = unsafe { options.prometheus.as_ref() } {
419416
if custom_meter.is_some() {
420417
return Err(anyhow::anyhow!(

crates/sdk-core-c-bridge/src/testing.rs

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -203,24 +203,23 @@ impl TryFrom<&DevServerOptions> for ephemeral_server::TemporalDevServerConfig {
203203

204204
fn try_from(options: &DevServerOptions) -> anyhow::Result<Self> {
205205
let test_server_options = unsafe { &*options.test_server };
206-
let ui_port = if options.ui_port != 0 && options.ui {
207-
Some(options.ui_port)
208-
} else {
209-
None
210-
};
211206
Ok(ephemeral_server::TemporalDevServerConfig::builder()
212207
.exe(test_server_options.exe())
213208
.namespace(options.namespace.to_string())
214209
.ip(options.ip.to_string())
215210
.ui(options.ui)
211+
.maybe_ui_port(if options.ui_port != 0 && options.ui {
212+
Some(options.ui_port)
213+
} else {
214+
None
215+
})
216216
.log((
217217
options.log_format.to_string(),
218218
options.log_level.to_string(),
219219
))
220220
.extra_args(test_server_options.extra_args())
221221
.maybe_port(test_server_options.port())
222222
.maybe_db_filename(options.database_filename.to_option_string())
223-
.maybe_ui_port(ui_port)
224223
.build())
225224
}
226225
}

crates/sdk-core-c-bridge/src/worker.rs

Lines changed: 9 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1163,7 +1163,7 @@ impl TryFrom<&WorkerOptions> for temporalio_sdk_core::WorkerConfig {
11631163

11641164
fn try_from(opt: &WorkerOptions) -> anyhow::Result<Self> {
11651165
let converted_tuner: temporalio_sdk_core::TunerHolder = (&opt.tuner).try_into()?;
1166-
let config = WorkerConfig::builder()
1166+
WorkerConfig::builder()
11671167
.namespace(opt.namespace.to_str())
11681168
.task_queue(opt.task_queue.to_str())
11691169
.versioning_strategy({
@@ -1271,8 +1271,7 @@ impl TryFrom<&WorkerOptions> for temporalio_sdk_core::WorkerConfig {
12711271
.collect::<Vec<_>>(),
12721272
)
12731273
.build()
1274-
.map_err(|e| anyhow::anyhow!(e))?;
1275-
Ok(config)
1274+
.map_err(|e| anyhow::anyhow!(e))
12761275
}
12771276
}
12781277

@@ -1322,21 +1321,17 @@ impl TryFrom<&TunerHolder> for temporalio_sdk_core::TunerHolder {
13221321
bail!("All resource-based slot suppliers must have the same ResourceBasedTunerOptions",);
13231322
}
13241323

1325-
let resource_based = first.map(|f| {
1326-
temporalio_sdk_core::ResourceBasedSlotsOptions::builder()
1327-
.target_mem_usage(f.target_memory_usage)
1328-
.target_cpu_usage(f.target_cpu_usage)
1329-
.build()
1330-
});
1331-
1332-
let builder = temporalio_sdk_core::TunerHolderOptions::builder()
1324+
temporalio_sdk_core::TunerHolderOptions::builder()
13331325
.workflow_slot_options(holder.workflow_slot_supplier.try_into()?)
13341326
.activity_slot_options(holder.activity_slot_supplier.try_into()?)
13351327
.local_activity_slot_options(holder.local_activity_slot_supplier.try_into()?)
13361328
.nexus_slot_options(holder.nexus_task_slot_supplier.try_into()?)
1337-
.maybe_resource_based_options(resource_based);
1338-
1339-
builder
1329+
.maybe_resource_based_options(first.map(|f| {
1330+
temporalio_sdk_core::ResourceBasedSlotsOptions::builder()
1331+
.target_mem_usage(f.target_memory_usage)
1332+
.target_cpu_usage(f.target_cpu_usage)
1333+
.build()
1334+
}))
13401335
.build()
13411336
.map_err(|e| anyhow::anyhow!("Failed building tuner holder options: {}", e))?
13421337
.build_tuner_holder()

0 commit comments

Comments
 (0)