98 lines
1.9 KiB
Go
98 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"git.ksdemosapps.com/kylesoda/go-migrate/internal/app/config"
|
|
_ "github.com/microsoft/go-mssqldb"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func main() {
|
|
log.SetFormatter(&log.TextFormatter{
|
|
FullTimestamp: true,
|
|
TimestampFormat: time.StampMilli,
|
|
})
|
|
log.SetLevel(log.DebugLevel)
|
|
|
|
db, err := sql.Open("sqlserver", config.App.SourceDbUrl)
|
|
if err != nil {
|
|
log.Fatal("Unexpected error connecting to mssql", err)
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
schema := "Cartografia"
|
|
table := "MANZANA"
|
|
|
|
query := buildExtractSqlSentence(schema, table, []string{})
|
|
|
|
rows, err := db.QueryContext(ctx, query)
|
|
if err != nil {
|
|
log.Fatal("Unexpected error extracting data", err)
|
|
}
|
|
defer rows.Close()
|
|
|
|
cols, _ := rows.Columns()
|
|
|
|
values := make([]any, len(cols))
|
|
scanArgs := make([]any, len(cols))
|
|
for i := range values {
|
|
scanArgs[i] = &values[i]
|
|
}
|
|
|
|
colTypes, err := rows.ColumnTypes()
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
for _, col := range colTypes {
|
|
log.Debugf("%+v", col)
|
|
}
|
|
|
|
count := 0
|
|
for rows.Next() {
|
|
if err := rows.Scan(scanArgs...); err != nil {
|
|
log.Fatal("Error scan", err)
|
|
}
|
|
|
|
count++
|
|
if count%100000 == 0 {
|
|
log.Infof("Procesadas %d filas...\n", count)
|
|
}
|
|
|
|
if count < 2 {
|
|
log.Debugf("Processed row (%d):", count)
|
|
|
|
for i, col := range cols {
|
|
log.Debugf("%s: %v", col, values[i])
|
|
}
|
|
}
|
|
}
|
|
|
|
log.Infof("Rows processed %d", count)
|
|
}
|
|
|
|
func buildExtractSqlSentence(schema, table string, colNames []string) string {
|
|
var sbColumns strings.Builder
|
|
|
|
if len(colNames) == 0 {
|
|
sbColumns.WriteString("*")
|
|
} else {
|
|
for i, col := range colNames {
|
|
sbColumns.WriteString(`[`)
|
|
sbColumns.WriteString(col)
|
|
sbColumns.WriteString(`]`)
|
|
if i < len(colNames)-1 {
|
|
sbColumns.WriteString(", ")
|
|
}
|
|
}
|
|
}
|
|
|
|
return fmt.Sprintf(`SELECT %s FROM [%s].[%s] WITH (NOLOCK)`, sbColumns.String(), schema, table)
|
|
}
|