kylesoda/go-migrate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: enhance mssql configuration and update extraction script

Browse Source
This commit is contained in:
Kylesoda
2026-04-03 22:55:30 -05:00
parent 7eaa3478e1
commit b04f1315af
2 changed files with 40 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
internal/app/config/main.go
View File

@@ -8,8 +8,10 @@ import (
) )
type appConfig struct { type appConfig struct {
SourceDbUrl string SourceDbUrl string
TargetDbUrl string SourceDbType string
TargetDbUrl string
TargetDbType string
} }
func loadEnv() { func loadEnv() {
@@ -27,14 +29,26 @@ func getAppConfig() appConfig {
log.Fatal("SOURCE_DB_URL environment variable not set") log.Fatal("SOURCE_DB_URL environment variable not set")
} }
sourceDbType := os.Getenv("SOURCE_DB_TYPE")
if sourceDbType == "" {
log.Fatal("SOURCE_DB_TYPE environment variable not set")
}
targetDbUrl := os.Getenv("TARGET_DB_URL") targetDbUrl := os.Getenv("TARGET_DB_URL")
if targetDbUrl == "" { if targetDbUrl == "" {
log.Fatal("TARGET_DB_URL environment variable not set") log.Fatal("TARGET_DB_URL environment variable not set")
} }
targetDbType := os.Getenv("TARGET_DB_TYPE")
if targetDbType == "" {
log.Fatal("TARGET_DB_TYPE environment variable not set")
}
return appConfig{ return appConfig{
SourceDbUrl: sourceDbUrl, SourceDbUrl: sourceDbUrl,
TargetDbUrl: targetDbUrl, SourceDbType: sourceDbType,
TargetDbUrl: targetDbUrl,
TargetDbType: targetDbType,
} }
} }

32
scripts/mssql-test/main.go
View File

@@ -26,11 +26,10 @@ func main() {
ctx := context.Background() ctx := context.Background()
schema := "test" schema := "Cartografia"
table := "migration_test" table := "MANZANA"
colNames := []string{"id", "nombre_producto", "descripcion", "stock", "precio", "es_activo", "fecha_creacion", "ultima_actualizacion", "configuracion_json", "etiquetas", "binario_test", "ip_servidor", "rango_prueba_min", "rango_prueba_max"}
query := buildExtractSqlSentence(schema, table, colNames) query := buildExtractSqlSentence(schema, table, []string{})
rows, err := db.QueryContext(ctx, query) rows, err := db.QueryContext(ctx, query)
if err != nil { if err != nil {
@@ -46,6 +45,15 @@ func main() {
scanArgs[i] = &values[i] scanArgs[i] = &values[i]
} }
colTypes, err := rows.ColumnTypes()
if err != nil {
log.Fatal(err)
}
for _, col := range colTypes {
log.Debugf("%+v", col)
}
count := 0 count := 0
for rows.Next() { for rows.Next() {
if err := rows.Scan(scanArgs...); err != nil { if err := rows.Scan(scanArgs...); err != nil {
@@ -72,12 +80,16 @@ func main() {
func buildExtractSqlSentence(schema, table string, colNames []string) string { func buildExtractSqlSentence(schema, table string, colNames []string) string {
var sbColumns strings.Builder var sbColumns strings.Builder
for i, col := range colNames { if len(colNames) == 0 {
sbColumns.WriteString(`[`) sbColumns.WriteString("*")
sbColumns.WriteString(col) } else {
sbColumns.WriteString(`]`) for i, col := range colNames {
if i < len(colNames)-1 { sbColumns.WriteString(`[`)
sbColumns.WriteString(", ") sbColumns.WriteString(col)
sbColumns.WriteString(`]`)
if i < len(colNames)-1 {
sbColumns.WriteString(", ")
}
} }
} }