TestMu Conference is Back! Future of Testing + AI LambdaTest G2 Announcement

How to use runMigrations method of main Package

Best Testkube code snippet using main.runMigrations

migrate.go

Source:migrate.go

...45	defer f.Close()46	entry.Info("migration created successfully")47	return nil48}49func runMigrations(core *core) {50	if logger == nil {51		panic("[runMigrations] logger is not configured, please run buildCore first")52	}53	if core == nil {54		logger.Fatal("[runMigrations] core cannot be nil")55	}56	if dbx == nil {57		logger.Fatal("[runMigrations] DB has not been initialized")58	}59	if !cfg.MySQL.Migrate {60		return61	}62	files, err := filepath.Glob(fmt.Sprintf("%s*.sql", migrationDir))63	if err != nil {64		core.logger.WithError(err).Fatal("[runMigrations] failed to read migrations")65	}66	sort.Strings(files)67	var ctx = context.Background()68	err = core.repos.migrations.InitializeMigrationsTable(ctx, cfg.MySQL.DB, "migrations")69	if err != nil {70		core.logger.WithError(err).Fatal("[runMigrations] failed to initialize migrations table")71	}72	core.logger.Info("[runMigrations] migration table initialized, checking migrations")73	for _, file := range files {74		name := strings.TrimPrefix(file, migrationDir)75		entry := core.logger.WithField("migration", name)76		_, err := core.repos.migrations.Migration(ctx, name)77		if err != nil && !errors.Is(err, sql.ErrNoRows) {78			entry.WithError(err).Fatal("[runMigrations] failed to check if migration has been executed")79		}80		if err == nil {81			continue82		}83		entry.Info("new migration found, executing migration")84		f, err := os.Open(file)85		if err != nil {86			entry.WithError(err).Fatal("[runMigrations] failed to open migration file")87		}88		data, err := io.ReadAll(f)89		if err != nil {90			entry.WithError(err).Fatal("[runMigrations] failed to read migration file")91		}92		if len(data) == 0 {93			entry.WithError(err).Fatal("[runMigrations] empty migration file detected, halting execution")94		}95		query := string(data)96		tx, err := dbx.BeginTxx(ctx, nil)97		if err != nil {98			entry.WithError(err).Fatal("[runMigrations] failed initialize transaction")99		}100		m := &ledger.Migration{Name: name, Executed: false}101		m, err = core.repos.migrations.CreateMigrationTx(ctx, tx, m)102		if err != nil {103			entry.WithError(err).Fatal("[runMigrations] failed to create migration")104		}105		_, err = dbx.ExecContext(ctx, query)106		if err != nil {107			entry.WithError(err).Fatal("[runMigrations] failed to execute migration")108		}109		m.Executed = true110		_, err = core.repos.migrations.UpdateMigrationTx(ctx, tx, m)111		if err != nil {112			entry.WithError(err).Fatal("[runMigrations] failed to update migration")113		}114		err = tx.Commit()115		if err != nil {116			entry.WithError(err).Fatal("[runMigrations] failed to commit transactions")117		}118		entry.Info("migration executed successfully")119	}120}...

goose.go

Source:goose.go

