support COUNT_DISTINCT

Toru-Takagi · Feb 12, 2024 · 2afa561 · 2afa561
1 parent 806dc78
commit 2afa561
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/formatter/format_test.go b/formatter/format_test.go
@@ -241,6 +241,15 @@ SELECT
   count(*)
 FROM users u
 GROUP BY u.name, u.age
+`,
+		},
+		{
+			name: "SELECT_COUNT_DISTINCT",
+			sql:  `select count(distinct user_name) from users`,
+			want: `
+SELECT
+  count(DISTINCT user_name)
+FROM users
 `,
 		},
 		{

diff --git a/formatter/node_formatter/format_funccall.go b/formatter/node_formatter/format_funccall.go
@@ -81,6 +81,10 @@ func FormatFuncCallArgs(ctx context.Context, funcCall *pg_query.Node_FuncCall, i
 			bu.WriteString("$")
 			bu.WriteString(fmt.Sprint(n.ParamRef.Number))
 		case *pg_query.Node_ColumnRef:
+			if funcCall.FuncCall.AggDistinct {
+				bu.WriteString("DISTINCT ")
+			}
+
 			field, err := FormatColumnRefFields(ctx, n)
 			if err != nil {
 				return "", err