YevhenBondarenko
2 years ago
23 changed files with 382 additions and 382 deletions
@ -0,0 +1,128 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.dao; |
|||
|
|||
import org.springframework.jdbc.core.BatchPreparedStatementSetter; |
|||
import org.springframework.jdbc.core.PreparedStatementCreator; |
|||
import org.springframework.jdbc.core.SqlProvider; |
|||
import org.springframework.jdbc.support.GeneratedKeyHolder; |
|||
import org.springframework.jdbc.support.KeyHolder; |
|||
import org.thingsboard.server.dao.sqlts.insert.AbstractInsertRepository; |
|||
|
|||
import java.sql.Connection; |
|||
import java.sql.PreparedStatement; |
|||
import java.sql.SQLException; |
|||
import java.util.ArrayList; |
|||
import java.util.List; |
|||
import java.util.Map; |
|||
|
|||
public abstract class AbstractSequenceInsertRepository<T> extends AbstractInsertRepository { |
|||
|
|||
public static final String SEQ_NUMBER = "seq_number"; |
|||
|
|||
public List<Long> saveOrUpdate(List<T> entities) { |
|||
return transactionTemplate.execute(status -> { |
|||
List<Long> seqNumbers = new ArrayList<>(entities.size()); |
|||
|
|||
KeyHolder keyHolder = new GeneratedKeyHolder(); |
|||
|
|||
int[] updateResult = onBatchUpdate(entities, keyHolder); |
|||
|
|||
List<Map<String, Object>> seqNumbersList = keyHolder.getKeyList(); |
|||
|
|||
int notUpdatedCount = entities.size() - seqNumbersList.size(); |
|||
|
|||
List<Integer> toInsertIndexes = new ArrayList<>(notUpdatedCount); |
|||
List<T> insertEntities = new ArrayList<>(notUpdatedCount); |
|||
int keyHolderIndex = 0; |
|||
for (int i = 0; i < updateResult.length; i++) { |
|||
if (updateResult[i] == 0) { |
|||
insertEntities.add(entities.get(i)); |
|||
seqNumbers.add(0L); |
|||
toInsertIndexes.add(i); |
|||
} else { |
|||
seqNumbers.add((Long) seqNumbersList.get(keyHolderIndex).get(SEQ_NUMBER)); |
|||
keyHolderIndex++; |
|||
} |
|||
} |
|||
|
|||
if (insertEntities.isEmpty()) { |
|||
return seqNumbers; |
|||
} |
|||
|
|||
onInsertOrUpdate(insertEntities, keyHolder); |
|||
|
|||
seqNumbersList = keyHolder.getKeyList(); |
|||
|
|||
for (int i = 0; i < seqNumbersList.size(); i++) { |
|||
seqNumbers.set(toInsertIndexes.get(i), (Long) seqNumbersList.get(i).get(SEQ_NUMBER)); |
|||
} |
|||
|
|||
return seqNumbers; |
|||
}); |
|||
} |
|||
|
|||
private int[] onBatchUpdate(List<T> entities, KeyHolder keyHolder) { |
|||
return jdbcTemplate.batchUpdate(new SequencePreparedStatementCreator(getBatchUpdateQuery()), new BatchPreparedStatementSetter() { |
|||
@Override |
|||
public void setValues(PreparedStatement ps, int i) throws SQLException { |
|||
setOnBatchUpdateValues(ps, i, entities); |
|||
} |
|||
|
|||
@Override |
|||
public int getBatchSize() { |
|||
return entities.size(); |
|||
} |
|||
}, keyHolder); |
|||
} |
|||
|
|||
private void onInsertOrUpdate(List<T> insertEntities, KeyHolder keyHolder) { |
|||
jdbcTemplate.batchUpdate(new SequencePreparedStatementCreator(getInsertOrUpdateQuery()), new BatchPreparedStatementSetter() { |
|||
@Override |
|||
public void setValues(PreparedStatement ps, int i) throws SQLException { |
|||
setOnInsertOrUpdateValues(ps, i, insertEntities); |
|||
} |
|||
|
|||
@Override |
|||
public int getBatchSize() { |
|||
return insertEntities.size(); |
|||
} |
|||
}, keyHolder); |
|||
} |
|||
|
|||
protected abstract void setOnBatchUpdateValues(PreparedStatement ps, int i, List<T> entities) throws SQLException; |
|||
|
|||
protected abstract void setOnInsertOrUpdateValues(PreparedStatement ps, int i, List<T> entities) throws SQLException; |
|||
|
|||
protected abstract String getBatchUpdateQuery(); |
|||
|
|||
protected abstract String getInsertOrUpdateQuery(); |
|||
|
|||
private record SequencePreparedStatementCreator(String sql) implements PreparedStatementCreator, SqlProvider { |
|||
|
|||
private static final String[] COLUMNS = {SEQ_NUMBER}; |
|||
|
|||
@Override |
|||
public PreparedStatement createPreparedStatement(Connection con) throws SQLException { |
|||
return con.prepareStatement(sql, COLUMNS); |
|||
} |
|||
|
|||
@Override |
|||
public String getSql() { |
|||
return this.sql; |
|||
} |
|||
} |
|||
} |
|||
@ -1,27 +0,0 @@ |
|||
/** |
|||
* Copyright © 2016-2024 The Thingsboard Authors |
|||
* |
|||
* Licensed under the Apache License, Version 2.0 (the "License"); |
|||
* you may not use this file except in compliance with the License. |
|||
* You may obtain a copy of the License at |
|||
* |
|||
* http://www.apache.org/licenses/LICENSE-2.0
|
|||
* |
|||
* Unless required by applicable law or agreed to in writing, software |
|||
* distributed under the License is distributed on an "AS IS" BASIS, |
|||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|||
* See the License for the specific language governing permissions and |
|||
* limitations under the License. |
|||
*/ |
|||
package org.thingsboard.server.dao.sql.attributes; |
|||
|
|||
import org.springframework.stereotype.Repository; |
|||
import org.springframework.transaction.annotation.Transactional; |
|||
import org.thingsboard.server.dao.util.SqlDao; |
|||
|
|||
@Repository |
|||
@Transactional |
|||
@SqlDao |
|||
public class SqlAttributesInsertRepository extends AttributeKvInsertRepository { |
|||
|
|||
} |
|||
Loading…
Reference in new issue