Skip to content

Commit dc03d7c

Browse files
authored
update to fluent-kit #153 (#137)
1 parent 271dae1 commit dc03d7c

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

Sources/FluentPostgresDriver/FluentPostgresDatabase.swift

+8-7
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ extension _FluentPostgresDatabase: Database {
1414
.convert(query)
1515
switch query.action {
1616
case .create:
17-
expression = PostgresReturning(expression)
17+
expression = PostgresReturningKey(key: query.idKey, base: expression)
1818
default: break
1919
}
2020
let (sql, binds) = self.serialize(expression)
@@ -70,14 +70,15 @@ extension _FluentPostgresDatabase: PostgresDatabase {
7070
}
7171
}
7272

73-
private struct PostgresReturning: SQLExpression {
73+
private struct PostgresReturningKey: SQLExpression {
74+
let key: String
7475
let base: SQLExpression
75-
init(_ base: SQLExpression) {
76-
self.base = base
77-
}
7876

7977
func serialize(to serializer: inout SQLSerializer) {
80-
self.base.serialize(to: &serializer)
81-
serializer.write(#" RETURNING id as "fluentID""#)
78+
serializer.statement {
79+
$0.append(self.base)
80+
$0.append("RETURNING")
81+
$0.append(SQLIdentifier(self.key))
82+
}
8283
}
8384
}

0 commit comments

Comments
 (0)