1 /*
2 * Licensed to the Apache Software Foundation (ASF) under one or more
3 * contributor license agreements. See the NOTICE file distributed with
4 * this work for additional information regarding copyright ownership.
5 * The ASF licenses this file to You under the Apache License, Version 2.0
6 * (the "License"); you may not use this file except in compliance with
7 * the License. You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 */
17
18 package org.apache.shardingsphere.driver.executor.batch;
19
20 import lombok.RequiredArgsConstructor;
21
22 import java.sql.SQLException;
23 import java.sql.Statement;
24 import java.util.LinkedList;
25 import java.util.List;
26
27 /**
28 * Batch executor for {@link Statement}.
29 */
30 @RequiredArgsConstructor
31 public final class BatchStatementExecutor {
32
33 private final Statement statement;
34
35 private final List<String> batchedSQLs = new LinkedList<>();
36
37 /**
38 * Add SQL for batched execution.
39 *
40 * @param sql SQL
41 */
42 public void addBatch(final String sql) {
43 batchedSQLs.add(sql);
44 }
45
46 /**
47 * Execute batched SQLs.
48 *
49 * @return execute results
50 * @throws SQLException SQL exception
51 */
52 public int[] executeBatch() throws SQLException {
53 int[] result = new int[batchedSQLs.size()];
54 int index = 0;
55 for (String each : batchedSQLs) {
56 result[index++] = statement.executeUpdate(each);
57 }
58 return result;
59 }
60
61 /**
62 * Clear batched SQLs.
63 */
64 public void clear() {
65 batchedSQLs.clear();
66 }
67 }