一文教你使用stream流轻松解决java项目数据枚举显示问题

news2024/12/23 21:02:17

本篇文章主要讲解java枚举、vo实体类通过stream流形式输出枚举参数的方法。
日期：2024年5月26日
作者：任聪聪
本文代码实例附件：https://download.csdn.net/download/hj960511/89361611

实际效果

在这里插入图片描述

可以看到在枚举参数的基础上，我们增加了Text的显示类型。

实现思路

0.创建枚举类

package com.enumsDemo.enums;

public enum StatusEnums {
    HAPPY(0, "快乐"), 
    PLEASURE(1, "愉悦"), 
    LUCKY(2, "幸运");

    private int code;
    private String value;

    StatusEnums(int code, String value) {
        this.code = code;
        this.value = value;
    }

    public int getCode() {
        return code;
    }

    public String getValue() {
        return value;
    }

    public static StatusEnums getByCode(int code) {
        for (StatusEnums e : StatusEnums.values()) {
            if (e.getCode() == code) {
                return e;
            }
        }
        return null;
    }
}

1.创建pojo实体类，用于我们的主要数据的显示和设置

package com.enumsDemo.pojo;

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

@Data
@NoArgsConstructor
@AllArgsConstructor
public class User {

    private String id;
    private String name;
    private String age;
    private Integer Status;

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getAge() {
        return age;
    }

    public void setAge(String age) {
        this.age = age;
    }

    public Integer getStatus() {
        return Status;
    }

    public void setStatus(Integer status) {
        Status = status;
    }

    @Override
    public String toString() {
        return "List{" +
                "id='" + id + '\'' +
                ", name='" + name + '\'' +
                ", age='" + age + '\'' +
                ", Status='" + Status + '\'' +
                '}';
    }
}

2.创建vo类

package com.enumsDemo.pojo.vo;

import com.enumsDemo.pojo.User;
import lombok.Data;

@Data
public class UserVo extends User {

    private String statusText;

    public UserVo(User user) {
        super.setStatus(user.getStatus());
        super.setId(user.getId());
        super.setName(user.getName());
        super.setAge(user.getAge());
        this.statusText = com.enumsDemo.enums.StatusEnums.getByCode(user.getStatus()).getValue();
    }
}

3.api接口处输出

package com.enumsDemo.controller;

import com.enumsDemo.pojo.User;
import com.enumsDemo.pojo.vo.UserVo;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

@RestController
@RequestMapping("/api")
public class ApiController {
    //返回类型可自定义，如果自己有封装好数据格式只需要将listObjVo放入即可！
    @GetMapping("/list")
    public List<UserVo> list() {
        //模拟数据库查询出的数据类型
        List<User> listObj = new ArrayList<>();
        listObj.add(new User("1","张三","13",1));
        listObj.add(new User("2","李四","14",2));
        listObj.add(new User("3","王五","15",0));
        listObj.add(new User("4","任六","16",2));
        listObj.add(new User("5","赵七","17",1));
        //进入枚举处理逻辑
        List<UserVo> listObjVo = listObj.stream().map(UserVo::new).collect(Collectors.toList());
        //输出处理好的数据
        return listObjVo;
    }
}