@@ -46,11 +46,13 @@ import (
4646 "go.opentelemetry.io/otel/trace"
4747 "go.opentelemetry.io/otel/trace/noop"
4848 "golang.org/x/sys/unix"
49+ "google.golang.org/grpc"
4950
5051 "github.com/parca-dev/parca-agent/analytics"
5152 "github.com/parca-dev/parca-agent/config"
5253 "github.com/parca-dev/parca-agent/flags"
5354 "github.com/parca-dev/parca-agent/reporter"
55+ "github.com/parca-dev/parca-agent/uploader"
5456)
5557
5658var (
@@ -136,6 +138,14 @@ func mainWithExitCode() flags.ExitCode {
136138 return code
137139 }
138140
141+ if f .OfflineMode .Upload {
142+ code , err := uploader .OfflineModeDoUpload (f )
143+ if err != nil {
144+ log .Errorf ("failed to upload offline mode logs: %v" , err )
145+ }
146+ return code
147+ }
148+
139149 reg := prometheus .NewRegistry ()
140150 reg .MustRegister (
141151 collectors .NewBuildInfoCollector (),
@@ -164,12 +174,17 @@ func mainWithExitCode() flags.ExitCode {
164174 }
165175 }
166176
167- grpcConn , err := f .RemoteStore .WaitGrpcEndpoint (ctx , reg , tp )
168- if err != nil {
169- log .Errorf ("failed to connect to server: %v" , err )
170- return flags .ExitFailure
177+ isOfflineMode := len (f .OfflineMode .StoragePath ) > 0
178+
179+ var grpcConn * grpc.ClientConn
180+ if ! isOfflineMode {
181+ grpcConn , err = f .RemoteStore .WaitGrpcEndpoint (ctx , reg , tp )
182+ if err != nil {
183+ log .Errorf ("failed to connect to server: %v" , err )
184+ return flags .ExitFailure
185+ }
186+ defer grpcConn .Close ()
171187 }
172- defer grpcConn .Close ()
173188
174189 presentCores , err := numcpus .GetPresent ()
175190 if err != nil {
@@ -291,17 +306,32 @@ func mainWithExitCode() flags.ExitCode {
291306 intervals := times .New (5 * time .Second , f .Profiling .Duration , f .Profiling .ProbabilisticInterval )
292307 times .StartRealtimeSync (mainCtx , f .ClockSyncInterval )
293308
309+ var client profilestoregrpc.ProfileStoreServiceClient
310+ var debuginfoClient debuginfogrpc.DebuginfoServiceClient
311+ if grpcConn != nil {
312+ client = profilestoregrpc .NewProfileStoreServiceClient (grpcConn )
313+ debuginfoClient = debuginfogrpc .NewDebuginfoServiceClient (grpcConn )
314+ }
315+
316+ var offlineModeConfig * reporter.OfflineModeConfig
317+ if isOfflineMode {
318+ offlineModeConfig = & reporter.OfflineModeConfig {
319+ StoragePath : f .OfflineMode .StoragePath ,
320+ RotationInterval : f .OfflineMode .RotationInterval ,
321+ }
322+ }
323+
294324 // Network operations to CA start here
295325 // Connect to the collection agent
296326 parcaReporter , err := reporter .New (
297327 memory .DefaultAllocator ,
298- profilestoregrpc . NewProfileStoreServiceClient ( grpcConn ) ,
299- debuginfogrpc . NewDebuginfoServiceClient ( grpcConn ) ,
328+ client ,
329+ debuginfoClient ,
300330 externalLabels ,
301331 f .Profiling .Duration ,
302332 f .Debuginfo .Strip ,
303333 f .Debuginfo .UploadMaxParallel ,
304- f .Debuginfo .UploadDisable ,
334+ f .Debuginfo .UploadDisable || isOfflineMode ,
305335 int64 (f .Profiling .CPUSamplingFrequency ),
306336 traceHandlerCacheSize ,
307337 f .Debuginfo .UploadQueueSize ,
@@ -310,6 +340,7 @@ func mainWithExitCode() flags.ExitCode {
310340 relabelConfigs ,
311341 buildInfo .VcsRevision ,
312342 reg ,
343+ offlineModeConfig ,
313344 )
314345 if err != nil {
315346 return flags .Failure ("Failed to start reporting: %v" , err )
@@ -413,8 +444,10 @@ func mainWithExitCode() flags.ExitCode {
413444
414445 log .Info ("Stop processing ..." )
415446 rep .Stop ()
416- if err := grpcConn .Close (); err != nil {
417- log .Fatalf ("Stopping connection of OTLP client client failed: %v" , err )
447+ if grpcConn != nil {
448+ if err := grpcConn .Close (); err != nil {
449+ log .Fatalf ("Stopping connection of OTLP client client failed: %v" , err )
450+ }
418451 }
419452
420453 log .Info ("Exiting ..." )
0 commit comments