fuzzycsv.nav.ExIterator.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.nav

import fuzzycsv.FuzzyCSVTable
import groovy.transform.CompileStatic
import groovy.transform.stc.ClosureParams
import groovy.transform.stc.FirstParam
import groovy.transform.stc.FromString

import java.util.concurrent.atomic.AtomicInteger

@CompileStatic
class IterHelper {
    static  T last(Iterator iter) {
        T tmp = null
        while (iter.hasNext()) tmp = iter.next()
        return tmp
    }

}

@CompileStatic
trait ExIterator implements Iterator {
    E last() {
        return IterHelper.last(this)
    }

    SELF skip() {
        return skip(1)
    }

    SELF skip(int steps) {

        if (!hasNext()) {
            throw new NoSuchElementException("no element to skip")
        }

        def j = this
        steps.times { Integer i ->
            j.next()
        }

        return this as SELF
    }
}


@CompileStatic
class NavIterator implements ExIterator {
    private static AtomicInteger i = new AtomicInteger()

    Navigator curr
    FuzzyCSVTable table
    FuzzyCSVTable pathTrack
    private boolean selfFinished = false
    private boolean markPath = false
    private Closure stopper
    private Closure next

    //internal use
    private Integer id = i.incrementAndGet()
    private int steps = 0

    static NavIterator from(Navigator curr, FuzzyCSVTable table = curr.table) {
        return new NavIterator(curr: curr.table(table), table: table)
    }

    NavIterator withStopper(@ClosureParams(value = FromString, options = ["fuzzycsv.FuzzyCSVTable", "fuzzycsv.Navigator"]) Closure stopper) {
        this.stopper = stopper
        return this
    }

    NavIterator withStepper(@ClosureParams(FirstParam.FirstGenericType) Closure next) {
        this.next = next
        return this
    }

    NavIterator markPath(FuzzyCSVTable t = table.copy()) {
        markPath = true
        pathTrack = t
        return this
    }

    @Override
    boolean hasNext() {
        if(!selfFinished && curr.inBounds(table)) return true
        return stopper(table, curr)
    }
    @Override
    Navigator next() {

        if (!selfFinished) {
            selfFinished = true
        } else {
            curr = next(curr)
        }

        if (markPath) curr?.mark("$id-${steps++}|", pathTrack)

        return curr
    }


}