一文教你利用Stream API批量Mock数据的方法

 更新时间:2022年09月05日 08:41:56   作者:鸭血粉丝Tang  
在日常开发的过程中我们经常会遇到需要mock一些数据的场景,比如说 mock 一些接口的返回或者说 mock 一些测试消息用于队列生产者发送消息。本文将教你如何通过 Stream API 批量 Mock 数据,需要的可以参考一下

在日常开发的过程中我们经常会遇到需要 mock 一些数据的场景,比如说 mock 一些接口的返回或者说 mock 一些测试消息用于队列生产者发送消息,可能很多时候我们都是使用一些固定的 case 或者一条相同的数据重复使用。今天阿粉就教大家用 Stream 去构造一些伪真实的一些数据。

Mock 任意个 UUID

首先我们通过普通写法来构造 100 个 UUID,代码如下相信大家都会写,就不多说了。

  public static List<UUID> listUUID(int size) {
    List<UUID> list = new ArrayList<>();
    for (int i = 0; i < size; i++) {
      UUID uuid = UUID.randomUUID();
      list.add(uuid);
    }
    return list;
  }

下面再提供 Stream 的写法,代码如下,一行搞定

  public static List<UUID> listUUID2(int size) {
    return Stream.generate(UUID::randomUUID).limit(size).collect(Collectors.toList());
  }

这里我们使用了 Stream 的 generate 方法,该方法接收一个 Supplier 类型的参数,Supplier 是一个功能接口,只有一个 get 方法,返回一个对象,不接收任何参数,上面我们就是通过 UUID 静态引用的方式获得一个 UUID 对象,另外我们使用 limit 方法来进行截断只获取 100 个。

Mock 消息

接下来我们再使用 Stream API 批量构造一批消息,作为队列的生产者进行数据发送

定义消息体

package com.example.demo.dto;

/**
 * <br>
 * <b>Function:</b><br>
 * <b>Author:</b>@author Java 极客技术<br>
 * <b>Date:</b>2022-09-03 11:49<br>
 * <b>Desc:</b>无<br>
 */
public class Message {
  int id;
  String message;

  public Message(int id, String message) {
    this.id = id;
    this.message = message;
  }

  @Override
  public String toString() {
    return "Message{" +
      "id=" + id +
      ", message='" + message + '\'' +
      '}';
  }
}

测试代码

  public static void main(String[] args) {
    List<Message> messages = genMessage(10);
    messages.forEach(System.out::println);
  }

  public static List<Message> genMessage(int size) {
    AtomicInteger atomicInteger = new AtomicInteger();
    Supplier<Message> supplier = () -> {
      Message message = new Message(new Random().nextInt(), "Message : " + atomicInteger.getAndIncrement());
      System.out.println("inner:" + message.toString());
      return message;
    };
    System.out.println(99);
    return Stream.generate(supplier).limit(size).collect(Collectors.toList());
  }

先看下运行结果,我们再来分析,可以看到第一个 case 我们是使用静态引用来返回一个 UUID 对象,这个 case 我们通过创建 lambda 表达式的形式来实现一个 Supplier,在表达式中我们进行 message 对象的构造,然后进行返回。其实上文的静态引用,本质上也是一个 lambda,所以跟下面的实现是一个原理,只不过是一些语法糖而已。

  public static List<UUID> listUUID2(int size) {
    Supplier<UUID> supplier = () -> UUID.randomUUID();
    return Stream.generate(supplier).limit(size).collect(Collectors.toList());
  }

如果对 Stream 流有理解的可以看到,我们这里有两个点需要注意,一个是我们这里的输出 99 是在 inner 之前的,另一个是我们这里使用的 limit 方法,不然会一直进行输出不会停止的,这两点其实都是流的基本特性,就不多说了。

Supplier 是个啥

上文提到 Stream  的 generate 方法接收的是一个 Supplier 类型的参数,那么这个 Supplier  是个啥呢?我们来仔细看一下。

package java.util.function;

@FunctionalInterface
public interface Supplier<T> {

    /**
     * Gets a result.
     *
     * @return a result
     */
    T get();
}

通过代码我们可以看到首先 Supplier 是个接口,既然是接口那就可以进行具体的实现,并且这个接口只有一个方法 get 返回指定的类型,同时该接口还有一个 @FunctionalInterface 注解,表名这个接口是一个函数是编程的接口,函数式接口是指仅仅只包含一个抽象方法的接口。

