Save all data available from the ws stream.

main.go

@@ -22,6 +22,37 @@ func main() {
 	log.Printf("Config: symbol=%s, hot_retention=%dh, feature_retention=%dd",
 		cfg.Symbol, cfg.HotRetentionHours, cfg.FeatureRetentionDays)
 
+    if len(os.Args) > 1 {
+
+        switch os.Args[1] {
+
+        case "recover":
+            sm, err := NewStorageManager(cfg)
+            if err != nil {
+                log.Fatal(err)
+            }
+
+            if err := sm.StartupRecovery(); err != nil {
+                log.Fatal(err)
+            }
+
+            log.Println("Recovery completed.")
+            return
+
+        case "maintain":
+            sm, err := NewStorageManager(cfg)
+            if err != nil {
+                log.Fatal(err)
+            }
+
+            sm.RunHourlyMaintenance()
+
+            log.Println("Maintenance completed.")
+            return
+        }
+    }
+
+
 	// Initialize storage (creates dirs, databases, tables)
 	sm, err := NewStorageManager(cfg)
 	if err != nil {