From f5f12e1560c629da318a1e58ce251595aab4ff51 Mon Sep 17 00:00:00 2001
From: Niko Storni <niko@lbry.io>
Date: Tue, 27 Oct 2020 19:50:10 +0100
Subject: [PATCH] add support for multiple queues

add support for blockchain.db pruning via new status
---
 main.go               | 5 +++--
 manager/manager.go    | 7 +++++--
 manager/s3_storage.go | 4 +++-
 shared/shared.go      | 5 ++++-
 4 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/main.go b/main.go
index 13080eb..7ac017b 100644
--- a/main.go
+++ b/main.go
@@ -53,7 +53,8 @@ func main() {
 	cmd.Flags().BoolVar(&cliFlags.UpgradeMetadata, "upgrade-metadata", false, "Upgrade videos if they're on the old metadata version")
 	cmd.Flags().BoolVar(&cliFlags.DisableTransfers, "no-transfers", false, "Skips the transferring process of videos, channels and supports")
 	cmd.Flags().BoolVar(&cliFlags.QuickSync, "quick", false, "Look up only the last 50 videos from youtube")
-	cmd.Flags().StringVar(&cliFlags.SyncStatus, "status", "", "Specify which queue to pull from. Overrides --update")
+	cmd.Flags().StringVar(&cliFlags.Status, "status", "", "Specify which queue to pull from. Overrides --update")
+	cmd.Flags().StringVar(&cliFlags.SecondaryStatus, "status2", "", "Specify which secondary queue to pull from.")
 	cmd.Flags().StringVar(&cliFlags.ChannelID, "channelID", "", "If specified, only this channel will be synced.")
 	cmd.Flags().Int64Var(&cliFlags.SyncFrom, "after", time.Unix(0, 0).Unix(), "Specify from when to pull jobs [Unix time](Default: 0)")
 	cmd.Flags().Int64Var(&cliFlags.SyncUntil, "before", time.Now().AddDate(1, 0, 0).Unix(), "Specify until when to pull jobs [Unix time](Default: current Unix time)")
@@ -87,7 +88,7 @@ func ytSync(cmd *cobra.Command, args []string) {
 		util.InitSlack(os.Getenv("SLACK_TOKEN"), os.Getenv("SLACK_CHANNEL"), hostname)
 	}
 
-	if cliFlags.SyncStatus != "" && !util.InSlice(cliFlags.SyncStatus, shared.SyncStatuses) {
+	if cliFlags.Status != "" && !util.InSlice(cliFlags.Status, shared.SyncStatuses) {
 		log.Errorf("status must be one of the following: %v\n", shared.SyncStatuses)
 		return
 	}
diff --git a/manager/manager.go b/manager/manager.go
index 8a09b77..9abe824 100644
--- a/manager/manager.go
+++ b/manager/manager.go
@@ -76,13 +76,16 @@ func (s *SyncManager) Start() error {
 			shouldInterruptLoop = true
 		} else {
 			var queuesToSync []string
-			if s.CliFlags.SyncStatus != "" {
-				queuesToSync = append(queuesToSync, s.CliFlags.SyncStatus)
+			if s.CliFlags.Status != "" {
+				queuesToSync = append(queuesToSync, s.CliFlags.Status)
 			} else if s.CliFlags.SyncUpdate {
 				queuesToSync = append(queuesToSync, shared.StatusSyncing, shared.StatusSynced)
 			} else {
 				queuesToSync = append(queuesToSync, shared.StatusSyncing, shared.StatusQueued)
 			}
+			if s.CliFlags.SecondaryStatus != "" {
+				queuesToSync = append(queuesToSync, s.CliFlags.SecondaryStatus)
+			}
 		queues:
 			for _, q := range queuesToSync {
 				channels, err := s.ApiConfig.FetchChannels(q, &s.CliFlags)
diff --git a/manager/s3_storage.go b/manager/s3_storage.go
index 8b78c89..a21d233 100644
--- a/manager/s3_storage.go
+++ b/manager/s3_storage.go
@@ -97,7 +97,9 @@ func (s *Sync) downloadBlockchainDB() error {
 			return errors.Err(err)
 		}
 	}
-
+	if s.DbChannelData.WipeDB {
+		return nil
+	}
 	downloader, err := s.getS3Downloader()
 	if err != nil {
 		return errors.Err(err)
diff --git a/shared/shared.go b/shared/shared.go
index 28b8077..016d6d2 100644
--- a/shared/shared.go
+++ b/shared/shared.go
@@ -25,6 +25,7 @@ type YoutubeChannel struct {
 	LengthLimit        int    `json:"length_limit"`
 	SizeLimit          int    `json:"size_limit"`
 	LastUploadedVideo  string `json:"last_uploaded_video"`
+	WipeDB             bool   `json:"wipe_db"`
 }
 
 var NeverRetryFailures = []string{
@@ -53,7 +54,8 @@ type SyncFlags struct {
 	MaxTries                int
 	Refill                  int
 	Limit                   int
-	SyncStatus              string
+	Status                  string
+	SecondaryStatus         string
 	ChannelID               string
 	SyncFrom                int64
 	SyncUntil               int64
@@ -86,6 +88,7 @@ const (
 	StatusPendingUpgrade = "pendingupgrade" // in sync queue. will be synced soon
 	StatusSyncing        = "syncing"        // syncing now
 	StatusSynced         = "synced"         // done
+	StatusWipeDb         = "pendingdbwipe"  // in sync queue. lbryum database will be pruned
 	StatusFailed         = "failed"
 	StatusFinalized      = "finalized" // no more changes allowed
 	StatusAbandoned      = "abandoned" // deleted on youtube or banned