whatcanGOwrong

go/pkg/mod/github.com/golang-migrate/migrate/v4@v4.17.1/database/stub/stub.go

@@ -0,0 +1,97 @@
+package stub
+
+import (
+	"io"
+	"reflect"
+
+	"go.uber.org/atomic"
+
+	"github.com/golang-migrate/migrate/v4/database"
+)
+
+func init() {
+	database.Register("stub", &Stub{})
+}
+
+type Stub struct {
+	Url               string
+	Instance          interface{}
+	CurrentVersion    int
+	MigrationSequence []string
+	LastRunMigration  []byte // todo: make []string
+	IsDirty           bool
+	isLocked          atomic.Bool
+
+	Config *Config
+}
+
+func (s *Stub) Open(url string) (database.Driver, error) {
+	return &Stub{
+		Url:               url,
+		CurrentVersion:    database.NilVersion,
+		MigrationSequence: make([]string, 0),
+		Config:            &Config{},
+	}, nil
+}
+
+type Config struct{}
+
+func WithInstance(instance interface{}, config *Config) (database.Driver, error) {
+	return &Stub{
+		Instance:          instance,
+		CurrentVersion:    database.NilVersion,
+		MigrationSequence: make([]string, 0),
+		Config:            config,
+	}, nil
+}
+
+func (s *Stub) Close() error {
+	return nil
+}
+
+func (s *Stub) Lock() error {
+	if !s.isLocked.CAS(false, true) {
+		return database.ErrLocked
+	}
+	return nil
+}
+
+func (s *Stub) Unlock() error {
+	if !s.isLocked.CAS(true, false) {
+		return database.ErrNotLocked
+	}
+	return nil
+}
+
+func (s *Stub) Run(migration io.Reader) error {
+	m, err := io.ReadAll(migration)
+	if err != nil {
+		return err
+	}
+	s.LastRunMigration = m
+	s.MigrationSequence = append(s.MigrationSequence, string(m[:]))
+	return nil
+}
+
+func (s *Stub) SetVersion(version int, state bool) error {
+	s.CurrentVersion = version
+	s.IsDirty = state
+	return nil
+}
+
+func (s *Stub) Version() (version int, dirty bool, err error) {
+	return s.CurrentVersion, s.IsDirty, nil
+}
+
+const DROP = "DROP"
+
+func (s *Stub) Drop() error {
+	s.CurrentVersion = database.NilVersion
+	s.LastRunMigration = nil
+	s.MigrationSequence = append(s.MigrationSequence, DROP)
+	return nil
+}
+
+func (s *Stub) EqualSequence(seq []string) bool {
+	return reflect.DeepEqual(seq, s.MigrationSequence)
+}

go/pkg/mod/github.com/golang-migrate/migrate/v4@v4.17.1/database/stub/stub_test.go

@@ -0,0 +1,43 @@
+package stub
+
+import (
+	"github.com/golang-migrate/migrate/v4"
+	"github.com/golang-migrate/migrate/v4/source"
+	"github.com/golang-migrate/migrate/v4/source/stub"
+	"testing"
+
+	dt "github.com/golang-migrate/migrate/v4/database/testing"
+)
+
+func Test(t *testing.T) {
+	s := &Stub{}
+	d, err := s.Open("")
+	if err != nil {
+		t.Fatal(err)
+	}
+	dt.Test(t, d, []byte("/* foobar migration */"))
+}
+
+func TestMigrate(t *testing.T) {
+	s := &Stub{}
+	d, err := s.Open("")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	stubMigrations := source.NewMigrations()
+	stubMigrations.Append(&source.Migration{Version: 1, Direction: source.Up, Identifier: "CREATE 1"})
+	stubMigrations.Append(&source.Migration{Version: 1, Direction: source.Down, Identifier: "DROP 1"})
+	src := &stub.Stub{}
+	srcDrv, err := src.Open("")
+	if err != nil {
+		t.Fatal(err)
+	}
+	srcDrv.(*stub.Stub).Migrations = stubMigrations
+	m, err := migrate.NewWithInstance("stub", srcDrv, "", d)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	dt.TestMigrate(t, m)
+}