Count unedifact segments

UnedifactUtil.groovy

import org.milyn.smooks.edi.EDIWritable

class UnedifactUtil {

    public static int countSegments(final EDIWritable origmessage) {

        def isSegmentGroup = { it?.class?.canonicalName?.contains('SegmentGroup') }

        def countInDepthSegments
        countInDepthSegments = { EDIWritable message ->

            def notNullFieldValues = message
                    .getClass()
                    .getDeclaredFields()
                    .collect( { message."$it.name" } )
                    .grep({ it })

            def firstLevelEdifactSegments = notNullFieldValues
                    .grep({it instanceof EDIWritable})

            def inListsEdifactSegments = notNullFieldValues
                    .grep({it instanceof List})
                    .flatten()
                    .grep({ !isSegmentGroup(it) })

            def inDepthEdifactSegments = notNullFieldValues
                    .grep({it instanceof List})
                    .flatten()
                    .grep(isSegmentGroup)
                    .collect(countInDepthSegments)
                    .flatten()

            firstLevelEdifactSegments + inListsEdifactSegments + inDepthEdifactSegments
        }

        origmessage ? countInDepthSegments(origmessage).size() : 0

    }
}