/
util.clj
267 lines (224 loc) · 12.8 KB
/
util.clj
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
(ns methodical.util
"Utility functions for performing additional operations on multifns and their components not specified in one of the
interfaces. These functions are compositions of those methods."
(:refer-clojure :exclude [prefers prefer-method remove-all-methods])
(:require [methodical.impl.standard :as impl.standard]
[methodical.interface :as i]))
(defn multifn?
"True if `x` is a Methodical multifn (i.e., if it is an instance of `StandardMultiFn`)."
[x]
(impl.standard/multifn? x))
(defn primary-method
"Get the primary method *explicitly specified* for `dispatch-value`. This function does not return methods that would
otherwise still be applicable (e.g., methods for ancestor dispatch values) -- just the methods explicitly defined
for this exact match. (If you want methods that will be used, including those of ancestors dispatch values, you can
use [[applicable-primary-method]] or [[effective-primary-method]] instead.)
Note that the primary method will not have any implicit args (e.g. `next-method`) bound the way it normally would
when combined into an effective method; you will need to supply this yourself (or pass `nil` for no `next-method`)."
[multifn dispatch-val]
(get (i/primary-methods multifn) dispatch-val))
(defn matching-primary-methods
"Return a sequence of applicable primary methods for `dispatch-value`, sorted from most-specific to least-specific.
Methods include the `^:dispatch-valueue` with which they were defined as metadata. The standard dispatcher also checks
to make sure methods in the sequence are not ambiguously specific, replacing ambiguous methods with ones that will
throw an Exception when invoked."
([multifn dispatch-val]
(i/matching-primary-methods multifn multifn dispatch-val))
([dispatcher method-table dispatch-val]
(i/matching-primary-methods dispatcher method-table dispatch-val)))
(defn applicable-primary-method
"Return the primary method that would be use for `dispatch-value`, including ones from ancestor dispatch values or the
default dipsatch value. Method includes `^:dispatch-valueue` metadata indicating the actual dispatch value for which
the applicable method was defined.
Like `primary-method`, the method returned will not have any implicit args (such as `next-method`) bound."
[multifn dispatch-val]
(first (matching-primary-methods multifn dispatch-val)))
(defn effective-primary-method
"Build and effective method equivalent that would be used for this `dispatch-value` if it had no applicable auxiliary
methods. Implicit args (such as `next-method`) will be bound appropriately. Method has `^:dispatch-valueue` metadata
for the dispatch value with which the most-specific primary method was defined."
[multifn dispatch-val]
(let [[most-specific-primary-method :as primary-methods] (matching-primary-methods multifn dispatch-val)]
(some-> (i/combine-methods multifn primary-methods nil)
(with-meta (meta most-specific-primary-method)))))
(defn aux-methods
"Get all auxiliary methods *explicitly specified* for `dispatch-value`. This function does not include methods that
would otherwise still be applicable (e.g., methods for ancestor dispatch values) -- the methods explicitly defined
for this exact match.
* With 1 arg: methods come back as a map of `qualifier` -> `dispatch value` -> `[method]`.
* With 2 args: methods come back as a map of `qualifier` -> `[method]`.
* With 3 args: methods come back as sequence of `methods`."
([multifn]
(i/aux-methods multifn))
([multifn dispatch-val]
(let [qualifier->dispatch-val->fns (i/aux-methods multifn)]
(when (seq qualifier->dispatch-val->fns)
(into {} (for [[qualifier dispatch-val->fns] qualifier->dispatch-val->fns
:let [fns (get dispatch-val->fns dispatch-val)]
:when (seq fns)]
[qualifier fns])))))
([multifn qualifier dispatch-val]
(get-in (i/aux-methods multifn) [qualifier dispatch-val])))
(defn matching-aux-methods
"Return a map of aux method qualifier -> sequence of applicable methods for `dispatch-value`, sorted from
most-specific to least-specific. Methods should have the `^:dispatch-valueue` with which they were defined as
metadata."
([multifn dispatch-val]
(i/matching-aux-methods multifn multifn dispatch-val))
([dispatcher method-table dispatch-val]
(i/matching-aux-methods dispatcher method-table dispatch-val)))
(defn default-primary-method
"Get the default primary method associated with this `mutlifn`, if one exists."
[multifn]
(primary-method multifn (i/default-dispatch-value multifn)))
(defn default-aux-methods
"Get a map of aux qualifer -> methods for the default dispatch value, if any exist."
[multifn]
(aux-methods multifn (i/default-dispatch-value multifn)))
(defn default-effective-method
"Return the effective (combined) method for the default dispatch value, if one can be computed."
[multifn]
(i/effective-method multifn (i/default-dispatch-value multifn)))
(defn effective-dispatch-value
"Return the least-specific dispatch value that would return the same effective method as `dispatch-value`. e.g. if
`dispatch-value` is `Integer` and the effective method is a result of combining a `Object` primary method and a
`Number` aux method, the effective dispatch value is `Number`, since `Number` is the most specific thing out of the
primary and aux methods and would get the same effective method as `Integer`."
[multifn dispatch-val]
(:dispatch-value (meta (i/effective-method multifn dispatch-val))))
(defn dispatch-value
"Calculate the dispatch value that `multifn` will use given `args`."
;; since protocols can't define varargs, we have to wrap the `dispatch-value` method from the protocol and apply
;; varargs for > 4 args. The various < 4 args arities are there as an optimization because it's a little faster than
;; calling apply every time.
([multifn a] (i/dispatch-value multifn a))
([multifn a b] (i/dispatch-value multifn a b))
([multifn a b c] (i/dispatch-value multifn a b c))
([multifn a b c d] (i/dispatch-value multifn a b c d))
([multifn a b c d & more] (i/dispatch-value multifn a b c d more)))
(defn dispatch-fn
"Return a function that can be used to calculate dispatch values of given arg(s)."
[multifn]
(partial dispatch-value multifn))
(defn remove-all-primary-methods
"Remove all primary methods, for all dispatch values (including the default value), for this `multifn` or method
table."
[multifn]
(reduce
(fn [multifn dispatch-val]
(i/remove-primary-method multifn dispatch-val))
multifn
(keys (i/primary-methods multifn))))
(defn remove-all-aux-methods
"With one arg, remove *all* auxiliary methods for a `multifn`. With two args, remove all auxiliary methods for the
given `qualifier` (e.g. `:before`). With three args, remove all auxiliary methods for a given `qualifier` and
`dispatch-value`. "
([multifn]
(reduce remove-all-aux-methods multifn (keys (i/aux-methods multifn))))
([multifn qualifier]
(reduce
(fn [multifn dispatch-val]
(remove-all-aux-methods multifn qualifier dispatch-val))
multifn
(keys (get (i/aux-methods multifn) qualifier))))
([multifn qualifier dispatch-val]
(reduce
(fn [multifn f]
(i/remove-aux-method multifn qualifier dispatch-val f))
multifn
(get-in (i/aux-methods multifn) [qualifier dispatch-val]))))
;; TODO -- consider renaming to `remove-all-aux-methods-for-dispatch-val` for consistency with everything else
(defn remove-all-aux-methods-for-dispatch-val
"Remove all auxiliary methods for `dispatch-value` for *all* qualifiers."
[multifn dispatch-val]
(reduce
(fn [multifn qualifier]
(remove-all-aux-methods multifn qualifier dispatch-val))
multifn
(keys (i/aux-methods multifn))))
(defn remove-aux-method-with-unique-key
"Remove an auxiliary method that was added by [[add-aux-method-with-unique-key]], if one exists. Returns `multifn`."
[multifn qualifier dispatch-val unique-key]
{:pre [(some? multifn)]}
(if-let [method (some
(fn [method]
(when (= (::unique-key (meta method)) unique-key)
method))
(aux-methods multifn qualifier dispatch-val))]
(i/remove-aux-method multifn qualifier dispatch-val method)
multifn))
(defn add-aux-method-with-unique-key
"Adds an auxiliary method with a `unique-key` stored in its metadata. This unique key can later be used to remove the
auxiliary method with [[remove-aux-method-with-unique-key]]. If a method with this key already exists for this
qualifier and dispatch value, replaces the original."
[multifn qualifier dispatch-val f unique-key]
{:pre [(some? multifn)]}
(-> multifn
(remove-aux-method-with-unique-key qualifier dispatch-val unique-key)
(i/add-aux-method qualifier dispatch-val (vary-meta f assoc ::unique-key unique-key))))
(defn remove-all-methods
"Remove all primary and auxiliary methods, including default implementations."
[multifn]
(-> multifn remove-all-primary-methods remove-all-aux-methods))
;;;; #### Low-level destructive operations
(defn alter-var-root+
"Like [[clojure.core/alter-var-root]], but handles vars that are aliases of other vars, e.g. ones that have been
imported via Potemkin [[potemkin/import-vars]]."
[multifn-var f & args]
(let [{var-ns :ns, var-name :name} (meta multifn-var)
varr (if (and var-ns var-name)
(ns-resolve var-ns var-name)
multifn-var)]
(apply alter-var-root varr f args)))
(defn add-primary-method!
"Destructive version of [[add-primary-method]]. Operates on a var defining a Methodical multifn."
[multifn-var dispatch-val f]
(alter-var-root+ multifn-var i/add-primary-method dispatch-val f))
(defn remove-primary-method!
"Destructive version of [[remove-primary-method]]. Operates on a var defining a Methodical multifn."
[multifn-var dispatch-val]
(alter-var-root+ multifn-var i/remove-primary-method dispatch-val))
(defn remove-all-primary-methods!
"Destructive version of [[remove-all-primary-methods]]. Operates on a var defining a Methodical multifn."
[multifn-var]
(alter-var-root+ multifn-var remove-all-primary-methods))
(defn add-aux-method!
"Destructive version of [[add-aux-method]]. Operates on a var defining a Methodical multifn."
[multifn-var qualifier dispatch-val f]
(alter-var-root+ multifn-var i/add-aux-method qualifier dispatch-val f))
(defn remove-aux-method!
"Destructive version of [[remove-aux-method]]. Operates on a var defining a Methodical multifn."
[multifn-var qualifier dispatch-val f]
(alter-var-root+ multifn-var i/remove-aux-method qualifier dispatch-val f))
(defn remove-all-aux-methods!
"Destructive version of [[remove-all-aux-methods]]. Operates on a var defining a Methodical multifn."
([multifn-var]
(alter-var-root+ multifn-var remove-all-aux-methods))
([multifn-var qualifier]
(alter-var-root+ multifn-var remove-all-aux-methods qualifier))
([multifn-var qualifier dispatch-val]
(alter-var-root+ multifn-var remove-all-aux-methods qualifier dispatch-val)))
(defn remove-all-aux-methods-for-dispatch-val!
"Destructive version of [[remove-all-aux-methods-for-dispatch-val]]. Operates on a var defining a Methodical multifn."
[multifn-var dispatch-val]
(alter-var-root+ multifn-var remove-all-aux-methods-for-dispatch-val dispatch-val))
(defn add-aux-method-with-unique-key!
"Destructive version of [[add-aux-method-with-unique-key]]. Operates on a var defining a Methodical multifn."
[multifn-var qualifier dispatch-val f unique-key]
(alter-var-root+ multifn-var add-aux-method-with-unique-key qualifier dispatch-val f unique-key))
(defn remove-aux-method-with-unique-key!
"Destructive version of [[remove-aux-method-with-unique-key]]. Operates on a var defining a Methodical multifn."
[multifn-var qualifier dispatch-val unique-key]
(alter-var-root+ multifn-var remove-aux-method-with-unique-key qualifier dispatch-val unique-key))
(defn remove-all-methods!
"Destructive version of [[remove-all-methods]]. Operates on a var defining a Methodical multifn."
[multifn-var]
(alter-var-root+ multifn-var remove-all-methods))
(defn prefer-method!
"Destructive version of [[prefer-method]]. Operates on a var defining a Methodical multifn.
Note that vanilla Clojure [[clojure.core/prefer-method]] is actually itself destructive, so this function is
actually the Methodical equivalent of that function. `prefer-method!` is used by Methodical to differentiate the
operation from our nondestructive [[prefer-method]], which returns a copy of the multifn with an altered dispatch
table."
[multifn-var dispatch-val-x dispatch-val-y]
(alter-var-root+ multifn-var i/prefer-method dispatch-val-x dispatch-val-y))