1 files changed, 16 insertions, 4 deletions
diff --git a/src/main/java/at/gv/egiz/moazs/repository/InMemoryDeliveryRepository.java b/src/main/java/at/gv/egiz/moazs/repository/InMemoryDeliveryRepository.java
index 570296f..58ec92c 100644
--- a/src/main/java/at/gv/egiz/moazs/repository/InMemoryDeliveryRepository.java
+++ b/src/main/java/at/gv/egiz/moazs/repository/InMemoryDeliveryRepository.java
@@ -30,6 +30,10 @@ public class InMemoryDeliveryRepository implements DeliveryRepository {
             .expireAfterWrite(30, TimeUnit.MINUTES)
             .build();
 
+    private static final Cache<String, byte[]> signedStatusRepository = CacheBuilder.newBuilder()
+            .expireAfterWrite(30, TimeUnit.MINUTES)
+            .build();
+
     @Override
     public void add(DeliveryRequestType request) {
         var key = request.getMetaData().getAppDeliveryID();
@@ -38,8 +42,7 @@ public class InMemoryDeliveryRepository implements DeliveryRepository {
 
     @Override
     public Optional<DeliveryRequestType> getDeliveryRequest(String appDeliveryID) {
-        var deliveryRequestType = requestRepository.getIfPresent(appDeliveryID);
-        return ofNullable(deliveryRequestType);
+        return ofNullable(requestRepository.getIfPresent(appDeliveryID));
     }
 
     @Override
@@ -51,7 +54,16 @@ public class InMemoryDeliveryRepository implements DeliveryRepository {
 
     @Override
     public Optional<DeliveryRequestStatusType> getDeliveryRequestStatus(String appDeliveryID) {
-        var deliveryRequestType = statusRepository.getIfPresent(appDeliveryID);
-        return ofNullable(deliveryRequestType);
+        return ofNullable(statusRepository.getIfPresent(appDeliveryID));
+    }
+
+    @Override
+    public void addSignedDeliveryRequestStatus(byte[] bytes, String appDeliveryId) {
+        signedStatusRepository.put(appDeliveryId, bytes);
+    }
+
+    @Override
+    public Optional<byte[]> getSignedDeliveryRequestStatus(String appDeliveryID) {
+        return ofNullable(signedStatusRepository.getIfPresent(appDeliveryID));
     }
 }