File tree Expand file tree Collapse file tree 1 file changed +8
-7
lines changed
Sources/FluentPostgresDriver Expand file tree Collapse file tree 1 file changed +8
-7
lines changed Original file line number Diff line number Diff 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}
You can’t perform that action at this time.
0 commit comments