fuzzycsv.FuzzyCSVUtils.groovy Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of fuzzy-csv Show documentation
A groovy/java tabular Data (from CSV,SQL,JSON) processing library that supports fuzzy column matching,tranformations/merging/querying etc
There is a newer version: 1.9.1-groovy4
Show newest version
package fuzzycsv

import groovy.time.TimeCategory
import groovy.transform.CompileStatic
import org.slf4j.Logger
import org.slf4j.LoggerFactory

@CompileStatic
class FuzzyCSVUtils {

    private static Logger log = LoggerFactory.getLogger(FuzzyCSVUtils)


    static List toNumbers(List list) {
        def rt = list.collect { Object it -> coerceToNumber(it) }
        return rt
    }

    static Object safeGet(List it, int idx) {
        if (idx < 0 || it == null) return null
        return it.size() > idx ? it.get(idx) : null
    }

    static List moveElem2Idx(List h, from, int idx) {
        move(h, h.indexOf(from), idx)
    }

    static List moveElems(List h, from, to) {
        move(h, h.indexOf(from), h.indexOf(to))
    }

    static List move(List h, int from, int to) {
        def get = h.get(from)
        h.remove(from)
        h.add(to, get)
        return h
    }

    static  List replace(List source, T from, T to) {
        def idx = source.indexOf(from)
        if (idx != -1)
            source.set(idx, to)
        return source
    }

    static Number coerceToNumber(obj, Class preferredType = Integer) {
        toNumber(obj, false, preferredType)
    }


    private static Number toNumber(obj, boolean strict = true, Class preferredType = Integer) {
        if (obj == null)
            return preferredType == Integer || preferredType == BigInteger ? 0 : 0.0

        if (obj instanceof Aggregator)
            obj = obj.getValue()

        if (obj instanceof Number)
            return obj as Number

        if (obj instanceof Boolean)
            return obj ? 1.toBigInteger() : 0.toBigDecimal()

        def strValue = obj.toString()

        try {
            return Integer.parseInt(strValue) as Number
        } catch (Exception x) {
        }

        try {
            return Double.parseDouble(strValue) as Number
        } catch (Exception x) {
            def msg = "FuzzyCSVUtils:toNumbers() Could not convert [$strValue] to a Number ($x)"

            if (strict)
                throw new NumberFormatException(msg)
            else
                log.trace(msg)
        }

        return preferredType == Integer || preferredType == BigInteger ? 0 : 0.0
    }


    static  T time(String name, Closure worker) {
        def padding = repeat('    ', IndentHelper.get())
        IndentHelper.increment()
        println "$padding ### Task: {$name}..."
        def start = System.currentTimeMillis()
        try {
            def rt = worker.call()
            def stop = System.currentTimeMillis()

            def time = TimeCategory.minus(new Date(stop), new Date(start))
            println "$padding -----> Completed in {$name} in ${time}".toString()

            return rt
        } finally {
            IndentHelper.decrement()
        }

    }

    //added cause groovy 3 multiply now uses charsequence
    private static String repeat(String self, Number factor) {
        int size = factor.intValue();
        if (size == 0)
            return "";
        else if (size < 0) {
            throw new IllegalArgumentException("multiply() should be called with a number of 0 or greater not: " + size);
        }
        StringBuilder answer = new StringBuilder(self);
        for (int i = 1; i < size; i++) {
            answer.append(self);
        }
        return answer.toString();
    }

    static void closeQuietly(Closeable c) {
        if (c != null) {
            try {
                c.close()
            } catch (IOException e) {
                /* ignore */
            }
        }
    }

    static void closeQuietly(AutoCloseable c) {
        if (c != null) {
            try {
                c.close()
            } catch (IOException e) {
                /* ignore */
            }
        }
    }

    static Map toProperties(Object o) {
        final Map properties = o.properties
        properties.remove("class")
        return properties
    }

}

class IndentHelper {
    private static ThreadLocal indent = new ThreadLocal() {
        @Override
        protected Integer initialValue() {
            return 0
        }
    }

    static increment() {
        indent.set(++indent.get())
    }

    static decrement() {
        indent.set(--indent.get())
    }

    static Integer get() {
        indent.get()
    }

    static clear() {
        indent.remove()
    }
}