Add command to recalc all clans

Author: Swan

cmd/console/cmd.go

@@ -54,6 +54,7 @@ func init() {
 	RootCmd.AddCommand(commands.DeleteScoreCmd)
 	RootCmd.AddCommand(commands.FixStatsCmd)
 	RootCmd.AddCommand(commands.UpdateStripePriceId)
+	RootCmd.AddCommand(commands.ClanRecalculateCommand)
 
 	// Migrations
 	RootCmd.AddCommand(migrations.MigrationPlaylistMapsetCmd)

cmd/console/commands/clan_recalc.go

@@ -0,0 +1,29 @@
+package commands
+
+import (
+	"github.com/Quaver/api2/db"
+	"github.com/sirupsen/logrus"
+	"github.com/spf13/cobra"
+)
+
+var ClanRecalculateCommand = &cobra.Command{
+	Use:   "clan:recalc",
+	Short: "Recalculates all clans",
+	Run: func(cmd *cobra.Command, args []string) {
+		clans, err := db.GetClans()
+
+		if err != nil {
+			logrus.Error(err)
+			return
+		}
+
+		for _, clan := range clans {
+			if err := db.PerformFullClanRecalculation(clan); err != nil {
+				logrus.Error(err)
+				return
+			}
+
+			logrus.Info("Recalculated clan: ", clan.Id)
+		}
+	},
+}

db/clan_stats.go

@@ -99,6 +99,9 @@ func PerformFullClanRecalculation(clan *Clan) error {
 			return err
 		}
 
+		if err := UpdateClanLeaderboard(clan, mode); err != nil {
+			return err
+		}
 	}
 
 	return nil

db/clans.go

@@ -81,6 +81,21 @@ func (clan *Clan) Insert() error {
 	return nil
 }
 
+// GetClans Retrieves all clans from the db
+func GetClans() ([]*Clan, error) {
+	clans := make([]*Clan, 0)
+
+	result := SQL.
+		Preload("Stats").
+		Find(&clans)
+
+	if result.Error != nil {
+		return nil, result.Error
+	}
+
+	return clans, nil
+}
+
 // GetClanById Gets a clan from the database by its id
 func GetClanById(id int) (*Clan, error) {
 	var clan *Clan