org.vertexium.util.MultiFieldStringEncoder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vertexium-core Show documentation
There is a newer version: 4.10.0
package org.vertexium.util;

import org.vertexium.VertexiumException;

import java.io.UnsupportedEncodingException;
import java.net.URLDecoder;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.time.Instant;
import java.time.ZoneId;
import java.time.ZonedDateTime;
import java.util.regex.Pattern;

public class MultiFieldStringEncoder {
    private static final String ENC = StandardCharsets.UTF_8.toString();
    private final String separator;
    private final String separatorRegex;
    private final Integer expectedPartCount;

    public MultiFieldStringEncoder(String separator, Integer expectedPartCount) {
        this.separator = separator;
        this.separatorRegex = Pattern.quote(separator);
        this.expectedPartCount = expectedPartCount;
    }

    public String[] decode(String str) {
        String[] parts = str.split(separatorRegex);
        if (expectedPartCount != null && parts.length != expectedPartCount) {
            throw new VertexiumException(String.format("Expected %d parts found %d in string \"%s\" while looking for \"%s\"", expectedPartCount, parts.length, str, separator));
        }
        for (int i = 0; i < parts.length; i++) {
            parts[i] = decodeField(parts[i]);
        }
        return parts;
    }

    private String decodeField(String field) {
        try {
            return URLDecoder.decode(field, ENC);
        } catch (UnsupportedEncodingException e) {
            throw new VertexiumException("Bad encoder", e);
        }
    }

    public static ZonedDateTime timestampFromString(String str, ZoneId zoneId) {
        return ZonedDateTime.ofInstant(Instant.ofEpochMilli(Long.parseLong(str, 16)), zoneId);
    }

    public static String timestampToString(ZonedDateTime timestamp) {
        return String.format("%08x", timestamp.toInstant().toEpochMilli());
    }

    public String encode(String... fields) {
        if (expectedPartCount != null && fields.length != expectedPartCount) {
            throw new VertexiumException(String.format("Expected %d parts found %d", expectedPartCount, fields.length));
        }
        StringBuilder result = new StringBuilder();
        for (String field : fields) {
            if (result.length() > 0) {
                result.append(separator);
            }
            result.append(encodeField(field));
        }
        return result.toString();
    }

    private String encodeField(String field) {
        try {
            return URLEncoder.encode(field, ENC);
        } catch (UnsupportedEncodingException e) {
            throw new VertexiumException("Bad encoder", e);
        }
    }
}