...50	return filenames51}52func lastMigration(gaggle string) string {53	if gaggle != "" {54		runMigrations := strings.Split(gaggle, "\n")55		return runMigrations[len(runMigrations)-1]56	} else {57		return ""58	}59}60func runMigrations(filenames []string) {61	config, err := toml.LoadFile(".gooserc")62	checkError(err)63	executable := config.Get("goose.path").(string)64	database := config.Get("goose.db").(string)65	hostname := config.Get("goose.hostname").(string)66	port := config.Get("goose.port").(string)67	name := config.Get("goose.name").(string)68	username := config.Get("goose.username").(string)69	password := config.Get("goose.password").(string)70	if database == "postgres" {71		executedFiles := make([]string, 0)72		for i := 0; i < (len(filenames)); i++ {73			fmt.Println(fmt.Sprintf("Running Migration: %v", filenames[i]))74			cmd := exec.Command(executable, "-h", hostname, "-p", port, "-d", name, "-U", username, "-f", fmt.Sprintf("migrations/%s", strings.TrimLeft(filenames[i], " ")))75			cmd.Env = []string{fmt.Sprintf("PGPASSWORD=%s", password)}76			var out bytes.Buffer77			var stderr bytes.Buffer78			cmd.Stdout = &out79			cmd.Stderr = &stderr80			err := cmd.Run()81			if err != nil {82				fmt.Println(fmt.Sprint(err) + ": " + stderr.String())83				checkError(err)84			} else {85				fmt.Println(out.String())86				executedFiles = append(executedFiles, filenames[i])87			}88		}89		writeGaggle(executedFiles)90	}91}92func migrate() {93	data, err := ioutil.ReadFile(".gaggle")94	lastMigration := lastMigration(strings.Trim(string(data), "\n"))95	if (err != nil) || (lastMigration == "") {96		unrunMigrations := allMigrations()97		runMigrations(unrunMigrations)98		fmt.Println("Migrations complete.")99	} else {100		filenames := allMigrations()101		i := sort.SearchStrings(filenames, lastMigration)102		if (i + 1) == len(filenames) {103			fmt.Println("Migrations up to date.")104		} else {105			unrunMigrations := filenames[(i + 1):]106			runMigrations(unrunMigrations)107			fmt.Println("Migrations complete.")108		}109	}110}111func checkForRc() {112	_, err := os.Stat(".gooserc")113	if err != nil {114		fmt.Println("Can't find .gooserc. Please create it with with your database connection settings and try again.")115		os.Exit(0)116	}117}118func main() {119	checkForRc()120	command := os.Args[1]...

run.go

Source:run.go

2import (3	"github.com/centrifuge/go-centrifuge/cmd"4	"github.com/spf13/cobra"5)6var runMigrations bool7func init() {8	// runCmd represents the run command9	var runCmd = &cobra.Command{10		Use:   "run",11		Short: "run a centrifuge node",12		Long:  ``,13		Run: func(cm *cobra.Command, args []string) {14			// cm requires a config file15			cfgFile := ensureConfigFile()16			// Check if migrations should run17			if runMigrations {18				err := doMigrate()19				if err != nil {20					log.Fatal(err)21				}22			}23			// the following call will block24			cmd.RunBootstrap(cfgFile)25		},26	}27	runCmd.Flags().BoolVarP(&runMigrations, "runmigrations", "m", true, "Run Migrations at startup (-m=false)")28	rootCmd.AddCommand(runCmd)29}...

runMigrations

Using AI Code Generation

1import (2func main() {3	db, err := sql.Open("postgres", os.Getenv("DATABASE_URL"))4	if err != nil {5		log.Fatal(err)6	}7	if err := goose.RunMigrations(db, "./", os.Args[1:]); err != nil {8		log.Fatal(err)9	}10}11import (12func init() {13	goose.AddMigration(Up2, Down2)14}15func Up2(tx *sql.Tx) error {16	fmt.Println("Up2")17}18func Down2(tx *sql.Tx) error {19	fmt.Println("Down2")20}21import (22func init() {23	goose.AddMigration(Up3, Down3)24}25func Up3(tx *sql.Tx) error {26	fmt.Println("Up3")27}28func Down3(tx *sql.Tx) error {29	fmt.Println("Down3")30}31import (32func init() {33	goose.AddMigration(Up4, Down4)34}35func Up4(tx *sql.Tx) error {36	fmt.Println("Up4")37}38func Down4(tx *sql.Tx) error {39	fmt.Println("Down4")40}41import (42func init() {43	goose.AddMigration(Up5, Down5)44}45func Up5(tx *sql.Tx) error {46	fmt.Println("Up5")47}48func Down5(tx *sql.Tx) error {49	fmt.Println("Down5")50}51import (

runMigrations

Using AI Code Generation

1func main() {2    main := &main{}3    main.runMigrations()4}5func main() {6    main := &main{}7    main.runMigrations()8}9func main() {10    main := &main{}11    main.runMigrations()12}13func main() {14    main := &main{}15    main.runMigrations()16}

runMigrations

Using AI Code Generation

1import (2func main() {3	fmt.Println("Hello World")4	migrations.RunMigrations()5}6import (7func RunMigrations() {8	fmt.Println("Running migrations")9	migration.RunMigration()10}11import (12func RunMigration() {13	fmt.Println("Running migrations")14}15import (16func main() {17	fmt.Println("Hello World")18}19import (20func init() {21	fmt.Println("Running migrations")22	migration.RunMigration()23}24import (25func RunMigration() {26	fmt.Println("Running migrations")27}

runMigrations

Using AI Code Generation

1import (2func main() {3	migrate.RunMigrations()4	fmt.Println("Migration has been completed")5}6import (7func main() {8	migrate.RunMigrations()9	fmt.Println("Migration has been completed")10}11./1.go:4:2: import "migrate" is a program, not an importable package12./1.go:4:2: import "migrate" is a program, not an importable package

runMigrations

Using AI Code Generation

1func main() {2    m := &main{}3    m.runMigrations()4}5func main() {6    m := &main{}7    m.runMigrations()8}9func main() {10    m := &main{}11    m.runMigrations()12}13func main() {14    m := &main{}15    m.runMigrations()16}17func main() {18    m := &main{}19    m.runMigrations()20}21func main() {22    m := &main{}23    m.runMigrations()24}25func main() {26    m := &main{}27    m.runMigrations()28}29func main() {30    m := &main{}31    m.runMigrations()32}33func main() {34    m := &main{}35    m.runMigrations()36}37func main() {

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.