资讯专栏INFORMATION COLUMN

如何以并发方式在同一个流上执行多种操作?--复制流

王晗 / 3561人阅读

摘要:正常情况下,一个流在执行一次终端操作之后便结束了。本文通过复制流内数据的方式,曲折的实现了同一个流上执行多次操作。只是思路,其性能并不一定高效,尤其是数据都在内存中处理时复制的开销很大。但如果流涉及大量,也许性能会有提高。

正常情况下,一个流在执行一次终端操作之后便结束了。本文通过复制流内数据的方式,曲折的实现了同一个流上执行多次操作。
Demo只是思路,其性能并不一定高效,尤其是数据都在内存中处理时复制的开销很大。但如果流涉及大量I/O,也许性能会有提高。

</>复制代码

  1. public class StreamForker {
  2. private final Stream stream;
  3. private final Map, ?>> forks = new HashMap<>();
  4. public StreamForker(Stream stream) {
  5. this.stream = stream;
  6. }
  7. public StreamForker fork(Object key, Function, ?> f) {
  8. forks.put(key, f);
  9. return this;
  10. }
  11. public Results getResults() {
  12. ForkingStreamConsumer consumer = build();
  13. try {
  14. stream.sequential().forEach(consumer);
  15. } finally {
  16. consumer.finish();
  17. }
  18. return consumer;
  19. }
  20. private ForkingStreamConsumer build() {
  21. List> queues = new ArrayList<>();
  22. Map> actions = forks.entrySet().stream().reduce(new HashMap>(),
  23. (map, e) -> {
  24. map.put(e.getKey(), getOperationResult(queues, e.getValue()));
  25. return map;
  26. }, (m1, m2) -> {
  27. m1.putAll(m2);
  28. return m1;
  29. });
  30. return new ForkingStreamConsumer<>(queues, actions);
  31. }
  32. private Future getOperationResult(List> queues, Function, ?> f) {
  33. BlockingQueue queue = new LinkedBlockingQueue<>();
  34. queues.add(queue);
  35. Spliterator spliterator = new BlockingQueueSpliterator<>(queue);
  36. Stream source = StreamSupport.stream(spliterator, false);
  37. return CompletableFuture.supplyAsync(() -> f.apply(source));
  38. }
  39. }

accept方法将原始流中所有的数据添加到各个BlockingQueue内,此处实现了复制

</>复制代码

  1. class ForkingStreamConsumer implements Consumer, Results {
  2. static final Object END_OF_STREAM = new Object();
  3. private final List> queues;
  4. private final Map> actions;
  5. public ForkingStreamConsumer(List> queues, Map> actions) {
  6. this.queues = queues;
  7. this.actions = actions;
  8. }
  9. @Override
  10. public void accept(T t) {
  11. queues.forEach(q -> q.add(t));
  12. }
  13. @SuppressWarnings("unchecked")
  14. void finish() {
  15. accept((T) END_OF_STREAM);
  16. }
  17. @SuppressWarnings("unchecked")
  18. @Override
  19. public R get(Object key) {
  20. try {
  21. return ((Future) actions.get(key)).get();
  22. } catch (Exception e) {
  23. throw new RuntimeException(e);
  24. }
  25. }
  26. }

此处重写了tryAdvance接口,只是简单的从BlockingQueue中取出数据,执行action。业务逻辑中复制流是为了做什么事情,action就是这件事情。ForkingStreamConsumer.END_OF_STREAM是Queue中数据结束的标示

</>复制代码

  1. class BlockingQueueSpliterator implements Spliterator {
  2. private final BlockingQueue q;
  3. BlockingQueueSpliterator(BlockingQueue q) {
  4. this.q = q;
  5. }
  6. @Override
  7. public boolean tryAdvance(Consumer action) {
  8. T t;
  9. while (true) {
  10. try {
  11. t = q.take();
  12. break;
  13. } catch (InterruptedException e) {
  14. }
  15. }
  16. if (t != ForkingStreamConsumer.END_OF_STREAM) {
  17. action.accept(t);
  18. return true;
  19. }
  20. return false;
  21. }
  22. @Override
  23. public Spliterator trySplit() {
  24. return null;
  25. }
  26. @Override
  27. public long estimateSize() {
  28. return 0;
  29. }
  30. @Override
  31. public int characteristics() {
  32. return 0;
  33. }
  34. }

文章版权归作者所有,未经允许请勿转载,若此文章存在违规行为,您可以联系管理员删除。

转载请注明本文地址:https://www.ucloud.cn/yun/77289.html

相关文章

  • 从命令式到响应式(五)

    摘要:输出流只有在所有的输入流都完成以后才能完成,任何一条输入流上的错误都将立即推送到输出流上。如果没有转入输入流,输出流将会立即发出结束通知。返回值以数组形式获取到的每一个输入流的值,或者来自映射函数的值。返回值仅从最新的内部流上取值的流。 接着上一节的操作符继续,让我们直奔主题。 组合类操作符 组合类的操作符可以将不同流数据按一定的规则进行合并,从而获得所需要的完整数据。 combine...

    CoderBear 评论0 收藏0
  • 从命令式到响应式(四)

    摘要:使用的操作符这条从左到右的横线代表经过操作符转换后的输出流。返回值通过判定函数检测的值组成的流。返回值持续发出输入流上的值,直到通知流上发出值为止。 上期介绍过了rxjs中的三大件,Observable,subscription,subject,但是在开发过程我们最常接触到的东西非操作符莫属。比如上期代码中曾出现过的from就是一个操作符。rxjs中的操作符大致上可以分为几类,创建类,...

    jaysun 评论0 收藏0
  • 巧妙复制一个

    摘要:场景实际业务中可能出现重复消费一个可读流的情况,比如在前置过滤器解析请求体,拿到进行相关权限及身份认证认证通过后框架或者后置过滤器再次解析请求体传递给业务上下文。 场景 实际业务中可能出现重复消费一个可读流的情况,比如在前置过滤器解析请求体,拿到body进行相关权限及身份认证;认证通过后框架或者后置过滤器再次解析请求体传递给业务上下文。因此,重复消费同一个流的需求并不奇葩,这类似于js...

    wenzi 评论0 收藏0
  • 探索 RxJS - Core Concept

    摘要:但不同的是,在的遍历调用过程中,如果一个事件还没有触发完毕获取到返回值,就触发了下一个事件,则将忽略返回的值。这样,我们就可以避免异步的返回值因为返回较慢,反而覆盖了之后异步的返回值。 Steam in ReactiveX showImg(https://segmentfault.com/img/bVFReX?w=100&h=100); ReactiveX,又称 Reactive Ex...

    Neilyo 评论0 收藏0

发表评论

0条评论

王晗

|高级讲师

TA的文章

阅读更多
最新活动
阅读需要支付1元查看
<