Skip to content

Commit fe961aa

Browse files
authored
✨ Registration of model middlewares (#1)
1 parent 0da4d52 commit fe961aa

File tree

2 files changed

+14
-3
lines changed

2 files changed

+14
-3
lines changed

.swiftpm/xcode/package.xcworkspace/contents.xcworkspacedata

+7
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

Sources/FluentData/FluentDataContext.swift

+7-3
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,9 @@ public class FluentDataContext {
1313
/// ```swift
1414
/// Planet(name: "Earth").create(on: context.database)
1515
/// ```
16-
public lazy var database: any Database = {
16+
public var database: any Database {
1717
databases.database(.sqlite, logger: logger, on: eventLoopGroup.next())!
18-
}()
18+
}
1919

2020
private let databases: Databases
2121
private let eventLoopGroup: EventLoopGroup
@@ -49,7 +49,11 @@ public class FluentDataContext {
4949
// Register the context
5050
FluentDataContexts[contextKey, makeDefault] = self
5151
}
52-
52+
53+
public func use(middleware: AnyModelMiddleware) {
54+
databases.middleware.use(middleware, on: .sqlite)
55+
}
56+
5357
internal func deregister<Model: FluentKit.Model>(_ fluentQuery: FluentQuery<Model>) {
5458
registeredQueries.removeValue(forKey: fluentQuery.queryId)
5559
}

0 commit comments

Comments
 (0)