From 2024e58c150a2680eaf12d1f51ad0b2a1f804548 Mon Sep 17 00:00:00 2001
From: Bartek Nowotarski <bartek@nowotarski.info>
Date: Tue, 7 Jul 2020 14:18:34 +0200
Subject: [PATCH] services/horizon: Add `expingest trigger-state-rebuild`
 command (#2782)

This commit adds `horizon expingest trigger-state-rebuild` command which
sets ingest version in a DB to 0 and it makes Horizon rebuild the state.

Sometimes Horizon DB state becomes invalid, in most cases because of the
bugs during development. Usually to solve the issue we update the DB
manually but it's better to have a command that does this.
---
 services/horizon/cmd/ingest.go                | 24 ++++++++++++++++++-
 .../horizon/internal/db2/history/key_value.go |  4 ++--
 2 files changed, 25 insertions(+), 3 deletions(-)

diff --git a/services/horizon/cmd/ingest.go b/services/horizon/cmd/ingest.go
index db65c99f29..b744a1e118 100644
--- a/services/horizon/cmd/ingest.go
+++ b/services/horizon/cmd/ingest.go
@@ -9,6 +9,7 @@ import (
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 	"github.com/stellar/go/historyarchive"
+	"github.com/stellar/go/services/horizon/internal/db2/history"
 	"github.com/stellar/go/services/horizon/internal/expingest"
 	support "github.com/stellar/go/support/config"
 	"github.com/stellar/go/support/db"
@@ -207,6 +208,27 @@ var ingestStressTestCmd = &cobra.Command{
 	},
 }
 
+var ingestTriggerStateRebuildCmd = &cobra.Command{
+	Use:   "trigger-state-rebuild",
+	Short: "updates a database to trigger state rebuild, state will be rebuilt by a running Horizon instance, DO NOT RUN production DB, some endpoints will be unavailable until state is rebuilt",
+	Run: func(cmd *cobra.Command, args []string) {
+		initRootConfig()
+
+		horizonSession, err := db.Open("postgres", config.DatabaseURL)
+		if err != nil {
+			log.Fatalf("cannot open Horizon DB: %v", err)
+		}
+
+		historyQ := &history.Q{horizonSession}
+		err = historyQ.UpdateExpIngestVersion(0)
+		if err != nil {
+			log.Fatalf("cannot trigger state rebuild: %v", err)
+		}
+
+		log.Info("Triggered state rebuild")
+	},
+}
+
 func init() {
 	for _, co := range ingestVerifyRangeCmdOpts {
 		err := co.Init(ingestVerifyRangeCmd)
@@ -225,5 +247,5 @@ func init() {
 	viper.BindPFlags(ingestVerifyRangeCmd.PersistentFlags())
 
 	rootCmd.AddCommand(ingestCmd)
-	ingestCmd.AddCommand(ingestVerifyRangeCmd, ingestStressTestCmd)
+	ingestCmd.AddCommand(ingestVerifyRangeCmd, ingestStressTestCmd, ingestTriggerStateRebuildCmd)
 }
diff --git a/services/horizon/internal/db2/history/key_value.go b/services/horizon/internal/db2/history/key_value.go
index b5af30954b..bea4855d2f 100644
--- a/services/horizon/internal/db2/history/key_value.go
+++ b/services/horizon/internal/db2/history/key_value.go
@@ -96,10 +96,10 @@ func (q *Q) GetExpIngestVersion() (int, error) {
 }
 
 // UpdateExpIngestVersion updates the exp ingest version.
-func (q *Q) UpdateExpIngestVersion(ledgerSequence int) error {
+func (q *Q) UpdateExpIngestVersion(version int) error {
 	return q.updateValueInStore(
 		ingestVersion,
-		strconv.FormatUint(uint64(ledgerSequence), 10),
+		strconv.FormatUint(uint64(version), 10),
 	)
 }