com.fizzed.crux.util.ByteSize Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of crux-util Show documentation
The newest version!
package com.fizzed.crux.util;

import java.text.DecimalFormat;
import java.util.Objects;

public class ByteSize implements Comparable {

    final private long value;
    final private ByteSizeUnit displayUnit;

    public ByteSize(long value) {
        this(value, ByteSizeUnit.B);
    }

    public ByteSize(long value, ByteSizeUnit displayUnit) {
        Objects.requireNonNull(displayUnit, "displayUnit was null");
        this.value = value;
        this.displayUnit = displayUnit;
    }

    public ByteSizeUnit getDisplayUnit() {
        return displayUnit;
    }

    public long getValue() {
        return value;
    }

    public ByteSize withDisplayUnit(ByteSizeUnit displayUnit) {
        return new ByteSize(this.value, displayUnit);
    }

    public double asDouble(ByteSizeUnit unit) {
        return unit.fromByteSize(this.value);
    }

    public long asLong(ByteSizeUnit unit) {
        return (long)unit.fromByteSize(this.value);
    }

    static final DecimalFormat FMT = new DecimalFormat("#.####");

    @Override
    public String toString() {
        if (this.displayUnit == ByteSizeUnit.B) {
            return Long.toString(this.value);
        } else {
            double converted = this.displayUnit.fromByteSize(this.value);
//            String s = Double.toString(converted);
            String s = FMT.format(converted);
            if (s.endsWith(".0")) {
                s = s.substring(0, s.length()-2);
            }
            return s + this.displayUnit.getShortId();
        }
    }

    @Override
    public int compareTo(ByteSize o) {
        return Long.compare(this.value, o.value);
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (!(o instanceof ByteSize)) return false;

        ByteSize byteSize = (ByteSize) o;
        return value == byteSize.value;
    }

    @Override
    public int hashCode() {
        return (int) (value ^ (value >>> 32));
    }

    // helpers

    static public ByteSize parse(String s) {
        return parse(s, ByteSizeUnit.B);
    }

    static public ByteSize parse(String s, ByteSizeUnit defaultUnit) {
        // parse for first digit, then first non-digit
        int digitStartPos = -1;
        int unitStartPos = -1;

        for (int i = 0; i < s.length(); i++) {
            char c = s.charAt(i);
            if (Character.isDigit(c) || c == '.') {
                if (digitStartPos < 0) {
                    digitStartPos = i;
                } else {
                    // do nothing
                }
            } else {
                if (unitStartPos < 0) {
                    unitStartPos = i;       // done, we can exit early
                    break;
                } else {
                    // do nothing
                }
            }
        }

        if (digitStartPos < 0) {
            throw new IllegalArgumentException("Value '" + s + "' does not contain any digits");
        }

        String digitString = s.substring(digitStartPos, unitStartPos < 0 ? s.length() : unitStartPos).trim();
        ByteSizeUnit displayUnit = defaultUnit;

        if (unitStartPos > 0) {
            String unitString = s.substring(unitStartPos).trim();
            displayUnit = ByteSizeUnit.resolve(unitString);

            if (displayUnit == null) {
                throw new IllegalArgumentException("The part '" + unitString + "' in value '" + s + "' is not a valid byte size unit");
            }
        }

        long value;
        // as double or long?
        if (digitString.contains(".")) {
            double size = Double.parseDouble(digitString);
            value = displayUnit.toByteSize(size);
        } else {
            long size = Long.parseLong(digitString);
            value = displayUnit.toByteSize(size);
        }

        return new ByteSize(value, displayUnit);
    }

}