feat: add WKB to EWKB conversion with SRID handling and integrate into MSSQL transformation

cmd/go_migrate/mssql-transform.go

@@ -1,5 +1,9 @@
 package main
 
+import (
+	"encoding/binary"
+)
+
 func mssqlUuidToBigEndian(mssqlUuid []byte) []byte {
 	if len(mssqlUuid) != 16 {
 		return mssqlUuid
@@ -12,3 +16,37 @@ func mssqlUuidToBigEndian(mssqlUuid []byte) []byte {
 
 	return pgUuid
 }
+
+const sridFlag = 0x20000000
+
+func wkbToEwkbWithSrid(geometry []byte, srid int) []byte {
+	if len(geometry) < 5 {
+		return geometry
+	}
+
+	var byteOrder binary.ByteOrder
+	if geometry[0] == 0 {
+		byteOrder = binary.BigEndian
+	} else {
+		byteOrder = binary.LittleEndian
+	}
+
+	wkbType := byteOrder.Uint32(geometry[1:5])
+	if wkbType&sridFlag != 0 {
+		return geometry
+	}
+
+	ewkbType := wkbType | sridFlag
+
+	result := make([]byte, len(geometry)+4)
+
+	result[0] = geometry[0]
+
+	byteOrder.PutUint32(result[1:5], ewkbType)
+
+	byteOrder.PutUint32(result[5:9], uint32(srid))
+
+	copy(result[9:], geometry[5:])
+
+	return result
+}

cmd/go_migrate/process.go

@@ -79,6 +79,10 @@ func transformRowsMssql(columns []ColumnType, in <-chan []UnknownRowValues, out
 					if b, ok := value.([]byte); ok {
 						rowValues[i] = mssqlUuidToBigEndian(b)
 					}
+				} else if col.SystemType() == "geometry" || col.SystemType() == "geography" {
+					if b, ok := value.([]byte); ok {
+						rowValues[i] = wkbToEwkbWithSrid(b, 4326)
+					}
 				}
 			}
 		}
@@ -90,7 +94,7 @@ func transformRowsMssql(columns []ColumnType, in <-chan []UnknownRowValues, out
 func logSampleRow(job MigrationJob, columns []ColumnType, rowValues UnknownRowValues, tag string) {
 	log.Infof("[%s.%s] Sample row: (%s)", job.Schema, job.Table, tag)
 	for i, col := range columns {
-		log.Infof("%s: %v", col.Name(), rowValues[i])
+		log.Infof("%s (%T): %v", col.Name(), rowValues[i], rowValues[i])
 	}
 }