File tree 1 file changed +8
-7
lines changed
Sources/FluentPostgresDriver
1 file changed +8
-7
lines changed Original file line number Diff line number Diff line change @@ -14,7 +14,7 @@ extension _FluentPostgresDatabase: Database {
14
14
. convert ( query)
15
15
switch query. action {
16
16
case . create:
17
- expression = PostgresReturning ( expression)
17
+ expression = PostgresReturningKey ( key : query . idKey , base : expression)
18
18
default : break
19
19
}
20
20
let ( sql, binds) = self . serialize ( expression)
@@ -70,14 +70,15 @@ extension _FluentPostgresDatabase: PostgresDatabase {
70
70
}
71
71
}
72
72
73
- private struct PostgresReturning : SQLExpression {
73
+ private struct PostgresReturningKey : SQLExpression {
74
+ let key : String
74
75
let base : SQLExpression
75
- init ( _ base: SQLExpression ) {
76
- self . base = base
77
- }
78
76
79
77
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
+ }
82
83
}
83
84
}
You can’t perform that action at this time.
0 commit comments