snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/sql/ast/infer_clause.go

@@ -0,0 +1,34 @@
+package ast
+
+import "github.com/sqlc-dev/sqlc/internal/sql/format"
+
+type InferClause struct {
+	Tag NodeTag[InferClause] `json:"tag"`
+
+	IndexElems  *List   `json:"index_elems,omitempty"`
+	WhereClause Node    `json:"where_clause,omitempty"`
+	Conname     *string `json:"conname,omitempty"`
+	Location    int     `json:"location"`
+}
+
+func (n *InferClause) Pos() int {
+	return n.Location
+}
+
+func (n *InferClause) Format(buf *TrackedBuffer, d format.Dialect) {
+	if n == nil {
+		return
+	}
+	if n.Conname != nil && *n.Conname != "" {
+		buf.WriteString("ON CONSTRAINT ")
+		buf.WriteString(*n.Conname)
+	} else if items(n.IndexElems) {
+		buf.WriteString("(")
+		buf.join(n.IndexElems, d, ", ")
+		buf.WriteString(")")
+		if set(n.WhereClause) {
+			buf.WriteString(" WHERE ")
+			buf.astFormat(n.WhereClause, d)
+		}
+	}
+}