Initialize Archive Storage For All in One And Query Service

Signed-off-by: Mahad Zaryab <mahadzaryab1@gmail.com>

cmd/all-in-one/main.go

@@ -56,6 +56,10 @@ func main() {
 	if err != nil {
 		log.Fatalf("Cannot initialize storage factory: %v", err)
 	}
+	archiveStorageFactory, err := storage.NewFactory(storage.FactoryConfigFromEnvAndCLI(os.Args, os.Stderr), true)
+	if err != nil {
+		log.Fatalf("Cannot initialize archive storage factory: %v", err)
+	}
 	samplingStrategyFactoryConfig, err := ss.FactoryConfigFromEnv()
 	if err != nil {
 		log.Fatalf("Cannot initialize sampling strategy factory config: %v", err)
@@ -164,7 +168,7 @@ by default uses only in-memory database.`,
 			}
 			// query
 			querySrv := startQuery(
-				svc, qOpts, qOpts.BuildQueryServiceOptions(storageFactory, logger),
+				svc, qOpts, qOpts.BuildQueryServiceOptions(archiveStorageFactory, logger),
 				spanReader, dependencyReader, metricsQueryService,
 				tm, telset,
 			)

cmd/query/main.go

@@ -44,6 +44,11 @@ func main() {
 		log.Fatalf("Cannot initialize storage factory: %v", err)
 	}
 
+	archiveStorageFactory, err := storage.NewFactory(storage.FactoryConfigFromEnvAndCLI(os.Args, os.Stderr), true)
+	if err != nil {
+		log.Fatalf("Cannot initialize archive storage factory: %v", err)
+	}
+
 	fc := metricsPlugin.FactoryConfigFromEnv()
 	metricsReaderFactory, err := metricsPlugin.NewFactory(fc)
 	if err != nil {
@@ -98,7 +103,7 @@ func main() {
 			if err != nil {
 				logger.Fatal("Failed to create metrics query service", zap.Error(err))
 			}
-			queryServiceOptions := queryOpts.BuildQueryServiceOptions(storageFactory, logger)
+			queryServiceOptions := queryOpts.BuildQueryServiceOptions(archiveStorageFactory, logger)
 			queryService := querysvc.NewQueryService(
 				spanReader,
 				dependencyReader,