|
7 | 7 |
|
8 | 8 | import java.io.File; |
9 | 9 | import java.sql.Connection; |
10 | | -import java.util.Collections; |
| 10 | +import java.util.Arrays; |
11 | 11 | import java.util.List; |
12 | 12 | import java.util.concurrent.atomic.AtomicReference; |
13 | 13 | import java.util.function.Consumer; |
@@ -45,34 +45,68 @@ protected void flushConnection(Connection connection) { |
45 | 45 |
|
46 | 46 | @Override |
47 | 47 | protected List<String> toSaveStatement(String name, String[] keyColumns, String[] valueColumns) { |
48 | | - StringBuilder statement = new StringBuilder("INSERT OR REPLACE INTO `") |
| 48 | + StringBuilder insertStatement = new StringBuilder("INSERT OR IGNORE INTO `") |
49 | 49 | .append(name) |
50 | 50 | .append("` ("); |
51 | 51 | for (int i = 0; i < keyColumns.length + valueColumns.length; i++) { |
52 | | - statement.append("`") |
| 52 | + insertStatement.append("`") |
53 | 53 | .append(i < keyColumns.length ? keyColumns[i] : valueColumns[i - keyColumns.length]) |
54 | 54 | .append("`"); |
55 | 55 | if (i != keyColumns.length + valueColumns.length - 1) { |
56 | | - statement.append(", "); |
| 56 | + insertStatement.append(", "); |
57 | 57 | } |
58 | 58 | } |
59 | | - statement.append(") VALUES ("); |
| 59 | + insertStatement.append(") VALUES ("); |
60 | 60 | for (int i = 0; i < keyColumns.length + valueColumns.length; i++) { |
61 | | - statement.append("?"); |
| 61 | + insertStatement.append("?"); |
62 | 62 | if (i != keyColumns.length + valueColumns.length - 1) { |
63 | | - statement.append(", "); |
| 63 | + insertStatement.append(", "); |
64 | 64 | } |
65 | 65 | } |
66 | | - statement.append(");"); |
67 | | - return Collections.singletonList(statement.toString()); |
| 66 | + insertStatement.append(");"); |
| 67 | + |
| 68 | + StringBuilder updateStatement = new StringBuilder("UPDATE `") |
| 69 | + .append(name) |
| 70 | + .append("` SET "); |
| 71 | + for (int i = 0; i < valueColumns.length; i++) { |
| 72 | + updateStatement.append("`") |
| 73 | + .append(valueColumns[i]) |
| 74 | + .append("` = ?"); |
| 75 | + if (i != valueColumns.length - 1) { |
| 76 | + updateStatement.append(", "); |
| 77 | + } |
| 78 | + } |
| 79 | + updateStatement.append(" WHERE "); |
| 80 | + for (int i = 0; i < keyColumns.length; i++) { |
| 81 | + updateStatement.append("`") |
| 82 | + .append(keyColumns[i]) |
| 83 | + .append("` = ?"); |
| 84 | + if (i != keyColumns.length - 1) { |
| 85 | + updateStatement.append(" AND "); |
| 86 | + } |
| 87 | + } |
| 88 | + updateStatement.append(";"); |
| 89 | + |
| 90 | + return Arrays.asList( |
| 91 | + insertStatement.toString(), |
| 92 | + updateStatement.toString() |
| 93 | + ); |
68 | 94 | } |
69 | 95 |
|
70 | 96 | @Override |
71 | 97 | protected List<Object[]> toSaveValues(Object[] keys, Object[] values) { |
72 | | - Object[] queryValues = new Object[keys.length + values.length]; |
73 | | - System.arraycopy(keys, 0, queryValues, 0, keys.length); |
74 | | - System.arraycopy(values, 0, queryValues, keys.length, values.length); |
75 | | - return Collections.singletonList(queryValues); |
| 98 | + Object[] insertValues = new Object[keys.length + values.length]; |
| 99 | + System.arraycopy(keys, 0, insertValues, 0, keys.length); |
| 100 | + System.arraycopy(values, 0, insertValues, keys.length, values.length); |
| 101 | + |
| 102 | + Object[] updateValues = new Object[values.length + keys.length]; |
| 103 | + System.arraycopy(values, 0, updateValues, 0, values.length); |
| 104 | + System.arraycopy(keys, 0, updateValues, values.length, keys.length); |
| 105 | + |
| 106 | + return Arrays.asList( |
| 107 | + insertValues, |
| 108 | + updateValues |
| 109 | + ); |
76 | 110 | } |
77 | 111 |
|
78 | 112 | @Override |
|
0 commit comments