File tree Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Original file line number Diff line number Diff 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 ( )
Original file line number Diff line number Diff 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 ( )
You can’t perform that action at this time.
0 commit comments