[英]Clojure: implementing stateful Java interface
Kafka Streams有一个接口, Processor
,其实现是有状态的。 开发人员指南中给出的示例实现是:
public class WordCountProcessor implements Processor<String, String> {
private ProcessorContext context;
private KeyValueStore<String, Long> kvStore;
@Override
@SuppressWarnings("unchecked")
public void init(ProcessorContext context) {
// keep the processor context locally because we need it in punctuate() and commit()
this.context = context;
// call this processor's punctuate() method every 1000 time units.
this.context.schedule(1000);
// retrieve the key-value store named "Counts"
kvStore = (KeyValueStore) context.getStateStore("Counts");
}
@Override
public void process(String dummy, String line) {
String[] words = line.toLowerCase().split(" ");
for (String word : words) {
Long oldValue = kvStore.get(word);
if (oldValue == null) {
kvStore.put(word, 1L);
} else {
kvStore.put(word, oldValue + 1L);
}
}
}
@Override
public void punctuate(long timestamp) {
KeyValueIterator<String, Long> iter = this.kvStore.all();
while (iter.hasNext()) {
KeyValue<String, Long> entry = iter.next();
context.forward(entry.key, entry.value.toString());
}
iter.close();
// commit the current processing progress
context.commit();
}
@Override
public void close() {
// close the key-value store
kvStore.close();
}
}
init
方法初始化WordCountProcessor
的内部状态,例如检索键值存储。 其他方法(如process
和close
)使用此状态。
这不是我清楚如何reify
Clojure中这样的接口。 我们如何将init
检索到的状态传递给process
, close
等?
我有一个想法是使用一个闭包:
(let [ctx (atom nil)]
(reify Processor
(close [this]
;; Do something w/ ctx
)
(init [this context]
(reset! ctx context))
(process [this k v]
;; Do something w/ ctx
)
(punctuate [this timestamp]
;; Do something w/ ctx
)))
令人讨厌的是,我们每次都必须从ProcessorContext
对象开始,因此键值存储代码将在需要键值存储的所有方法中重复。
我没有看到(通用)方法,虽然根据具体情况,我们可以用方法所需的更具体的状态替换ctx
原子。
有没有更好的办法?
关闭原子将是实现它的主要方式。 您的原始类有两个字段,因此您可以关闭两个原子以获得相同的效果
(let [ctx (atom nil)
kv-store (atom nil)]
(reify Processor
,,,
(init [this context]
(reset! ctx context)
(reset! kv-store (.getStateStore context "Counts")))
,,,))
如果这仍然太乏味,那么你可以添加一些便利函数,这些函数也会关闭原子
(let [ctx (atom nil)
kv-store (atom nil)]
(def kv-get [key]
(.get @kv-store key))
(def kv-all []
(iterator-seq (.all @kv-store)))
(def kv-put [key value]
(.put @kv-store key value))
(reify Processor
,,,
(init [this context]
(reset! ctx context)
(reset! kv-store (.getStateStore context "Counts")))
,,,
(punctuate [this timestamp]
(do-seq [x (kv-all)]
,,,)
)))
替代方案是使用gen-class ,但认为你会更好地使用reify。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.