Skip to content

Commit 7de6838

Browse files
committed
Small tweaks
1 parent bcc41d3 commit 7de6838

4 files changed

Lines changed: 30 additions & 13 deletions

File tree

Sources/Compiler/Gen/SwiftLanguage.swift

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -215,8 +215,8 @@ public struct SwiftLanguage: Language {
215215

216216
for query in queries {
217217
let associatedType = query.name.capitalizedFirst
218-
writer.write(line: "associatedtype ", associatedType, ": ", query.typealiasName)
219-
writer.write(line: "var ", query.variableName, ": ", associatedType, " { get }")
218+
writer.write(line: "associatedtype _", associatedType, ": ", query.typealiasName)
219+
writer.write(line: "var ", query.variableName, ": _", associatedType, " { get }")
220220
}
221221

222222
writer.unindent()
@@ -312,14 +312,8 @@ public struct SwiftLanguage: Language {
312312

313313
writer.write(line: ") { input, tx in")
314314
writer.indent()
315-
316-
if query.input == .void {
317-
writer.write(line: "let")
318-
} else {
319-
writer.write(line: "var")
320-
}
321-
322-
writer.write(" statement = try Otter.Statement(")
315+
316+
writer.write(line: "let statement = try Otter.Statement(")
323317
writer.indent()
324318
multilineStringLiteral(of: query.sourceSql)
325319
writer.write(",")

Sources/Compiler/Parse/Parsers.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1148,7 +1148,7 @@ enum Parsers {
11481148
kind = .table(table)
11491149
}
11501150
case .openParen:
1151-
if state.peek.kind == .select {
1151+
if state.peek.kind == .select || state.peek.kind == .values {
11521152
let subquery = try parens(state: &state, value: selectStmt)
11531153
let alias = maybeAlias(state: &state, asRequired: false)
11541154
kind = .subquery(subquery, alias: alias)

Sources/Otter/QueryObservation.swift

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,5 +113,3 @@ extension QueryObservation {
113113
}
114114
}
115115
}
116-
117-
public final class QueryObservation2 {}

Tests/CompilerTests/Compiler/CompileSimpleSelects.sql

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -257,3 +257,28 @@ LIMIT ?;
257257
-- CHECK: TABLES
258258
-- CHECK: foo
259259
SELECT id = 1 AS isOne FROM foo;
260+
261+
-- CHECK: SIGNATURE
262+
-- CHECK: OUTPUT_CHUNKS
263+
-- CHECK: CHUNK
264+
-- CHECK: OUTPUT
265+
-- CHECK: b INTEGER
266+
-- CHECK: c INTEGER
267+
WITH a AS (
268+
SELECT column1 AS b, column2 AS c FROM (VALUES (1, 2))
269+
)
270+
SELECT a.b, a.c FROM a;
271+
272+
-- CHECK: SIGNATURE
273+
-- CHECK: OUTPUT_CHUNKS
274+
-- CHECK: CHUNK
275+
-- CHECK: OUTPUT
276+
-- CHECK: b INTEGER
277+
-- CHECK: e INTEGER
278+
WITH a AS (
279+
SELECT column1 AS b, column2 AS c FROM (VALUES (1, 2))
280+
),
281+
d AS (
282+
SELECT column1 AS e, column2 AS f FROM (VALUES (1, 2))
283+
)
284+
SELECT a.b, d.e FROM a INNER JOIN d ON a.b = d.e;

0 commit comments

Comments
 (0)