-
Notifications
You must be signed in to change notification settings - Fork 129
/
PredicateBindings.swift
55 lines (49 loc) · 1.9 KB
/
PredicateBindings.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
//===----------------------------------------------------------------------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2022-2023 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See https://swift.org/LICENSE.txt for license information
// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//
@_nonSendable
@available(macOS 9999, iOS 9999, tvOS 9999, watchOS 9999, *)
public struct PredicateBindings {
// Store as a values as an array instead of a dictionary (since it is almost always very few elements, this reduces heap allocation and hashing overhead)
private var storage: [(id: PredicateExpressions.VariableID, value: Any)]
public init<each T>(_ value: repeat (PredicateExpressions.Variable<each T>, each T)) {
storage = []
_ = (repeat storage.append(((each value).0.key, (each value).1)))
}
public subscript<T>(_ variable: PredicateExpressions.Variable<T>) -> T? {
get {
storage.first {
$0.id == variable.key
}?.value as? T
}
set {
let found = storage.firstIndex {
$0.id == variable.key
}
guard let newValue else {
if let found {
storage.remove(at: found)
}
return
}
if let found {
storage[found].value = newValue
} else {
storage.append((variable.key, newValue))
}
}
}
public func binding<T>(_ variable: PredicateExpressions.Variable<T>, to value: T) -> Self {
var mutable = self
mutable[variable] = value
return mutable
}
}