@@ -67,9 +67,8 @@ public final class InsertStatement extends DMLStatement {
67
67
* Append generate key token.
68
68
*
69
69
* @param shardingRule databases and tables sharding rule
70
- * @param parametersSize parameters size
71
70
*/
72
- public void appendGenerateKeyToken (final ShardingRule shardingRule , final int parametersSize ) {
71
+ public void appendGenerateKeyToken (final ShardingRule shardingRule ) {
73
72
if (null != generatedKey ) {
74
73
return ;
75
74
}
@@ -82,26 +81,22 @@ public void appendGenerateKeyToken(final ShardingRule shardingRule, final int pa
82
81
return ;
83
82
}
84
83
ItemsToken valuesToken = new ItemsToken (generatedKeysToken .get ().getBeginPosition ());
85
- if (0 == parametersSize ) {
86
- appendGenerateKeyToken (shardingRule , tableRule .get (), valuesToken );
87
- } else {
88
- appendGenerateKeyToken (shardingRule , tableRule .get (), valuesToken , parametersSize );
89
- }
84
+ appendGenerateKeyToken (shardingRule , tableRule .get (), valuesToken );
90
85
getSqlTokens ().remove (generatedKeysToken .get ());
91
86
getSqlTokens ().add (valuesToken );
92
87
}
93
88
94
89
private void appendGenerateKeyToken (final ShardingRule shardingRule , final TableRule tableRule , final ItemsToken valuesToken ) {
95
- Number generatedKey = shardingRule . generateKey ( tableRule . getLogicTable ());
96
- valuesToken . getItems (). add ( generatedKey . toString ());
97
- getConditions (). add ( new Condition ( new Column ( tableRule . getGenerateKeyColumn (), tableRule . getLogicTable ()), new SQLNumberExpression ( generatedKey )), shardingRule );
98
- this . generatedKey = new GeneratedKey ( tableRule .getLogicTable (), - 1 , generatedKey );
99
- }
100
-
101
- private void appendGenerateKeyToken ( final ShardingRule shardingRule , final TableRule tableRule , final ItemsToken valuesToken , final int parametersSize ) {
102
- valuesToken . getItems ().add (Symbol . QUESTION . getLiterals () );
103
- getConditions (). add ( new Condition ( new Column (tableRule .getGenerateKeyColumn (), tableRule . getLogicTable ()), new SQLPlaceholderExpression ( parametersSize )), shardingRule );
104
- generatedKey = new GeneratedKey ( tableRule . getGenerateKeyColumn (), parametersSize , null );
90
+ if ( 0 == getParametersIndex ()) {
91
+ Number generatedKey = shardingRule . generateKey ( tableRule . getLogicTable ());
92
+ valuesToken . getItems (). add ( generatedKey . toString () );
93
+ getConditions (). add ( new Condition ( new Column ( tableRule .getGenerateKeyColumn (), tableRule . getLogicTable ()), new SQLNumberExpression ( generatedKey )), shardingRule );
94
+ this . generatedKey = new GeneratedKey ( tableRule . getLogicTable (), - 1 , generatedKey );
95
+ } else {
96
+ valuesToken . getItems (). add ( Symbol . QUESTION . getLiterals ());
97
+ getConditions ().add (new Condition ( new Column ( tableRule . getGenerateKeyColumn (), tableRule . getLogicTable ()), new SQLPlaceholderExpression ( getParametersIndex ())), shardingRule );
98
+ generatedKey = new GeneratedKey (tableRule .getGenerateKeyColumn (), getParametersIndex (), null );
99
+ }
105
100
}
106
101
107
102
private Optional <GeneratedKeyToken > findGeneratedKeyToken () {
0 commit comments