Skip to content

Commit e25e247

Browse files
author
zhushaoping
committed
support to add jdbc name parameter after LIMIT and TOP keywords
1 parent 091c5dd commit e25e247

File tree

4 files changed

+143
-84
lines changed

4 files changed

+143
-84
lines changed

‎src/main/java/net/sf/jsqlparser/statement/select/Limit.java

Lines changed: 11 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -21,61 +21,35 @@
2121
*/
2222
package net.sf.jsqlparser.statement.select;
2323

24-
import net.sf.jsqlparser.expression.JdbcParameter;
24+
import net.sf.jsqlparser.expression.Expression;
2525

2626
/**
2727
* A limit clause in the form [LIMIT {[offset,] row_count) | (row_count | ALL)
2828
* OFFSET offset}]
2929
*/
3030
public class Limit {
3131

32-
private long offset;
33-
private long rowCount;
34-
private JdbcParameter rowCountJdbcParameter;
35-
private JdbcParameter offsetJdbcParameter;
32+
private Expression rowCount;
33+
private Expression offset;
3634
private boolean limitAll;
3735
private boolean limitNull = false;
3836

39-
public long getOffset() {
37+
public Expression getOffset() {
4038
return offset;
4139
}
4240

43-
public long getRowCount() {
41+
public Expression getRowCount() {
4442
return rowCount;
4543
}
4644

47-
public void setOffset(long l) {
45+
public void setOffset(Expression l) {
4846
offset = l;
4947
}
5048

51-
public void setRowCount(long l) {
49+
public void setRowCount(Expression l) {
5250
rowCount = l;
5351
}
5452

55-
public JdbcParameter getRowCountJdbcParameter() {
56-
return rowCountJdbcParameter;
57-
}
58-
59-
public void setRowCountJdbcParameter(JdbcParameter rowCountJdbcParameter) {
60-
this.rowCountJdbcParameter = rowCountJdbcParameter;
61-
}
62-
63-
public JdbcParameter getOffsetJdbcParameter() {
64-
return offsetJdbcParameter;
65-
}
66-
67-
public void setOffsetJdbcParameter(JdbcParameter offsetJdbcParameter) {
68-
this.offsetJdbcParameter = offsetJdbcParameter;
69-
}
70-
71-
public boolean isRowCountJdbcParameter(){
72-
return null != this.rowCountJdbcParameter;
73-
}
74-
75-
public boolean isOffsetJdbcParameter(){
76-
return null != this.offsetJdbcParameter;
77-
}
78-
7953
/**
8054
* @return true if the limit is "LIMIT ALL [OFFSET ...])
8155
*/
@@ -100,11 +74,11 @@ public String toString() {
10074
if (limitNull) {
10175
retVal += "NULL";
10276
} else {
103-
if (offset > 0 || null != offsetJdbcParameter) {
104-
retVal += (null != offsetJdbcParameter ? offsetJdbcParameter : Long.toString(offset)) + ", ";
77+
if (null != offset) {
78+
retVal += offset + ", ";
10579
}
106-
if ( rowCount >= 0 || null != rowCountJdbcParameter) {
107-
retVal += (null !=rowCountJdbcParameter ? rowCountJdbcParameter : Long.toString(rowCount));
80+
if (null != rowCount) {
81+
retVal += rowCount;
10882
}
10983
}
11084

‎src/main/java/net/sf/jsqlparser/util/deparser/LimitDeparser.java

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -36,15 +36,11 @@ public void deParse(Limit limit) {
3636
if (limit.isLimitNull()) {
3737
buffer.append("NULL");
3838
} else {
39-
if (null != limit.getOffsetJdbcParameter()) {
40-
buffer.append(limit.getOffsetJdbcParameter()).append(", ");
41-
} else if (limit.getOffset() != 0) {
39+
if (null != limit.getOffset()) {
4240
buffer.append(limit.getOffset()).append(", ");
4341
}
4442

45-
if (null != limit.getRowCountJdbcParameter()) {
46-
buffer.append(limit.getRowCountJdbcParameter());
47-
} else if (limit.getRowCount() >= 0) {
43+
if (null != limit.getRowCount()) {
4844
buffer.append(limit.getRowCount());
4945
}
5046
}

‎src/main/jjtree/net/sf/jsqlparser/parser/JSqlParserCC.jjt

Lines changed: 15 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1446,7 +1446,6 @@ OrderByElement OrderByElement():
14461446
Limit LimitWithOffset():
14471447
{
14481448
Limit limit = new Limit();
1449-
limit.setRowCount(-1l);
14501449
Token token = null;
14511450
}
14521451
{
@@ -1455,17 +1454,21 @@ Limit LimitWithOffset():
14551454
// mysql-> LIMIT offset,row_count
14561455
<K_LIMIT>
14571456
(
1458-
token=<S_LONG> { limit.setOffset(Long.parseLong(token.image)); }
1457+
token=<S_LONG> { limit.setOffset(new LongValue(token.image)); }
14591458
|
1460-
"?" { limit.setOffsetJdbcParameter(new JdbcParameter()); } [ LOOKAHEAD(2) token = <S_LONG> { limit.getOffsetJdbcParameter().setIndex(Integer.valueOf(token.image)); } ]
1459+
"?" { limit.setOffset(new JdbcParameter()); } [ LOOKAHEAD(2) token = <S_LONG> { ((JdbcParameter)limit.getOffset()).setIndex(Integer.valueOf(token.image)); } ]
1460+
|
1461+
":" { limit.setOffset(new JdbcNamedParameter()); } [ LOOKAHEAD(2) token = <S_IDENTIFIER> { ((JdbcNamedParameter)limit.getOffset()).setName(token.image); } ]
14611462

14621463
)
14631464
","
14641465

14651466
(
1466-
token=<S_LONG> { limit.setRowCount(Long.parseLong(token.image)); }
1467+
token=<S_LONG> { limit.setRowCount(new LongValue(token.image)); }
14671468
|
1468-
"?" { limit.setRowCountJdbcParameter(new JdbcParameter()); } [ LOOKAHEAD(2) token = <S_LONG> { limit.getRowCountJdbcParameter().setIndex(Integer.valueOf(token.image)); } ]
1469+
"?" { limit.setRowCount(new JdbcParameter()); } [ LOOKAHEAD(2) token = <S_LONG> { ((JdbcParameter)limit.getRowCount()).setIndex(Integer.valueOf(token.image)); } ]
1470+
|
1471+
":" { limit.setRowCount(new JdbcNamedParameter()); } [ LOOKAHEAD(2) token = <S_IDENTIFIER> { ((JdbcNamedParameter)limit.getRowCount()).setName(token.image); } ]
14691472
)
14701473
|
14711474
limit = PlainLimit()
@@ -1478,16 +1481,17 @@ Limit LimitWithOffset():
14781481
Limit PlainLimit():
14791482
{
14801483
Limit limit = new Limit();
1481-
limit.setRowCount(-1l);
14821484
Token token = null;
14831485
}
14841486
{
14851487
// mysql-postgresql-> LIMIT (row_count | ALL | NULL)
14861488
<K_LIMIT>
14871489
(
1488-
token=<S_LONG> { limit.setRowCount(Long.parseLong(token.image)); }
1490+
token=<S_LONG> { limit.setRowCount(new LongValue(token.image)); }
1491+
|
1492+
"?" { limit.setRowCount(new JdbcParameter()); } [ LOOKAHEAD(2) token = <S_LONG> { ((JdbcParameter)limit.getRowCount()).setIndex(Integer.valueOf(token.image)); } ]
14891493
|
1490-
"?" { limit.setRowCountJdbcParameter(new JdbcParameter()); } [ LOOKAHEAD(2) token = <S_LONG> { limit.getRowCountJdbcParameter().setIndex(Integer.valueOf(token.image)); } ]
1494+
":" { limit.setRowCount(new JdbcNamedParameter()); } [ LOOKAHEAD(2) token = <S_IDENTIFIER> { ((JdbcNamedParameter)limit.getRowCount()).setName(token.image); } ]
14911495
|
14921496
<K_ALL> { limit.setLimitAll(true);}
14931497
|
@@ -1551,7 +1555,9 @@ Top Top():
15511555
(
15521556
token=<S_LONG> { top.setExpression(new LongValue(token.image)); }
15531557
|
1554-
"?" { top.setExpression(new JdbcParameter()); }
1558+
"?" { top.setExpression(new JdbcParameter()); } [ LOOKAHEAD(2) token = <S_LONG> { ((JdbcParameter)(top.getExpression())).setIndex(Integer.valueOf(token.image)); } ]
1559+
|
1560+
":" { top.setExpression(new JdbcNamedParameter()); } [ LOOKAHEAD(2) token = <S_IDENTIFIER> { ((JdbcNamedParameter)top.getExpression()).setName(token.image); } ]
15551561
|
15561562
"("
15571563
expr=AdditiveExpression() { top.setExpression(expr); }

0 commit comments

Comments
 (0)