(Feat): Initial Commit

cmd/db/seed.go

@@ -0,0 +1,77 @@
+package db
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+
+	"allaboutapps.dev/aw/go-starter/internal/api"
+	"allaboutapps.dev/aw/go-starter/internal/config"
+	data "allaboutapps.dev/aw/go-starter/internal/data/fixtures"
+	"allaboutapps.dev/aw/go-starter/internal/util"
+	"allaboutapps.dev/aw/go-starter/internal/util/command"
+	dbutil "allaboutapps.dev/aw/go-starter/internal/util/db"
+	"github.com/aarondl/sqlboiler/v4/boil"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/cobra"
+)
+
+func newSeed() *cobra.Command {
+	return &cobra.Command{
+		Use:   "seed",
+		Short: "Inserts or updates fixtures to the database.",
+		Long:  `Uses upsert to add test data to the current environment.`,
+		Run: func(_ *cobra.Command, _ []string) {
+			seedCmdFunc()
+		},
+	}
+}
+
+func seedCmdFunc() {
+	err := command.WithServer(context.Background(), config.DefaultServiceConfigFromEnv(), func(ctx context.Context, s *api.Server) error {
+		log := util.LogFromContext(ctx)
+
+		err := ApplySeedFixtures(ctx, s.Config)
+		if err != nil {
+			log.Err(err).Msg("Error while applying seed fixtures")
+			return err
+		}
+
+		log.Info().Msg("Successfully applied seed fixtures")
+
+		return nil
+	})
+	if err != nil {
+		log.Fatal().Err(err).Msg("Failed to apply migrations")
+	}
+}
+
+func ApplySeedFixtures(ctx context.Context, config config.Server) error {
+	log := util.LogFromContext(ctx)
+
+	db, err := sql.Open("postgres", config.Database.ConnectionString())
+	if err != nil {
+		return fmt.Errorf("failed to open the database: %w", err)
+	}
+
+	defer db.Close()
+
+	if err := db.PingContext(ctx); err != nil {
+		return fmt.Errorf("failed to ping the database: %w", err)
+	}
+
+	// insert fixtures in an auto-managed db transaction
+	return dbutil.WithTransaction(ctx, db, func(tx boil.ContextExecutor) error {
+		fixtures := data.Upserts()
+
+		for _, fixture := range fixtures {
+			if err := fixture.Upsert(ctx, tx, true, nil, boil.Infer(), boil.Infer()); err != nil {
+				log.Error().Err(err).Msg("Failed to upsert fixture")
+				return err
+			}
+		}
+
+		log.Info().Int("fixturesCount", len(fixtures)).Msg("Successfully upserted fixtures")
+		return nil
+	})
+}