EdgeAPI/internal/setup/sql_dump_test.go

package setup

import (
	"encoding/json"
	"testing"
	"time"

	"github.com/iwind/TeaGo/dbs"
)

func TestSQLDump_Dump(t *testing.T) {
	db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
		Driver: "mysql",
		Dsn:    "root:123456@tcp(127.0.0.1:3306)/db_edge?charset=utf8mb4&timeout=30s",
		Prefix: "edge",
	})
	if err != nil {
		t.Fatal(err)
	}
	defer func() {
		_ = db.Close()
	}()

	dump := NewSQLDump()
	result, err := dump.Dump(db, true)
	if err != nil {
		t.Fatal(err)
	}

	// Table
	for _, table := range result.Tables {
		_ = table
		//t.Log(table.Name, table.Engine, table.Charset)

		/**for _, field := range table.Fields {
			t.Log("===", field.Name, ":", field.Definition)
		}**/
		/**for _, index := range table.Indexes {
			t.Log("===", index.Name, ":", index.Definition)
		}**/

		/**for _, record := range table.Records {
			t.Log(record.Id, record.Values)
		}**/
	}

	data, err := json.Marshal(result)
	if err != nil {
		t.Fatal(err)
	}
	t.Log(len(data), "bytes")
}

func TestSQLDump_Apply(t *testing.T) {
	db, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
		Driver: "mysql",
		Dsn:    "root:123456@tcp(127.0.0.1:3306)/db_edge?charset=utf8mb4&timeout=30s",
		Prefix: "edge",
	})
	if err != nil {
		t.Fatal(err)
	}
	defer func() {
		_ = db.Close()
	}()

	var dump = NewSQLDump()
	result, err := dump.Dump(db, true)
	if err != nil {
		t.Fatal(err)
	}

	var before = time.Now()
	defer func() {
		t.Log("cost:", time.Since(before))
	}()

	db2, err := dbs.NewInstanceFromConfig(&dbs.DBConfig{
		Driver: "mysql",
		Dsn:    "edge:123456@tcp(192.168.2.60:3306)/db_edge_new?charset=utf8mb4&timeout=30s",
		Prefix: "edge",
	})
	if err != nil {
		t.Fatal(err)
	}
	defer func() {
		_ = db2.Close()
	}()
	ops, err := dump.Apply(db2, result, false)
	if err != nil {
		t.Fatal(err)
	}
	t.Log("ok")
	/**if len(ops) > 0 {
		for _, op := range ops {
			t.Log("", op)
		}
	}**/
	_ = ops
}