Prechádzať zdrojové kódy

Status list includes persons, birth certificates and census entries

Andrew Grant 2 mesiacov pred
rodič
commit
979b589685

+ 2 - 0
src/main/java/scot/carricksoftware/grantswriter/repositories/ReadOnlyRepository.java

@@ -16,4 +16,6 @@ public interface ReadOnlyRepository<T, ID> extends Repository<T, ID> {
 
     List<T> findAll(Sort sort);
 
+    Integer count();
+
 }

+ 20 - 3
src/main/java/scot/carricksoftware/grantswriter/services/status/StatusServiceImpl.java

@@ -8,6 +8,9 @@ package scot.carricksoftware.grantswriter.services.status;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 import org.springframework.stereotype.Service;
+import scot.carricksoftware.grantswriter.repositories.censusentry.CensusEntryRepository;
+import scot.carricksoftware.grantswriter.repositories.certificates.birthcertificate.BirthCertificateRepository;
+import scot.carricksoftware.grantswriter.repositories.people.PersonRepository;
 
 import java.util.LinkedHashMap;
 
@@ -15,21 +18,35 @@ import java.util.LinkedHashMap;
 public class StatusServiceImpl implements StatusService {
 
     private static final Logger logger = LogManager.getLogger(StatusServiceImpl.class);
+    private final PersonRepository personRepository;
+    private final BirthCertificateRepository birthCertificateRepository;
+    private final CensusEntryRepository censusEntryRepository;
+
 
     private final LinkedHashMap<String, String> status = new LinkedHashMap<>();
 
+    public StatusServiceImpl(PersonRepository personRepository,
+                             BirthCertificateRepository birthCertificateRepository,
+                             CensusEntryRepository censusEntryRepository) {
+        this.personRepository = personRepository;
+        this.birthCertificateRepository = birthCertificateRepository;
+        this.censusEntryRepository = censusEntryRepository;
+    }
+
+
     @Override
     public LinkedHashMap<String, String> getStatus() {
         logger.info("StatusService::getStatus");
 
-        status.put("People", "0");
-        status.put("Birth Certificates", "0");
+        status.put("People", personRepository.count().toString());
+        status.put("Birth Certificates", birthCertificateRepository.count().toString());
         status.put("Marriage Certificates", "0");
         status.put("Divorce Certificates", "0");
         status.put("Death Certificates", "0");
         status.put("", "");
         status.put("Censuses", "0");
-        status.put("Census Entries", "0");
+        status.put("Census Entries", censusEntryRepository.count().toString());
         return status;
     }
+
 }