com.virtusa.gto.nyql.engine.transform.JdbcResultTransformer.groovy Maven / Gradle / Ivy

Go to download
package com.virtusa.gto.nyql.engine.transform

import com.virtusa.gto.nyql.engine.impl.NyQLResult
import groovy.transform.CompileStatic

@java.lang.SuppressWarnings('JdbcResultSetReference')
import java.sql.ResultSet
import java.sql.ResultSetMetaData
/**
 * @author IWEERARATHNA
 */
@CompileStatic
class JdbcResultTransformer implements QJdbcResultTransformer>> {

    List> apply(ResultSet resultSet) {
        try {
            ResultSetMetaData metaData = resultSet.getMetaData()
            int cc = metaData.columnCount
            Map cols = [:]
            for (int i = 1; i <= cc; i++) {
                cols.put(i, metaData.getColumnLabel(i))
            }

            NyQLResult nyQLResult = new NyQLResult()
            while (resultSet.next()) {
                Map row = [:]
                for (int i = 1; i <= cc; i++) {
                    row.put(cols[i], resultSet.getObject(i))
                }
                nyQLResult.add(row)
            }
            return nyQLResult

        } finally {
            resultSet.close()
        }
    }

    @Override
    long convertUpdateResult(long val) {
        val
    }
}