Package org.postgresql.copy

Examples of org.postgresql.copy.PGCopyInputStream

31
32
33
34
35
36
37
38
39
40
41
    }
   
    public void testReadBytesCorrectlyHandlesEof() throws SQLException, IOException {
      insertSomeData();
     
      sut = new PGCopyInputStream((PGConnection) _conn, "COPY cpinstreamtest (i) TO STDOUT WITH (FORMAT CSV, HEADER false)");
     
      byte[] buf = new byte[100]; // large enough to read everything on the next step
      assertTrue(sut.read(buf) > 0);
     
      assertEquals(-1, sut.read(buf));
View Full Code Here
42
43
44
45
46
47
48
49
50
51
52
    }

    public void testReadBytesCorrectlyReadsDataInChunks() throws SQLException, IOException {
      insertSomeData();
     
      sut = new PGCopyInputStream((PGConnection) _conn, "COPY (select i from cpinstreamtest order by i asc) TO STDOUT WITH (FORMAT CSV, HEADER false)");
     
      byte[] buf = new byte[2]; // small enough to read in multiple chunks
      StringBuilder result = new StringBuilder(100);
      int chunks = 0;
      while (sut.read(buf) > 0) {
View Full Code Here
59
60
61
62
63
64
65
66
67
68
69
    }
   
    public void testStreamCanBeClosedAfterReadUp() throws SQLException, IOException {
      insertSomeData();
     
      sut = new PGCopyInputStream((PGConnection) _conn, "COPY (select i from cpinstreamtest order by i asc) TO STDOUT WITH (FORMAT CSV, HEADER false)");
     
      byte[] buff = new byte[100];
    while(sut.read(buff) > 0);
   
    sut.close();
View Full Code Here
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
//    }    
  }
 
  @Override
  public void read(ILineSender lineSender){
    PGCopyInputStream inputStream = null;
    if(sql.isEmpty()){
      logger.error("Sql for GreenplumReader is empty.");
      throw new WormholeException("Sql for GreenplumReader is empty.",JobStatus.READ_FAILED.getStatus()+ERROR_CODE_ADD); 
    }
    logger.info(String.format("GreenplumReader start to query %s .", sql));
    for(String sqlItem:sql.split(";")){
      sqlItem = sqlItem.trim();
      if(sqlItem.isEmpty()) {
        continue;
      }
      try {
        inputStream = new PGCopyInputStream((PGConnection) ((DelegatingConnection) conn).getInnermostDelegate(), sqlItem);
        fetchData(lineSender,inputStream);
      } catch (Exception e) {       
        logger.error("GreenplumReader error");
        WormholeException ex = new WormholeException(e,JobStatus.READ_FAILED.getStatus());
        if(e instanceof WormholeException) {
View Full Code Here
TOP

Related Classes of org.postgresql.copy.PGCopyInputStream

  • com.dp.nebula.wormhole.plugins.reader.greenplumreader.GreenplumReader
  • org.postgresql.test.jdbc4.PGCopyInputStreamTest
  • org.postgresql.util.PSQLException
  • java.io.IOException
    • Copyright © 2018 www.massapicom. All rights reserved.
    All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.