fix: construct ordered column with header separation first

deryrahman · deryrahman · commit 70ebef48da65 · 2024-12-09T16:24:34.000+07:00
diff --git a/mc2mc/internal/client/client.go b/mc2mc/internal/client/client.go
@@ -110,5 +110,6 @@ func addPartitionValueColumn(rawQuery []byte) []byte {
 }
 
 func constructQueryWithOrderedColumns(query []byte, orderedColumns []string) []byte {
-	return []byte(loader.ConstructQueryWithOrderedColumns(string(query), orderedColumns))
+	header, qr := loader.SeparateHeadersAndQuery(string(query))
+	return []byte(fmt.Sprintf("%s %s", header, loader.ConstructQueryWithOrderedColumns(qr, orderedColumns)))
 }

Original file line number	Diff line number	Diff line change
`@@ -110,5 +110,6 @@ func addPartitionValueColumn(rawQuery []byte) []byte {`
`110`	`110`	`}`
`111`	`111`
`112`	`112`	`func constructQueryWithOrderedColumns(query []byte, orderedColumns []string) []byte {`
`113`		`- return []byte(loader.ConstructQueryWithOrderedColumns(string(query), orderedColumns))`
	`113`	`+ header, qr := loader.SeparateHeadersAndQuery(string(query))`
	`114`	`+ return []byte(fmt.Sprintf("%s %s", header, loader.ConstructQueryWithOrderedColumns(qr, orderedColumns)))`
`114`	`115`	`}`