ai.digital.integration.server.util.PostgresDbUtil.groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of integratoin-server-gradle-plugin Show documentation

The easy way to get custom setup for Deploy up and running

The newest version!

package ai.digital.integration.server.util

import java.sql.ResultSet
import java.sql.SQLException
import java.sql.Statement

class PostgresDbUtil {
    def static resetSequences(logger, connection) {
        try {
            Statement seqStmt = connection.createStatement()
            seqStmt.closeOnCompletion()
            ResultSet rs = seqStmt.executeQuery("SELECT c.relname FROM pg_class c WHERE c.relkind = 'S';")
            while (rs.next()) {
                String sequence = rs.getString('relname')
                String table = sequence.replace("_ID_seq", "")
                Statement updStmt = connection.createStatement()
                updStmt.closeOnCompletion()
                updStmt.executeQuery("SELECT SETVAL('\"${sequence}\"', (SELECT MAX(\"ID\")+1 FROM \"${table}\"));")
            }
        } catch (SQLException e) {
            logger.error('Error occurred while resetting sequences.')
            e.printStackTrace()
            throw e
        }
    }
}