我们看到这个注解的 javadoc 里面大概的意思是这个注解是用来标识一个函数接口,函数式接口只有一个抽象方法,但是如果有 default 方法或者覆盖了 Object 的 public 方法都不算是抽象方法。还有一句讲的是函数式接口可以通过 lambda 表达式,方法引用或者构造方法引用来创建。我们上面的两个例子演示了 lambda 表达式和方法引用,构造函数其实也一样。

所以总结来说  Stream  的 generate 方法通过接收一个 Supplier  类型的参数来创建一个数据流,得到数据流以后就可以进行各种流的操作了。我们这篇文章更多的是通过 Stream 来构造 mock 数据,创建一个流,对于流的各种操作就不在本文的讨论范围之内了,阿粉之前也有相应的文章介绍过 Stream 感兴趣的小伙伴可以去翻翻看。

总结

工作中 mock 数据在很多场景都会遇到,但是可能很多时候我们都不会太关注 mock 的数据的形式,虽然说一个循环也可以 mock 到相应的数据,但是能写的优雅一点为什么我们不写的优雅一点呢?

以上就是一文教你利用Stream API批量Mock数据的方法的详细内容,更多关于Stream API批量Mock数据的资料请关注脚本之家其它相关文章!

相关文章

  • Mybatis-Plus 条件构造器 QueryWrapper 的基本用法

    Mybatis-Plus 条件构造器 QueryWrapper 的基本用法

    这篇文章主要介绍了Mybatis-Plus - 条件构造器 QueryWrapper 的使用,通过实例代码给大家介绍了查询示例代码及实现需求,代码简单易懂,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
    2021-09-09
  • java正则表达式获取指定HTML标签的指定属性值且替换的方法

    java正则表达式获取指定HTML标签的指定属性值且替换的方法

    下面小编就为大家带来一篇java正则表达式获取指定HTML标签的指定属性值且替换的方法。小编觉得挺不错的,现在就分享给大家,也给大家做个参考。一起跟随小编过来看看吧
    2016-12-12
  • spring data jpa开启批量插入、批量更新的问题解析

    spring data jpa开启批量插入、批量更新的问题解析

    这篇文章主要介绍了spring data jpa开启批量插入、批量更新问题,本文通过图文实例相结合给大家介绍的非常详细,需要的朋友可以参考下
    2021-07-07
  • Springboot之修改启动端口的两种方式(小结)

    Springboot之修改启动端口的两种方式(小结)

    这篇文章主要介绍了Springboot之修改启动端口的两种方式(小结),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-09-09
  • Java实现图形界面计算器

    Java实现图形界面计算器

    这篇文章主要为大家详细介绍了Java实现图形界面计算器,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2021-11-11
  • 使用Java通过OAuth协议验证发送微博的教程

    使用Java通过OAuth协议验证发送微博的教程

    这篇文章主要介绍了使用Java通过OAuth协议验证发送微博的教程,使用到了新浪微博为Java开放的API weibo4j,需要的朋友可以参考下
    2016-02-02
  • spring-shiro权限控制realm实战教程

    spring-shiro权限控制realm实战教程

    这篇文章主要介绍了spring-shiro权限控制realm实战教程,具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
    2021-10-10
  • YGC前后新生代是否变大分析详解

    YGC前后新生代是否变大分析详解

    要解释这个问题,我们先要弄清楚YGC的过程,parNew是新生代的gc算法,简单来说从gc roots开始扫描对象,当扫到一个只要是属于新生代的对象就将其挪到to space,但是老的对象还不会做释放,直到gc完成之后再看是否释放老的对象
    2022-01-01
  • Java对象的四种引用方式实例分析

    Java对象的四种引用方式实例分析

    这篇文章主要介绍了Java对象的四种引用方式,简单描述了四种引用方式的概念、应用场景并结合实例形式分析了弱引用所引用对象的垃圾回收过程,需要的朋友可以参考下
    2019-08-08
  • Java基于TCP协议socket网络编程的文件传送的实现

    Java基于TCP协议socket网络编程的文件传送的实现

    这篇文章主要介绍了Java基于TCP协议socket网络编程的文件传送的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12

最新评论