ChildRegisterService.java

package org.ei.drishti.service;

import org.ei.drishti.domain.Child;
import org.ei.drishti.domain.EligibleCouple;
import org.ei.drishti.domain.Mother;
import org.ei.drishti.domain.register.ChildRegister;
import org.ei.drishti.domain.register.ChildRegisterEntry;
import org.ei.drishti.repository.AllChildren;
import org.ei.drishti.repository.AllEligibleCouples;
import org.ei.drishti.repository.AllMothers;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import static ch.lambdaj.Lambda.*;
import static org.hamcrest.Matchers.equalTo;

@Service
public class ChildRegisterService {
    private final AllChildren allChildren;
    private AllMothers allMothers;
    private final AllEligibleCouples allEligibleCouples;

    @Autowired
    public ChildRegisterService(AllChildren allChildren,
                                AllMothers allMothers, AllEligibleCouples allEligibleCouples) {
        this.allChildren = allChildren;
        this.allMothers = allMothers;
        this.allEligibleCouples = allEligibleCouples;
    }

    public ChildRegister getRegisterForANM(String anmIdentifier) {
        ArrayList<ChildRegisterEntry> childRegisterEntries = new ArrayList<>();
        List<Child> children = allChildren.findAllOpenChildrenForANM(anmIdentifier);
        Collection<String> motherIds = selectDistinct(collect(children, on(Child.class).motherCaseId()));
        List<String> motherIdsList = new ArrayList<>();
        motherIdsList.addAll(motherIds);
        List<Mother> mothers = allMothers.findAll(motherIdsList);
        Collection<String> ecIDs =  selectDistinct(collect(mothers, on(Mother.class).ecCaseId()));
        List<String> ecIdsList = new ArrayList<>();
        ecIdsList.addAll(ecIDs);
        List<EligibleCouple> ecs = allEligibleCouples.findAll(ecIdsList);
        for (Child child : children) {
            Mother mother = selectUnique(mothers,
                    having(on(Mother.class).caseId(), equalTo(child.motherCaseId())));
            EligibleCouple ec = selectUnique(ecs,
                    having(on(EligibleCouple.class).caseId(), equalTo(mother.ecCaseId())));
            ChildRegisterEntry entry = new ChildRegisterEntry()
                    .withThayiCardNumber(mother.thayiCardNumber())
                    .withWifeName(ec.wifeName())
                    .withHusbandName(ec.husbandName())
                    .withWifeDOB(ec.wifeDOB().toString())
                    .withVillage(ec.location().village())
                    .withSubCenter(ec.location().subCenter())
                    .withDOB(child.dateOfBirth())
                    .withImmunizations(child.immunizations())
                    .withVitaminADoses(child.vitaminADoses());
            childRegisterEntries.add(entry);
        }
        return new ChildRegister(childRegisterEntries);
    }
}