以下是我的代码:
var query = Post.query(on: req.db)
if needFilter == true {
query = query.filter(\.$board.$id == boardId)
}
return try await query
.with(\.$profile).all()
上面的代码有点复杂.有没有办法把它写成这样的❓:
return try await query
.filter(\.$board.$id == boardId, enable: needFilter)
.with(\.$profile).all()
‘Enable:’表示此过滤器打开/关闭.
还是有其他方法可以做到这一点？有什么方便的方法吗？
先谢谢你.
您可以这样实现:
return try await query
.filter { needFilter ? $0.$board.$id == boardId : true }
.with(\.$profile)
.all()
编辑: 或者，如果您想要一种可靠的方式:
您可以为QueryBuilder创建一个扩展，添加一个定制的filterIf方法，该方法接受Bool参数enableFilterIf和过滤条件的闭包.下面是一个如何做到这一点的例子:
import Vapor
extension QueryBuilder where Model: Fluent.Model {
@discardableResult
func filterIf(_ enableFilterIf: Bool, _ condition: (QueryBuilder<Model>) throws -> ()) rethrows -> Self {
if enableFilterIf {
try condition(self)
}
return self
}
}
下面是如何使用此扩展的方法:
return try await query
.filterIf(needFilter) { query in
query.filter(\.$board.$id == boardId)
}
.with(\.$profile)
.all()