1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
/*-------------------------------------------------------------------------
*
* Copyright (c) 2009, PostgreSQL Global Development Group
*
* IDENTIFICATION
* $PostgreSQL: pgjdbc/org/postgresql/core/v3/CopyInImpl.java,v 1.1 2009/07/01 05:00:40 jurka Exp $
*
*-------------------------------------------------------------------------
*/
package org.postgresql.core.v3;
import java.sql.SQLException;
import org.postgresql.copy.CopyIn;
/**
* Anticipated flow of a COPY FROM STDIN operation:
*
* CopyManager.copyIn()
* ->QueryExecutor.startCopy()
* - sends given query to server
* ->processCopyResults():
* - receives CopyInResponse from Server
* - creates new CopyInImpl
* ->initCopy():
* - receives copy metadata from server
* ->CopyInImpl.init()
* ->lock() connection for this operation
* - if query fails an exception is thrown
* - if query returns wrong CopyOperation, copyIn() cancels it before throwing exception
* <-return: new CopyInImpl holding lock on connection
* repeat CopyIn.writeToCopy() for all data
* ->CopyInImpl.writeToCopy()
* ->QueryExecutorImpl.writeToCopy()
* - sends given data
* ->processCopyResults()
* - parameterized not to block, just peek for new messages from server
* - on ErrorResponse, waits until protocol is restored and unlocks connection
* CopyIn.endCopy()
* ->CopyInImpl.endCopy()
* ->QueryExecutorImpl.endCopy()
* - sends CopyDone
* - processCopyResults()
* - on CommandComplete
* ->CopyOperationImpl.handleCommandComplete()
* - sets updatedRowCount when applicable
* - on ReadyForQuery unlock() connection for use by other operations
* <-return: CopyInImpl.getUpdatedRowCount()
*/
public class CopyInImpl extends CopyOperationImpl implements CopyIn {
public void writeToCopy(byte[] data, int off, int siz) throws SQLException {
queryExecutor.writeToCopy(this, data, off, siz);
}
public void flushCopy() throws SQLException {
queryExecutor.flushCopy(this);
}
public long endCopy() throws SQLException {
return queryExecutor.endCopy(this);
}
}
|
