AnastasiiaKomar
4 years ago
9 changed files with 409 additions and 217 deletions
@ -0,0 +1,137 @@ |
|||||
|
/** |
||||
|
* Copyright © 2016-2022 The Thingsboard Authors |
||||
|
* <p> |
||||
|
* 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 |
||||
|
* <p> |
||||
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
* <p> |
||||
|
* 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.cache.ota.service; |
||||
|
|
||||
|
import lombok.RequiredArgsConstructor; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.commons.io.FileUtils; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.id.OtaPackageId; |
||||
|
|
||||
|
import java.io.*; |
||||
|
import java.sql.Blob; |
||||
|
import java.sql.SQLException; |
||||
|
import java.util.UUID; |
||||
|
import java.util.concurrent.ConcurrentHashMap; |
||||
|
import java.util.concurrent.ConcurrentMap; |
||||
|
import java.util.concurrent.CountDownLatch; |
||||
|
|
||||
|
@Slf4j |
||||
|
@RequiredArgsConstructor |
||||
|
@Component |
||||
|
public class BaseFileCacheService implements FileCacheService { |
||||
|
@Value("${java.io.tmpdir}/ota/") |
||||
|
private String PATH; |
||||
|
|
||||
|
private final TemporaryFileCleaner fileCleaner; |
||||
|
private final ConcurrentMap<OtaPackageId, CountDownLatch> files = new ConcurrentHashMap<>(); |
||||
|
|
||||
|
|
||||
|
public File saveDataTemporaryFile(InputStream inputStream) { |
||||
|
File path = new File(PATH); |
||||
|
try { |
||||
|
File tempFile = File.createTempFile(UUID.randomUUID().toString(), ".tmp", path); |
||||
|
FileUtils.copyInputStreamToFile(inputStream, tempFile); |
||||
|
return tempFile; |
||||
|
} catch (IOException e) { |
||||
|
log.error("Failed to create temp file", e); |
||||
|
throw new RuntimeException("Failed to create temp file for input stream"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public File getOtaDataFile(OtaPackageId otaPackageId) throws FileNotFoundException { |
||||
|
String fileName = PATH + otaPackageId.getId(); |
||||
|
if (exist(fileName)) { |
||||
|
fileCleaner.updateFileUsageStatus(otaPackageId); |
||||
|
return new File(fileName); |
||||
|
} |
||||
|
throw new FileNotFoundException(); |
||||
|
} |
||||
|
|
||||
|
public File loadToFile(OtaPackageId otaPackageId, Blob data) { |
||||
|
if (otaPackageId == null || data == null) { |
||||
|
log.error("Received null variables: {}", otaPackageId == null ? "otaPackageId" : "data"); |
||||
|
throw new NullPointerException(); |
||||
|
} |
||||
|
CountDownLatch latch = files.computeIfAbsent(otaPackageId, ota -> processFileSaving(ota, data)); |
||||
|
String fileName = PATH + otaPackageId.getId(); |
||||
|
if (latch.getCount() == 0) { |
||||
|
return new File(fileName); |
||||
|
} else { |
||||
|
try { |
||||
|
latch.await(); |
||||
|
return new File(fileName); |
||||
|
} catch (InterruptedException e) { |
||||
|
log.error("Failed to save file {} to file system", fileName, e); |
||||
|
throw new RuntimeException(e); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public void deleteFile(File file) { |
||||
|
try { |
||||
|
if (file.exists()) { |
||||
|
FileUtils.delete(file); |
||||
|
log.info("System file {} was deleted", file.getName()); |
||||
|
} |
||||
|
} catch (IOException e) { |
||||
|
log.error("Failed to delete file {}", file.getName(), e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
public InputStream getOtaDataStream(OtaPackageId otaPackageId) throws FileNotFoundException { |
||||
|
return new FileInputStream(getOtaDataFile(otaPackageId)); |
||||
|
} |
||||
|
|
||||
|
public InputStream loadOtaData(OtaPackageId otaPackageId, Blob data) { |
||||
|
try { |
||||
|
return new FileInputStream(loadToFile(otaPackageId, data)); |
||||
|
} catch (FileNotFoundException e) { |
||||
|
log.error("Failed to upload temporary file {} to system", otaPackageId.getId(), e); |
||||
|
throw new RuntimeException(e); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private CountDownLatch processFileSaving(OtaPackageId otaPackageId, Blob data) { |
||||
|
CountDownLatch latch = new CountDownLatch(1); |
||||
|
try { |
||||
|
InputStream binaryStream = data.getBinaryStream(); |
||||
|
String fileName = PATH + otaPackageId.getId(); |
||||
|
saveAsSystemFile(fileName, binaryStream); |
||||
|
latch.countDown(); |
||||
|
return latch; |
||||
|
} catch (SQLException e) { |
||||
|
log.error("Failed to get binary data from otaPackage {}", otaPackageId, e); |
||||
|
throw new RuntimeException("Failed to get binary data from database"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void saveAsSystemFile(String fileName, InputStream inputStream) { |
||||
|
try { |
||||
|
File file = new File(fileName); |
||||
|
FileUtils.copyInputStreamToFile(inputStream, file); |
||||
|
} catch (IOException e) { |
||||
|
log.error("Failed to copy stream to system file {}", fileName, e); |
||||
|
throw new RuntimeException("Failed to save file"); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private boolean exist(String name) { |
||||
|
File file = new File(name); |
||||
|
return file.exists(); |
||||
|
} |
||||
|
} |
||||
@ -1,183 +1,17 @@ |
|||||
/** |
|
||||
* Copyright © 2016-2022 The Thingsboard Authors |
|
||||
* <p> |
|
||||
* 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 |
|
||||
* <p> |
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
|
||||
* <p> |
|
||||
* 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.cache.ota.service; |
package org.thingsboard.server.cache.ota.service; |
||||
|
|
||||
import lombok.RequiredArgsConstructor; |
|
||||
import lombok.extern.slf4j.Slf4j; |
|
||||
import org.apache.commons.io.FileUtils; |
|
||||
import org.springframework.beans.factory.annotation.Value; |
|
||||
import org.springframework.stereotype.Component; |
|
||||
import org.thingsboard.server.common.data.id.OtaPackageId; |
import org.thingsboard.server.common.data.id.OtaPackageId; |
||||
|
|
||||
import javax.annotation.PostConstruct; |
import java.io.File; |
||||
import java.io.*; |
import java.io.FileNotFoundException; |
||||
|
import java.io.InputStream; |
||||
import java.sql.Blob; |
import java.sql.Blob; |
||||
import java.sql.SQLException; |
|
||||
import java.util.List; |
|
||||
import java.util.Map; |
|
||||
import java.util.UUID; |
|
||||
import java.util.concurrent.*; |
|
||||
import java.util.stream.Collectors; |
|
||||
|
|
||||
@Slf4j |
public interface FileCacheService { |
||||
@RequiredArgsConstructor |
void deleteFile(File file); |
||||
@Component |
InputStream getOtaDataStream(OtaPackageId otaPackageId) throws FileNotFoundException ; |
||||
public class FileCacheService { |
InputStream loadOtaData(OtaPackageId otaPackageId, Blob data); |
||||
@Value("${java.io.tmpdir}/ota/") |
File loadToFile(OtaPackageId otaPackageId, Blob data); |
||||
private String PATH; |
File saveDataTemporaryFile(InputStream inputStream); |
||||
private final static String FILE_PATH_TEMPLATE = "%s"; |
File getOtaDataFile(OtaPackageId otaPackageId) throws FileNotFoundException; |
||||
private final static long TEMPORARY_FILE_INACTIVITY_TIME = 300;//todo increase time
|
} |
||||
private final ConcurrentMap<OtaPackageId, CountDownLatch> files = new ConcurrentHashMap<>(); |
|
||||
private final ConcurrentMap<OtaPackageId, Long> lastActivityTimes = new ConcurrentHashMap<>(); |
|
||||
|
|
||||
@PostConstruct |
|
||||
private void createScheduledTaskToClear() { |
|
||||
createTempDirectoryIfNotExist(); |
|
||||
cleanDirectoryWithTemporaryFiles(); |
|
||||
ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1); |
|
||||
scheduler.scheduleAtFixedRate(this::deleteUnusedTemporaryFiles, 3, 3, TimeUnit.MINUTES); //todo increase time
|
|
||||
} |
|
||||
|
|
||||
private void createTempDirectoryIfNotExist() { |
|
||||
File directory = new File(PATH); |
|
||||
if(!directory.exists()){ |
|
||||
directory.mkdir(); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void cleanDirectoryWithTemporaryFiles() { |
|
||||
File directory = new File(PATH); |
|
||||
if (directory.isDirectory()) { |
|
||||
try { |
|
||||
FileUtils.cleanDirectory(directory); |
|
||||
log.info("Directory with temporary files cleaned"); |
|
||||
} catch (IOException e) { |
|
||||
log.error("Failed to clean directory with temporary files", e); |
|
||||
throw new RuntimeException(e); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void deleteUnusedTemporaryFiles() { |
|
||||
long currentTime = System.currentTimeMillis(); |
|
||||
List<OtaPackageId> toBeDeleted = lastActivityTimes.entrySet() |
|
||||
.stream() |
|
||||
.filter(entry -> currentTime - entry.getValue() > TEMPORARY_FILE_INACTIVITY_TIME) |
|
||||
.map(Map.Entry::getKey) |
|
||||
.collect(Collectors.toList()); |
|
||||
try { |
|
||||
toBeDeleted.forEach(otaId -> { |
|
||||
deleteFile(otaId.getId().toString()); |
|
||||
lastActivityTimes.remove(otaId); |
|
||||
}); |
|
||||
} catch (Exception e) { |
|
||||
log.error(",,", e); |
|
||||
} |
|
||||
log.info("Deleted {} unused temporary files", toBeDeleted.size()); |
|
||||
} |
|
||||
|
|
||||
private void deleteFile(String otaId) { |
|
||||
String fileName = PATH + otaId; |
|
||||
File file = new File(String.format(FILE_PATH_TEMPLATE, fileName)); |
|
||||
deleteFile(file); |
|
||||
} |
|
||||
|
|
||||
public void deleteFile(File file) { |
|
||||
try { |
|
||||
if (file.exists()) { |
|
||||
FileUtils.delete(file); |
|
||||
log.info("System file {} was deleted", file.getName()); |
|
||||
} |
|
||||
} catch (IOException e) { |
|
||||
log.error("Failed to delete file {}", file.getName(), e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
public InputStream getOtaDataStream(OtaPackageId otaPackageId) throws FileNotFoundException { |
|
||||
String fileName = PATH + otaPackageId.getId(); |
|
||||
if (exist(fileName)) { |
|
||||
lastActivityTimes.put(otaPackageId, System.currentTimeMillis()); |
|
||||
return new FileInputStream(String.format(FILE_PATH_TEMPLATE, fileName)); |
|
||||
} |
|
||||
throw new FileNotFoundException(); |
|
||||
} |
|
||||
|
|
||||
private InputStream load(OtaPackageId otaPackageId, Blob data) throws FileNotFoundException { |
|
||||
CountDownLatch latch = files.computeIfAbsent(otaPackageId, ota -> processFileSaving(ota, data)); |
|
||||
String fileName = PATH + otaPackageId.getId(); |
|
||||
if (latch.getCount() == 0) { |
|
||||
return new FileInputStream(String.format(FILE_PATH_TEMPLATE, fileName)); |
|
||||
} else { |
|
||||
try { |
|
||||
latch.await(); |
|
||||
return new FileInputStream(String.format(FILE_PATH_TEMPLATE, fileName)); |
|
||||
} catch (InterruptedException e) { |
|
||||
log.error("Failed to save file {} to file system", fileName, e); |
|
||||
throw new RuntimeException(e); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private CountDownLatch processFileSaving(OtaPackageId otaPackageId, Blob data) { |
|
||||
CountDownLatch latch = new CountDownLatch(1); |
|
||||
try { |
|
||||
InputStream binaryStream = data.getBinaryStream(); |
|
||||
String fileName =PATH + otaPackageId.getId(); |
|
||||
saveAsSystemFile(fileName, binaryStream); |
|
||||
latch.countDown(); |
|
||||
return latch; |
|
||||
} catch (SQLException e) { |
|
||||
log.error("Failed to get binary data from otaPackage {}", otaPackageId, e); |
|
||||
throw new RuntimeException("Failed to get binary data from database"); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private void saveAsSystemFile(String fileName, InputStream inputStream) { |
|
||||
try { |
|
||||
File file = new File(String.format(FILE_PATH_TEMPLATE, fileName)); |
|
||||
FileUtils.copyInputStreamToFile(inputStream, file); |
|
||||
} catch (IOException e) { |
|
||||
log.error("Failed to copy stream to system file {}", fileName, e); |
|
||||
throw new RuntimeException("Failed to save file"); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
private boolean exist(String name) { |
|
||||
File file = new File(String.format(FILE_PATH_TEMPLATE, name)); |
|
||||
return file.exists(); |
|
||||
} |
|
||||
|
|
||||
public InputStream loadOtaData(OtaPackageId otaPackageId, Blob data) { |
|
||||
try { |
|
||||
return load(otaPackageId, data); |
|
||||
} catch (FileNotFoundException e) { |
|
||||
log.error("Failed to upload temporary file {} to system", otaPackageId.getId(), e); |
|
||||
throw new RuntimeException(e); |
|
||||
} |
|
||||
} |
|
||||
|
|
||||
public File saveDataTemporaryFile(InputStream inputStream) { |
|
||||
File path = new File(PATH); |
|
||||
try { |
|
||||
File tempFile = File.createTempFile(UUID.randomUUID().toString(), ".tmp", path); |
|
||||
FileUtils.copyInputStreamToFile(inputStream, tempFile); |
|
||||
return tempFile; |
|
||||
} catch (IOException e) { |
|
||||
log.error("Failed to create temp file", e); |
|
||||
throw new RuntimeException("Failed to create temp file for input stream"); |
|
||||
} |
|
||||
} |
|
||||
} |
|
||||
|
|||||
@ -0,0 +1,88 @@ |
|||||
|
package org.thingsboard.server.cache.ota.service; |
||||
|
|
||||
|
import lombok.extern.apachecommons.CommonsLog; |
||||
|
import lombok.extern.slf4j.Slf4j; |
||||
|
import org.apache.commons.io.FileUtils; |
||||
|
import org.springframework.beans.factory.annotation.Value; |
||||
|
import org.springframework.stereotype.Component; |
||||
|
import org.thingsboard.server.common.data.id.OtaPackageId; |
||||
|
|
||||
|
import javax.annotation.PostConstruct; |
||||
|
import java.io.File; |
||||
|
import java.io.IOException; |
||||
|
import java.util.List; |
||||
|
import java.util.Map; |
||||
|
import java.util.concurrent.*; |
||||
|
import java.util.stream.Collectors; |
||||
|
|
||||
|
@Slf4j |
||||
|
@Component |
||||
|
public class TemporaryFileCleaner { |
||||
|
@Value("${java.io.tmpdir}/ota/") |
||||
|
private String PATH; |
||||
|
private final static long TEMPORARY_FILE_INACTIVITY_TIME = 900_000; |
||||
|
private final ConcurrentMap<OtaPackageId, Long> lastActivityTimes = new ConcurrentHashMap<>(); |
||||
|
|
||||
|
public void updateFileUsageStatus(OtaPackageId otaPackageId){ |
||||
|
lastActivityTimes.put(otaPackageId, System.currentTimeMillis()); |
||||
|
} |
||||
|
|
||||
|
@PostConstruct |
||||
|
public void createScheduledTaskToClear() { |
||||
|
createTempDirectoryIfNotExist(); |
||||
|
cleanDirectoryWithTemporaryFiles(); |
||||
|
ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1); |
||||
|
scheduler.scheduleAtFixedRate(this::deleteUnusedTemporaryFiles, 10, 10, TimeUnit.MINUTES); |
||||
|
} |
||||
|
|
||||
|
private void createTempDirectoryIfNotExist() { |
||||
|
File directory = new File(PATH); |
||||
|
if (!directory.exists()) { |
||||
|
directory.mkdir(); |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void cleanDirectoryWithTemporaryFiles() { |
||||
|
File directory = new File(PATH); |
||||
|
if (directory.isDirectory()) { |
||||
|
try { |
||||
|
FileUtils.cleanDirectory(directory); |
||||
|
log.info("Directory with temporary files cleaned"); |
||||
|
} catch (IOException e) { |
||||
|
log.error("Failed to clean directory with temporary files", e); |
||||
|
throw new RuntimeException(e); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
private void deleteUnusedTemporaryFiles() { |
||||
|
long currentTime = System.currentTimeMillis(); |
||||
|
List<OtaPackageId> toBeDeleted = lastActivityTimes.entrySet() |
||||
|
.stream() |
||||
|
.filter(entry -> currentTime - entry.getValue() > TEMPORARY_FILE_INACTIVITY_TIME) |
||||
|
.map(Map.Entry::getKey) |
||||
|
.collect(Collectors.toList()); |
||||
|
try { |
||||
|
toBeDeleted.forEach(otaId -> { |
||||
|
deleteFile(otaId.getId().toString()); |
||||
|
lastActivityTimes.remove(otaId); |
||||
|
}); |
||||
|
} catch (Exception e) { |
||||
|
log.error("Failed to delete unused files", e); |
||||
|
} |
||||
|
log.info("Deleted {} unused temporary files", toBeDeleted.size()); |
||||
|
} |
||||
|
|
||||
|
private void deleteFile(String otaId) { |
||||
|
String fileName = PATH + otaId; |
||||
|
File file = new File(fileName); |
||||
|
try { |
||||
|
if (file.exists()) { |
||||
|
FileUtils.delete(file); |
||||
|
log.info("System file {} was deleted", file.getName()); |
||||
|
} |
||||
|
} catch (IOException e) { |
||||
|
log.error("Failed to delete file {}", file.getName(), e); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
@ -0,0 +1,101 @@ |
|||||
|
package org.thingsboard.server.cache.ota.service; |
||||
|
|
||||
|
import lombok.SneakyThrows; |
||||
|
import org.hibernate.engine.jdbc.BlobProxy; |
||||
|
import org.junit.jupiter.api.BeforeEach; |
||||
|
import org.junit.jupiter.api.Test; |
||||
|
import org.mockito.Mockito; |
||||
|
import org.thingsboard.server.common.data.id.OtaPackageId; |
||||
|
|
||||
|
import java.io.*; |
||||
|
import java.security.MessageDigest; |
||||
|
import java.sql.Blob; |
||||
|
import java.util.Objects; |
||||
|
import java.util.UUID; |
||||
|
|
||||
|
import static org.junit.jupiter.api.Assertions.*; |
||||
|
|
||||
|
|
||||
|
class BaseFileCacheServiceTest { |
||||
|
private final static String PATH = "/home/anastasiia/IdeaProjects/thingsboard/common/cache"; |
||||
|
private final static String FILE_FILLING = "Hello, testing environment"; |
||||
|
private static final int ONE_MEGA_BYTE = 1_000_000; |
||||
|
private final static OtaPackageId OTA_PACKAGE_ID = new OtaPackageId(UUID.randomUUID()); |
||||
|
private final static Blob DATA = BlobProxy.generateProxy(FILE_FILLING.getBytes()); |
||||
|
File file; |
||||
|
BaseFileCacheService baseFileCacheService = new BaseFileCacheService(new TemporaryFileCleaner()); |
||||
|
|
||||
|
@BeforeEach |
||||
|
void setUp() { |
||||
|
file = Mockito.mock(File.class); |
||||
|
// fileCacheService = Mockito.mock(FileCacheService.class);
|
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testNotThrowExceptionsWhenDeleteFile() { |
||||
|
Mockito.when(file.exists()).thenReturn(true); |
||||
|
assertDoesNotThrow(() -> baseFileCacheService.deleteFile(new File("/download/alarm_rule.js"))); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testDataSavingWithNullBlob() { |
||||
|
assertThrows(NullPointerException.class, () -> baseFileCacheService.loadToFile(OTA_PACKAGE_ID, null)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
void testDataSavingWithNullOtaPackageId() { |
||||
|
assertThrows(NullPointerException.class, () -> baseFileCacheService.loadToFile(null, DATA)); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
@SneakyThrows |
||||
|
void testMultiSavingDataToFile() { |
||||
|
File directory = new File(PATH); |
||||
|
int beginning = Objects.requireNonNull(directory.list()).length; |
||||
|
Thread thread1 = new Thread(() -> baseFileCacheService.loadToFile(OTA_PACKAGE_ID, DATA)); |
||||
|
Thread thread2 = new Thread(() -> baseFileCacheService.loadToFile(OTA_PACKAGE_ID, DATA)); |
||||
|
thread1.start(); |
||||
|
thread2.start(); |
||||
|
thread1.join(); |
||||
|
thread2.join(); |
||||
|
File directory1 = new File(PATH); |
||||
|
int ending = Objects.requireNonNull(directory1.list()).length; |
||||
|
assertEquals(1, ending - beginning); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
@SneakyThrows |
||||
|
void testCorrectDataSavingToFile() { |
||||
|
String sha256 = calculateChecksumSHA256(new ByteArrayInputStream(FILE_FILLING.getBytes())); |
||||
|
File file = baseFileCacheService.loadToFile(OTA_PACKAGE_ID, DATA); |
||||
|
assertEquals(sha256, calculateChecksumSHA256(new FileInputStream(file))); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
@SneakyThrows |
||||
|
void testNotThrowExceptionWhenCreatingTemporaryFiles() { |
||||
|
BaseFileCacheService mock = Mockito.mock(BaseFileCacheService.class); |
||||
|
assertDoesNotThrow(() -> mock.saveDataTemporaryFile(DATA.getBinaryStream())); |
||||
|
} |
||||
|
|
||||
|
@Test |
||||
|
@SneakyThrows |
||||
|
void testGettingStreamWhenFileNotExist(){ |
||||
|
assertThrows(FileNotFoundException.class, ()->baseFileCacheService.getOtaDataStream(OTA_PACKAGE_ID)); |
||||
|
} |
||||
|
|
||||
|
@SneakyThrows |
||||
|
String calculateChecksumSHA256(InputStream stream) { |
||||
|
MessageDigest md = MessageDigest.getInstance("SHA-256"); |
||||
|
byte[] buffer = new byte[ONE_MEGA_BYTE]; |
||||
|
int count = 0; |
||||
|
while ((count = stream.read(buffer)) != -1) { |
||||
|
md.update(buffer, 0, count); |
||||
|
} |
||||
|
StringBuilder result = new StringBuilder(); |
||||
|
for (byte b : md.digest()) { |
||||
|
result.append(String.format("%02x", b)); |
||||
|
} |
||||
|
return result.toString(); |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue