added composite in support delegation

Author: adranwit

metadata/info/composite.go

@@ -0,0 +1,59 @@
+package info
+
+import "strings"
+
+// CompositeIn renders a dialect-aware composite IN predicate using generic '?'
+// placeholders. Placeholder numbering, when needed, is applied later by EnsurePlaceholders.
+func (d *Dialect) CompositeIn(columns []string, rowCount int) string {
+	if len(columns) == 0 || rowCount <= 0 {
+		return "1 = 0"
+	}
+
+	if len(columns) == 1 {
+		return scalarIn(columns[0], rowCount)
+	}
+
+	if d != nil && d.CompositeInRenderer != nil {
+		return d.CompositeInRenderer(columns, rowCount)
+	}
+	return defaultCompositeIn(columns, rowCount)
+}
+
+func scalarIn(column string, rowCount int) string {
+	if strings.TrimSpace(column) == "" || rowCount <= 0 {
+		return "1 = 0"
+	}
+	builder := &strings.Builder{}
+	builder.WriteString(column)
+	builder.WriteString(" IN (")
+	for i := 0; i < rowCount; i++ {
+		if i > 0 {
+			builder.WriteString(", ")
+		}
+		builder.WriteByte('?')
+	}
+	builder.WriteByte(')')
+	return builder.String()
+}
+
+func defaultCompositeIn(columns []string, rowCount int) string {
+	builder := &strings.Builder{}
+	builder.WriteByte('(')
+	builder.WriteString(strings.Join(columns, ", "))
+	builder.WriteString(") IN (")
+	for row := 0; row < rowCount; row++ {
+		if row > 0 {
+			builder.WriteString(", ")
+		}
+		builder.WriteByte('(')
+		for col := range columns {
+			if col > 0 {
+				builder.WriteString(", ")
+			}
+			builder.WriteByte('?')
+		}
+		builder.WriteByte(')')
+	}
+	builder.WriteByte(')')
+	return builder.String()
+}

metadata/info/composite_test.go

@@ -0,0 +1,55 @@
+package info
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/viant/sqlx/metadata/database"
+)
+
+func TestDialect_CompositeIn(t *testing.T) {
+	testCases := []struct {
+		description string
+		dialect     *Dialect
+		columns     []string
+		rowCount    int
+		expect      string
+	}{
+		{
+			description: "mysql composite",
+			dialect:     &Dialect{Product: database.Product{Name: "MySQL"}},
+			columns:     []string{"t.advertiser_id", "t.val"},
+			rowCount:    2,
+			expect:      "(t.advertiser_id, t.val) IN ((?, ?), (?, ?))",
+		},
+		{
+			description: "postgres composite",
+			dialect:     &Dialect{Product: database.Product{Name: "PostgreSQL"}},
+			columns:     []string{"t.advertiser_id", "t.val"},
+			rowCount:    2,
+			expect:      "(t.advertiser_id, t.val) IN ((?, ?), (?, ?))",
+		},
+		{
+			description: "bigquery composite",
+			dialect: &Dialect{Product: database.Product{Name: "BigQuery"}, CompositeInRenderer: func(columns []string, rowCount int) string {
+				return "(t.advertiser_id, t.val) IN (SELECT AS STRUCT ?, ? UNION ALL SELECT AS STRUCT ?, ?)"
+			}},
+			columns:  []string{"t.advertiser_id", "t.val"},
+			rowCount: 2,
+			expect:   "(t.advertiser_id, t.val) IN (SELECT AS STRUCT ?, ? UNION ALL SELECT AS STRUCT ?, ?)",
+		},
+		{
+			description: "empty rows",
+			dialect:     &Dialect{Product: database.Product{Name: "MySQL"}},
+			columns:     []string{"t.advertiser_id", "t.val"},
+			rowCount:    0,
+			expect:      "1 = 0",
+		},
+	}
+
+	for _, testCase := range testCases {
+		t.Run(testCase.description, func(t *testing.T) {
+			assert.Equal(t, testCase.expect, testCase.dialect.CompositeIn(testCase.columns, testCase.rowCount))
+		})
+	}
+}

metadata/info/dialect.go

@@ -7,11 +7,12 @@ import (
 	"strings"
 )
 
-//Dialect represents dialect
+// Dialect represents dialect
 type Dialect struct {
 	database.Product
 	Placeholder         string // prepare statement placeholder, default '?', but oracle uses ':'
 	PlaceholderResolver placeholder.Generator
+	CompositeInRenderer func(columns []string, rowCount int) string
 	Transactional       bool
 	Insert              dialect.InsertFeatures
 	Upsert              dialect.UpsertFeatures
@@ -29,18 +30,18 @@ type Dialect struct {
 	SpecialKeywordEscapeQuote byte
 }
 
-//Dialects represents dialects
+// Dialects represents dialects
 type Dialects []*Dialect
 
-//PlaceholderGetter returns PlaceholderResolver if not nil, otherwise returns function that returns Placeholder
+// PlaceholderGetter returns PlaceholderResolver if not nil, otherwise returns function that returns Placeholder
 func (d *Dialect) PlaceholderGetter() func() string {
 	if d.PlaceholderResolver != nil {
 		return d.PlaceholderResolver.Resolver()
 	}
 	return (&placeholder.DefaultGenerator{}).Resolver()
 }
 
-//EnsurePlaceholders converts '?' to specific dialect placeholders if needed
+// EnsurePlaceholders converts '?' to specific dialect placeholders if needed
 func (d *Dialect) EnsurePlaceholders(SQL string) string {
 	if d.Placeholder == placeholder.Default {
 		return SQL

metadata/product/bigquery/composite.go

@@ -0,0 +1,24 @@
+package bigquery
+
+import "strings"
+
+func compositeIn(columns []string, rowCount int) string {
+	builder := &strings.Builder{}
+	builder.WriteByte('(')
+	builder.WriteString(strings.Join(columns, ", "))
+	builder.WriteString(") IN (")
+	for row := 0; row < rowCount; row++ {
+		if row > 0 {
+			builder.WriteString(" UNION ALL ")
+		}
+		builder.WriteString("SELECT AS STRUCT ")
+		for col := range columns {
+			if col > 0 {
+				builder.WriteString(", ")
+			}
+			builder.WriteByte('?')
+		}
+	}
+	builder.WriteByte(')')
+	return builder.String()
+}

metadata/product/bigquery/meta.go

@@ -200,6 +200,7 @@ SESSION_USER() AS USER_NAME,
 	registry.RegisterDialect(&info.Dialect{
 		Product:                 bigQuery,
 		Placeholder:             "?",
+		CompositeInRenderer:     compositeIn,
 		Transactional:           false, //only script is transactional
 		Insert:                  dialect.InsertWithMultiValues,
 		Upsert:                  dialect.UpsertTypeMerge,