FunTester java 网格输出的类--练习 stream

FunTester · 2020年04月19日 · 488 次阅读

在做测试的时候,经常需要把一些信息输出到控制台,但是格式上比较乱,想弄成一个类似 SQL 客户端的那个输出格式,在参考了一些资料后自己写了一个简单的控制台网格输出的类,分享代码供大家参考。

使用方法:暂时支持了 map 和 list 两种类型的数据展示,并没有提供 header 功能。

public static void main(String[] args) {

        List<String> ss0 = Arrays.asList("234", "432", "54");
        List<String> ss3 = Arrays.asList("234", "432", "54", "54", "54");
        List<String> ss1 = Arrays.asList("6546", "7675");
        Map<String, String> sss = new HashMap<>();
        sss.put(getNanoMark() + EMPTY, "fdsf");
        sss.put(getNanoMark() + EMPTY, "fdsfdsaff");
        sss.put(getNanoMark() + EMPTY, "fdsf");
        sss.put(getNanoMark() + EMPTY, "fdsfafdsf");
        sss.put(getNanoMark() + EMPTY, "fdsf");
        sss.put(getMark() + EMPTY, "fdsf");
        show(sss);
        List<List<String>> rows = Arrays.asList(ss1, ss3, ss0);
        show(rows);
        JSONObject json = new JSONObject();
        json.put("3234", 32432);
        json.put("323dsa4", 32432);
        json.put("3fdsa234", 32432);
        json.put("323fdsf4", 32432);
        json.put("32d34", 32432);
        json.put("32fdsafdf34", 32432);
        show(json);
    }

效果展示:

类代码如下(下一步优化其他数据类型和 header 以及边栏支持):

package com.fun.utils;

import com.fun.frame.SourceCode;
import net.sf.json.JSONObject;
import org.apache.commons.lang3.StringUtils;

import java.util.*;
import java.util.stream.Collectors;

public class ConsoleTable extends SourceCode {

    List<Integer> rowLength = new ArrayList<>();

    public static void show(Map map) {
        new ConsoleTable(map);
    }

    public static void show(List<List<String>> rows) {
        new ConsoleTable(rows);
    }

    /**
     * 输出map
     *
     * @param map
     */
    private ConsoleTable(Map map) {
        Set set = map.keySet();
        int asInt0 = set.stream().mapToInt(key -> key.toString().length()).max().getAsInt();
        rowLength.add(asInt0 + 2);
        List<String> values = new ArrayList<>();
        set.forEach(key -> values.add(map.get(key).toString()));
        int asInt1 = values.stream().mapToInt(value -> value.length()).max().getAsInt();
        rowLength.add(asInt1 + 2);
        StringBuffer stringBuffer = new StringBuffer(LINE + getHeader());
        map.forEach((k, v) -> {
            stringBuffer.append(getCel(0, k.toString()));
            stringBuffer.append(getCel(1, v.toString()));
        });
        output(stringBuffer.append(LINE + getHeader()).toString());
    }

    /**
     * 输出list
     *
     * @param rows
     */
    private ConsoleTable(List<List<String>> rows) {
        for (int i = 0; i < rows.size(); i++) {
            List<String> line = rows.get(i);
            for (int j = 0; j < line.size(); j++) {
                String s = line.get(j);
                if (rowLength.size() <= j) rowLength.add(0);
                if (rowLength.get(j) < s.length()) rowLength.set(j, s.length());
            }
        }
        rowLength = rowLength.stream().map(n -> n + 2).collect(Collectors.toList());
        StringBuffer stringBuffer = new StringBuffer(LINE + getHeader());
        for (int i = 0; i < rows.size(); i++) {
            List<String> line = rows.get(i);
            for (int j = 0; j < rowLength.size(); j++) {
                stringBuffer.append(getCel(j, j < line.size() ? line.get(j) : EMPTY));
            }
        }
        output(stringBuffer.append(LINE + getHeader()).toString());
    }


    /**
     * 获取每一格的string
     *
     * @param colum   
     * @param content 格内容
     * @return
     */
    public String getCel(int colum, String content) {
        Integer integer = rowLength.get(colum);
        int i = integer - content.length();
        return (colum == 0 ? LINE + PART : PART) + getManyString(SPACE_1, i / 2) + content + getManyString(SPACE_1, i - i / 2) + (rowLength.size() - colum == 1 ? PART : EMPTY);
    }

    /**
     * 获取头尾行
     *
     * @return
     */
    private String getHeader() {
        List<String> collect = rowLength.stream().map(size -> getManyString("-", size)).collect(Collectors.toList());
        return "+" + StringUtils.join(collect.toArray(), "+") + "+";
    }


}

这里使用了一些 stream 的用法,虽然比较生疏,但是效果很不错,stream 语法很强大,使得编码效率急速提升。在 groovy 语言使用 java 的 stream 各种方法时,还是遇到了不少的坑,目前主要还是符号兼容的问题比较多,所以尽量还是 java stream 的自己的语法比较好。

技术类文章精选

非技术文章精选

大咖风采

如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
暂无回复。
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册