io.edurt.datacap.core.Utils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of datacap-driver-redis Show documentation
DataCap - Driver
The newest version!
package io.edurt.datacap.core;

import java.io.BufferedReader;
import java.io.StringReader;
import java.util.*;
import java.util.stream.Collectors;

public class Utils
{
    public static boolean isNumber(String str)
    {
        if (str == null || str.length() == 0) {
            return false;
        }
        for (int i = 0; i < str.length(); i++) {
            boolean digit = Character.isDigit(str.charAt(i));
            if (!digit) {
                return false;
            }
        }
        return true;
    }

    public static  List toList(T[] arr)
    {
        if (arr == null) {
            return null;
        }
        return Arrays.stream(arr)
                .collect(Collectors.toList());
    }

    public static Op parseSql(String rawSql)
    {
        // for IDEA database tool only
        if (rawSql.contains("SELECT 'keep alive'")) {
            return new Op(rawSql, null, "PING", new String[0]);
        }

        // hints
        List lines = new BufferedReader(new StringReader(rawSql))
                .lines()
                .collect(Collectors.toList());

        List hintLines = new ArrayList<>();
        List sqlLines = new ArrayList<>();
        lines.forEach(line -> {
            if (line.startsWith("--")) {
                hintLines.add(line);
            }
            else {
                sqlLines.add(line);
            }
        });

        List hints = hintLines
                .stream()
                .map(line -> {
                    String hintStr = line.replace("--", "")
                            .replaceAll(" ", "");
                    String[] arr = hintStr.split(":");
                    return new Hint(HintKey.fromString(arr[0]), arr[1]);
                }).collect(Collectors.toList());

        // sql to execute
        StringBuilder sb = new StringBuilder();
        sqlLines.forEach(sb::append);

        String sql = sb.toString();

        String[] arr = sql.split(" ");

        String commandString = arr[0];

        if (arr.length == 1) {
            return new Op(rawSql, hints, commandString, new String[0]);
        }
        else {
            String[] commandParams = Arrays.copyOfRange(arr, 1, arr.length);
            return new Op(rawSql, hints, commandString, commandParams);
        }
    }

    public static Map parseQueryStringToMap(String queryString)
    {
        String[] params = queryString.split("&");
        Map map = new HashMap<>();
        for (String param : params) {
            String[] p = param.split("=");
            if (p.length == 2) {
                map.put(p[0], p[1]);
            }
        }
        return map;
    }
}