updated code

src/main/java/net/gepafin/tendermanagement/dao/DashboardDao.java

@@ -0,0 +1,118 @@
+package net.gepafin.tendermanagement.dao;
+
+import net.gepafin.tendermanagement.entities.CompanyEntity;
+import net.gepafin.tendermanagement.entities.UserEntity;
+import net.gepafin.tendermanagement.enums.CallStatusEnum;
+import net.gepafin.tendermanagement.enums.RoleStatusEnum;
+import net.gepafin.tendermanagement.enums.UserStatusEnum;
+import net.gepafin.tendermanagement.model.response.BeneficiaryWidgetResponseBean;
+import net.gepafin.tendermanagement.model.response.Widget1;
+import net.gepafin.tendermanagement.model.response.SuperAdminWidgetResponseBean;
+import net.gepafin.tendermanagement.repositories.ApplicationRepository;
+import net.gepafin.tendermanagement.repositories.CallRepository;
+import net.gepafin.tendermanagement.repositories.CompanyRepository;
+import net.gepafin.tendermanagement.repositories.UserRepository;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.math.BigDecimal;
+
+@Component
+public class DashboardDao {
+
+	@Autowired
+	private CallRepository callRepository;
+
+	@Autowired
+	private UserRepository userRepository;
+
+	@Autowired
+	private ApplicationRepository applicationRepository;
+
+	@Autowired
+	private CompanyRepository companyRepository;
+
+	public SuperAdminWidgetResponseBean getDashboardWidget() {
+		SuperAdminWidgetResponseBean widgetResponseBean = new SuperAdminWidgetResponseBean();
+		widgetResponseBean.setWidget1(createWidget1());
+//        List<Object[]> widgetBars = callRepository.findApplicationsPerCall();
+//        widgetResponseBean.setWidgetBars(widgetBars);
+		return widgetResponseBean;
+	}
+
+	private Widget1 createWidget1() {
+		Widget1 widget1 = initializeWidget1();
+
+		setActiveCalls(widget1);
+		setRegisteredUsers(widget1);
+		setTotalActiveFinancing(widget1);
+		setSubmittedApplications(widget1);
+		setDraftApplications(widget1);
+		setNumberOfCompanies(widget1);
+
+		return widget1;
+	}
+
+	private Widget1 initializeWidget1() {
+		return Widget1.builder().numberOfActiveCalls(0L).numberOfCompany(0L).numberOfDraftApplications(0L)
+				.numberOfResgisteredUsers(0L).numberOfSubmittedApplications(0L).totalActiveFinancing(BigDecimal.ZERO)
+				.build();
+	}
+
+	private void setActiveCalls(Widget1 widget1) {
+		Long activeCalls = callRepository.countByStatus(CallStatusEnum.PUBLISH.getValue());
+		if (activeCalls != null) {
+			widget1.setNumberOfActiveCalls(activeCalls);
+		}
+	}
+
+	private void setRegisteredUsers(Widget1 widget1) {
+		Long activeUsers = userRepository.countByStatusAndRoleEntityRoleType(UserStatusEnum.ACTIVE.getValue(),
+				RoleStatusEnum.ROLE_BENEFICIARY.getValue());
+		if (activeUsers != null) {
+			widget1.setNumberOfResgisteredUsers(activeUsers);
+		}
+	}
+
+	private void setTotalActiveFinancing(Widget1 widget1) {
+		BigDecimal totalActiveFinancing = callRepository.findTotalAmountOfPublishedCalls();
+		widget1.setTotalActiveFinancing(totalActiveFinancing);
+	}
+
+	private void setSubmittedApplications(Widget1 widget1) {
+		Long submittedApplications = applicationRepository.countSubmittedApplications();
+		if (submittedApplications != null) {
+			widget1.setNumberOfSubmittedApplications(submittedApplications);
+		}
+	}
+
+	private void setDraftApplications(Widget1 widget1) {
+		Long draftApplications = applicationRepository.countDraftApplications();
+		if (draftApplications != null) {
+			widget1.setNumberOfDraftApplications(draftApplications);
+		}
+	}
+
+	private void setNumberOfCompanies(Widget1 widget1) {
+		Long numberOfCompanies = companyRepository.countTotalCompanies();
+		if (numberOfCompanies != null) {
+			widget1.setNumberOfCompany(numberOfCompanies);
+		}
+	}
+
+	public BeneficiaryWidgetResponseBean getDashboardWidgetForBeneficiary(UserEntity userEntity,
+			CompanyEntity company) {
+		BeneficiaryWidgetResponseBean beneficiaryWidgetResponseBean = BeneficiaryWidgetResponseBean.builder()
+				.numberOfApplications(0L).numberOfCalls(0L).numberOfIntegratedDocuments(0L).build();
+		Long activeCalls = callRepository.countByStatus(CallStatusEnum.PUBLISH.getValue());
+		if (activeCalls != null) {
+			beneficiaryWidgetResponseBean.setNumberOfCalls(activeCalls);
+		}
+		Long activeApplication = applicationRepository.countSubmittedApplicationsByUserId(userEntity.getId(),
+				company.getId());
+		if (activeApplication != null) {
+			beneficiaryWidgetResponseBean.setNumberOfApplications(activeApplication);
+		}
+		return beneficiaryWidgetResponseBean;
+	}
+}