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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
/*-------------------------------------------------------------------------
*
* Copyright (c) 2004-2008, PostgreSQL Global Development Group
* Copyright (c) 2004, Open Cloud Limited.
*
* IDENTIFICATION
* $PostgreSQL: pgjdbc/org/postgresql/core/v2/FastpathParameterList.java,v 1.10 2008/01/08 06:56:27 jurka Exp $
*
*-------------------------------------------------------------------------
*/
package org.postgresql.core.v2;
import org.postgresql.core.*;
import java.io.InputStream;
import java.io.IOException;
import java.sql.SQLException;
import java.util.Arrays;
import org.postgresql.util.PSQLException;
import org.postgresql.util.PSQLState;
import org.postgresql.util.StreamWrapper;
import org.postgresql.util.GT;
/**
* Implementation of fastpath parameter lists for the V2 protocol.
* The V2 protocol expects different representations of parameters in
* queries to that used in fastpath calls, so we do a separate implementation
* which supports only what fastpath needs here.
*
* @author Oliver Jowett (oliver@opencloud.com)
*/
class FastpathParameterList implements ParameterList {
FastpathParameterList(int paramCount) {
this.paramValues = new Object[paramCount];
}
public void registerOutParameter(int index, int sqlType ){}
public void registerOutParameter(int index, int sqlType, int precision ){};
public int getInParameterCount() {
return paramValues.length;
}
public int getOutParameterCount()
{
return 0;
}
public int getParameterCount()
{
return paramValues.length;
}
public int[] getTypeOIDs() {
return null;
}
public void setIntParameter(int index, int value) throws SQLException {
if (index < 1 || index > paramValues.length)
throw new PSQLException(GT.tr("The column index is out of range: {0}, number of columns: {1}.", new Object[]{new Integer(index), new Integer(paramValues.length)}), PSQLState.INVALID_PARAMETER_VALUE );
byte[] data = new byte[4];
data[3] = (byte)value;
data[2] = (byte)(value >> 8);
data[1] = (byte)(value >> 16);
data[0] = (byte)(value >> 24);
paramValues[index - 1] = data;
}
public void setLiteralParameter(int index, String value, int oid) throws SQLException {
// Not enough type info here for the V2 path (which requires binary reprs)
throw new IllegalArgumentException("can't setLiteralParameter() on a fastpath parameter");
}
public void setStringParameter(int index, String value, int oid) throws SQLException {
paramValues[index - 1] = value;
}
public void setBytea(int index, byte[] data, int offset, int length) throws SQLException {
if (index < 1 || index > paramValues.length)
throw new PSQLException(GT.tr("The column index is out of range: {0}, number of columns: {1}.", new Object[]{new Integer(index), new Integer(paramValues.length)}), PSQLState.INVALID_PARAMETER_VALUE );
paramValues[index - 1] = new StreamWrapper(data, offset, length);
}
public void setBytea(int index, final InputStream stream, final int length) throws SQLException {
if (index < 1 || index > paramValues.length)
throw new PSQLException(GT.tr("The column index is out of range: {0}, number of columns: {1}.", new Object[]{new Integer(index), new Integer(paramValues.length)}), PSQLState.INVALID_PARAMETER_VALUE );
paramValues[index - 1] = new StreamWrapper(stream, length);
}
public void setNull(int index, int oid) throws SQLException {
throw new IllegalArgumentException("can't setNull() on a v2 fastpath parameter");
}
public String toString(int index) {
if (index < 1 || index > paramValues.length)
throw new IllegalArgumentException("parameter " + index + " out of range");
return "<fastpath parameter>";
}
private void copyStream(PGStream pgStream, StreamWrapper wrapper) throws IOException {
byte[] rawData = wrapper.getBytes();
if (rawData != null)
{
pgStream.Send(rawData, wrapper.getOffset(), wrapper.getLength());
return ;
}
pgStream.SendStream(wrapper.getStream(), wrapper.getLength());
}
void writeV2FastpathValue(int index, PGStream pgStream) throws IOException {
--index;
if (paramValues[index] instanceof StreamWrapper)
{
StreamWrapper wrapper = (StreamWrapper)paramValues[index];
pgStream.SendInteger4(wrapper.getLength());
copyStream(pgStream, wrapper);
}
else if (paramValues[index] instanceof byte[])
{
byte[] data = (byte[])paramValues[index];
pgStream.SendInteger4(data.length);
pgStream.Send(data);
}
else if (paramValues[index] instanceof String)
{
byte[] data = pgStream.getEncoding().encode((String)paramValues[index]);
pgStream.SendInteger4(data.length);
pgStream.Send(data);
}
else
{
throw new IllegalArgumentException("don't know how to stream parameter " + index);
}
}
void checkAllParametersSet() throws SQLException {
for (int i = 0; i < paramValues.length; i++)
{
if (paramValues[i] == null)
throw new PSQLException(GT.tr("No value specified for parameter {0}.", new Integer(i + 1)), PSQLState.INVALID_PARAMETER_VALUE);
}
}
public ParameterList copy() {
FastpathParameterList newCopy = new FastpathParameterList(paramValues.length);
System.arraycopy(paramValues, 0, newCopy.paramValues, 0, paramValues.length);
return newCopy;
}
public void clear() {
Arrays.fill(paramValues, null);
}
private final Object[] paramValues;
}
|
