Update tests.

This commit is contained in:
Jon Chambers
2020-07-17 10:21:20 -04:00
committed by Jon Chambers
parent dfa94eac41
commit 215125de26
2 changed files with 36 additions and 65 deletions

View File

@@ -51,14 +51,8 @@ public class Messages {
this.database.getDatabase().registerRowMapper(new OutgoingMessageEntityRowMapper());
}
@VisibleForTesting
public void store(UUID guid, Envelope message, String destination, long destinationDevice) {
final Envelope messageWithGuid = message.toBuilder().setServerGuid(guid.toString()).build();
store(List.of(messageWithGuid), destination, destinationDevice);
}
public void store(List<Envelope> messages, String destination, long destinationDevice) {
database.use(jdbi ->jdbi.useHandle(handle -> {
database.use(jdbi -> jdbi.useHandle(handle -> {
try (Timer.Context ignored = storeTimer.time()) {
final PreparedBatch batch = handle.prepareBatch("INSERT INTO messages (" + GUID + ", " + TYPE + ", " + RELAY + ", " + TIMESTAMP + ", " + SERVER_TIMESTAMP + ", " + SOURCE + ", " + SOURCE_UUID + ", " + SOURCE_DEVICE + ", " + DESTINATION + ", " + DESTINATION_DEVICE + ", " + MESSAGE + ", " + CONTENT + ") " +
"VALUES (:guid, :type, :relay, :timestamp, :server_timestamp, :source, :source_uuid, :source_device, :destination, :destination_device, :message, :content)");