feat: merge and replace load method

goto · Oct 11, 2024 · d5d1894 · d5d1894
1 parent 42a6804
commit d5d1894
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/max2max/internal/loader/merge.go b/max2max/internal/loader/merge.go
@@ -14,10 +14,10 @@ func NewMergeLoader(logger *slog.Logger) *mergeLoader {
 	}
 }
 
-func (l *mergeLoader) GetQuery(tableID, query string) string {
-	return "-- TODO merge loader"
+func (l *mergeLoader) GetQuery(_, query string) string {
+	return query
 }
 
-func (l *mergeLoader) GetPartitionedQuery(tableID, query string, partitionName []string) string {
-	return "-- TODO merge loader"
+func (l *mergeLoader) GetPartitionedQuery(_, query string, _ []string) string {
+	return query
 }
diff --git a/max2max/internal/loader/replace.go b/max2max/internal/loader/replace.go
@@ -1,7 +1,9 @@
 package loader
 
 import (
+	"fmt"
 	"log/slog"
+	"strings"
 )
 
 type replaceLoader struct {
@@ -15,9 +17,9 @@ func NewReplaceLoader(logger *slog.Logger) *replaceLoader {
 }
 
 func (l *replaceLoader) GetQuery(tableID, query string) string {
-	return "-- TODO replace loader"
+	return fmt.Sprintf("INSERT OVERWRITE TABLE %s %s", tableID, query)
 }
 
-func (l *replaceLoader) GetPartitionedQuery(tableID, query string, partitionName []string) string {
-	return "-- TODO replace loader"
+func (l *replaceLoader) GetPartitionedQuery(tableID, query string, partitionNames []string) string {
+	return fmt.Sprintf("INSERT OVERWRITE TABLE %s PARTITION (%s) %s", tableID, strings.Join(partitionNames, ", "), query)
 }