Add support for WITH CTE remapping

src/query_handler_test.go

@@ -512,6 +512,11 @@ func TestHandleQuery(t *testing.T) {
 			"description": {"gss_authenticated", "encrypted"},
 			"values":      {},
 		},
+		// WITH
+		"WITH RECURSIVE simple_cte AS (SELECT oid, rolname FROM pg_roles WHERE rolname = 'postgres' UNION ALL SELECT oid, rolname FROM pg_roles) SELECT * FROM simple_cte": {
+			"description": {"oid", "rolname"},
+			"values":      {"10", "bemidb"},
+		},
 	}
 
 	for query, responses := range responsesByQuery {

src/select_remapper.go

@@ -88,6 +88,16 @@ func (selectRemapper *SelectRemapper) remapSelectStatement(selectStatement *pgQu
 		selectStatement = selectRemapper.remapperWhere.RemapWhereExpressions(selectStatement, selectStatement.WhereClause, indentLevel)
 	}
 
+	// WITH
+	if selectStatement.WithClause != nil {
+		selectRemapper.traceTreeTraversal("WITH CTE's", indentLevel)
+		for _, cte := range selectStatement.WithClause.Ctes {
+			if cteSelect := cte.GetCommonTableExpr().Ctequery.GetSelectStmt(); cteSelect != nil {
+				selectRemapper.remapSelectStatement(cteSelect, indentLevel+1) // recursive
+			}
+		}
+	}
+
 	// FROM
 	if len(selectStatement.FromClause) > 0 {
 		for i, fromNode := range selectStatement.FromClause {