Skip to content

Commit 25aa401

Browse files
authored
fix: support adhoc (#109)
1 parent e8b50b2 commit 25aa401

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

pyroscope_ffi/python/lib/src/lib.rs

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,11 +67,16 @@ pub extern "C" fn initialize_agent(
6767
.to_string();
6868

6969
// server_address
70-
let server_address = unsafe { CStr::from_ptr(server_address) }
70+
let mut server_address = unsafe { CStr::from_ptr(server_address) }
7171
.to_str()
7272
.unwrap()
7373
.to_string();
7474

75+
let adhoc_server_address = std::env::var("PYROSCOPE_ADHOC_SERVER_ADDRESS");
76+
if let Ok(adhoc_server_address) = adhoc_server_address {
77+
server_address = adhoc_server_address
78+
}
79+
7580
let auth_token = unsafe { CStr::from_ptr(auth_token) }
7681
.to_str()
7782
.unwrap()

pyroscope_ffi/ruby/ext/rbspy/src/lib.rs

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -134,11 +134,16 @@ pub extern "C" fn initialize_agent(
134134
.unwrap()
135135
.to_string();
136136

137-
let server_address = unsafe { CStr::from_ptr(server_address) }
137+
let mut server_address = unsafe { CStr::from_ptr(server_address) }
138138
.to_str()
139139
.unwrap()
140140
.to_string();
141141

142+
let adhoc_server_address = std::env::var("PYROSCOPE_ADHOC_SERVER_ADDRESS");
143+
if let Ok(adhoc_server_address) = adhoc_server_address {
144+
server_address = adhoc_server_address
145+
}
146+
142147
let auth_token = unsafe { CStr::from_ptr(auth_token) }
143148
.to_str()
144149
.unwrap()

0 commit comments

Comments
 (0)