Examples with PGConnection org.postgresql.PGConnection used on opensource projects

Example 1 with PGConnection

use of org.postgresql.PGConnection in project sqlg by pietermartin.

the class PostgresDialect method streamSql.

@Override
public Writer streamSql(SqlgGraph sqlgGraph, String sql) {
    Connection conn = sqlgGraph.tx().getConnection();
    PGConnection pgConnection;
    try {
        pgConnection = conn.unwrap(PGConnection.class);
        OutputStream out = new PGCopyOutputStream(pgConnection, sql);
        return new OutputStreamWriter(out, "UTF-8");
    } catch (SQLException | UnsupportedEncodingException e) {
        throw new RuntimeException(e);
    }
}

Example 2 with PGConnection

use of org.postgresql.PGConnection in project activemq-artemis by apache.

the class PostgresSequentialSequentialFileDriver method createFile.

@Override
public void createFile(JDBCSequentialFile file) throws SQLException {
    synchronized (connection) {
        try {
            connection.setAutoCommit(false);
            LargeObjectManager lobjManager = ((PGConnection) connection).getLargeObjectAPI();
            long oid = lobjManager.createLO();
            createFile.setString(1, file.getFileName());
            createFile.setString(2, file.getExtension());
            createFile.setLong(3, oid);
            createFile.executeUpdate();
            try (ResultSet keys = createFile.getGeneratedKeys()) {
                keys.next();
                file.setId(keys.getLong(1));
            }
            connection.commit();
        } catch (SQLException e) {
            connection.rollback();
            throw e;
        }
    }
}

Example 3 with PGConnection

use of org.postgresql.PGConnection in project hale by halestudio.

the class DatabaseIT method testConnection.

/**
 * @throws SQLException if the connection cannot be obtained
 */
@Test
public void testConnection() throws SQLException {
    Connection connection = waitForConnection();
    try {
        assertTrue(connection instanceof PGConnection);
        assertTrue(connection.createStatement().execute("SELECT 1"));
    } finally {
        connection.close();
    }
}

Example 4 with PGConnection

use of org.postgresql.PGConnection in project hale by halestudio.

the class PostGISGeometries method configureGeometryColumnType.

@Override
public Class<? extends Geometry> configureGeometryColumnType(PGConnection connection, BaseColumn<?> column, DefaultTypeDefinition type) {
    Connection con = (Connection) connection;
    String columnValueName = column.getParent().getName();
    String geometryType = null;
    try {
        Statement stmt = con.createStatement();
        // Get the srid, dimension and geometry type
        ResultSet rs = stmt.executeQuery("SELECT srid,type,coord_dimension FROM geometry_columns WHERE f_table_name = " + "'" + columnValueName + "'");
        if (rs.next()) {
            geometryType = rs.getString("type");
            String dimension = rs.getString("coord_dimension");
            // Get the epsg code for the srid
            String srid = rs.getString("srid");
            ResultSet r = stmt.executeQuery("SELECT auth_srid, auth_name, srtext FROM spatial_ref_sys WHERE srid = " + srid);
            if (r.next()) {
                // Create Constraint to save the informations
                GeometryMetadata columnTypeConstraint = new GeometryMetadata(r.getString("auth_srid"), Integer.parseInt(dimension), r.getString("srtext"), r.getString("auth_name"));
                type.setConstraint(columnTypeConstraint);
            }
        } else {
        // XXX what if no SRID information is present? is that a case
        // that may still be valid?
        }
    } catch (SQLException e) {
        e.printStackTrace();
    }
    // In this case we have no geometry column information
    if (geometryType == null) {
        // use geometry even if no geometry column is present describing it
        return Geometry.class;
    }
    // return the geometryType
    if (geometryType.equalsIgnoreCase("MultiPolygon")) {
        return MultiPolygon.class;
    } else if (geometryType.equalsIgnoreCase("MultiPoint")) {
        return MultiPoint.class;
    } else if (geometryType.equalsIgnoreCase("MultiLineString")) {
        return MultiLineString.class;
    } else // TODO: shouldn't this be LineString instead?
    if (geometryType.equalsIgnoreCase("LinearRing")) {
        return LinearRing.class;
    } else if (geometryType.equalsIgnoreCase("Point")) {
        return Point.class;
    } else if (geometryType.equalsIgnoreCase("Polygon")) {
        return Polygon.class;
    } else {
        return Geometry.class;
    }
}

Example 5 with PGConnection

use of org.postgresql.PGConnection in project eventuate-local by eventuate-local.

the class PostgresWalClient method connectAndRun.

private void connectAndRun(Optional<BinlogFileOffset> binlogFileOffset, Consumer<EVENT> eventConsumer) throws SQLException, InterruptedException, IOException {
    countDownLatchForStop = new CountDownLatch(1);
    Properties props = new Properties();
    PGProperty.USER.set(props, user);
    PGProperty.PASSWORD.set(props, password);
    PGProperty.ASSUME_MIN_SERVER_VERSION.set(props, "9.4");
    PGProperty.REPLICATION.set(props, "database");
    PGProperty.PREFER_QUERY_MODE.set(props, "simple");
    connection = DriverManager.getConnection(url, props);
    PGConnection replConnection = connection.unwrap(PGConnection.class);
    LogSequenceNumber lsn = binlogFileOffset.flatMap(offset -> Optional.ofNullable(offset.getOffset()).map(LogSequenceNumber::valueOf)).orElse(LogSequenceNumber.valueOf("0/0"));
    stream = replConnection.getReplicationAPI().replicationStream().logical().withSlotName(replicationSlotName).withSlotOption("include-xids", false).withStatusInterval(replicationStatusIntervalInMilliseconds, TimeUnit.MILLISECONDS).withStartPosition(lsn).start();
    logger.info("connection to postgres wal succeed");
    while (running) {
        ByteBuffer messageBuffer = stream.readPending();
        if (messageBuffer == null) {
            logger.info("Got empty message, sleeping");
            TimeUnit.MILLISECONDS.sleep(walIntervalInMilliseconds);
            continue;
        }
        String messageString = extractStringFromBuffer(messageBuffer);
        logger.info("Got message: " + messageString);
        postgresWalMessageParser.parse(JSonMapper.fromJson(messageString, PostgresWalMessage.class), stream.getLastReceiveLSN().asLong(), replicationSlotName).forEach(eventConsumer);
        stream.setAppliedLSN(stream.getLastReceiveLSN());
        stream.setFlushedLSN(stream.getLastReceiveLSN());
    }
    try {
        stream.close();
        connection.close();
    } catch (SQLException e) {
        logger.error(e.getMessage(), e);
        throw new RuntimeException(e);
    }
    countDownLatchForStop.countDown();
}