From 66ca3e74a516f3164126ad8ddb0543e7c6083238 Mon Sep 17 00:00:00 2001 From: Tor-Einar Skog <tor-einar.skog@nibio.no> Date: Tue, 11 Jun 2019 11:05:32 +0200 Subject: [PATCH] Adding multi org to forecast summaries and forecast config list --- .../controller/session/ForecastBean.java | 40 ++++++++++++++--- .../logic/entity/ForecastConfiguration.java | 4 +- .../vips/logic/service/LogicService.java | 44 +++++++++++++++++++ 3 files changed, 82 insertions(+), 6 deletions(-) diff --git a/src/main/java/no/nibio/vips/logic/controller/session/ForecastBean.java b/src/main/java/no/nibio/vips/logic/controller/session/ForecastBean.java index b6553a33..2a0a276b 100755 --- a/src/main/java/no/nibio/vips/logic/controller/session/ForecastBean.java +++ b/src/main/java/no/nibio/vips/logic/controller/session/ForecastBean.java @@ -344,6 +344,29 @@ public class ForecastBean { } } + public List<ForecastConfiguration> getForecastConfigurations(List<Integer> organizationIds, Date from, Date to) + { + List<VipsLogicUser> organizationUsers = em + .createNamedQuery("VipsLogicUser.findByOrganizationIds") + .setParameter("organizationIds", organizationIds) + .getResultList(); + + + if(!organizationUsers.isEmpty()) + { + return em + .createNamedQuery("ForecastConfiguration.findByVipsLogicUserIdsAndDate") + .setParameter("vipsLogicUserIds", organizationUsers) + .setParameter("from", from) + .setParameter("to", to) + .getResultList(); + } + else + { + return new ArrayList<>(); + } + } + public List<ForecastConfiguration> getForecastConfigurationsByWeatherStation(PointOfInterestWeatherStation weatherStation) { return em @@ -938,11 +961,18 @@ public class ForecastBean { summaryForForecast.add(s); mappedSummaries.put(s.getForecastSummaryPK().getForecastConfigurationId(), summaryForForecast); }); - List<ForecastConfiguration> configurations = em.createNamedQuery("ForecastConfiguration.findByForecastConfigurationIds").setParameter("forecastConfigurationIds", mappedSummaries.keySet()).getResultList(); - configurations.forEach((conf) -> { - conf.setForecastSummaries(mappedSummaries.get(conf.getForecastConfigurationId())); - }); - return configurations; + if(mappedSummaries.size() > 0) + { + List<ForecastConfiguration> configurations = em.createNamedQuery("ForecastConfiguration.findByForecastConfigurationIds").setParameter("forecastConfigurationIds", mappedSummaries.keySet()).getResultList(); + configurations.forEach((conf) -> { + conf.setForecastSummaries(mappedSummaries.get(conf.getForecastConfigurationId())); + }); + return configurations; + } + else + { + return new ArrayList<>(); + } } /** diff --git a/src/main/java/no/nibio/vips/logic/entity/ForecastConfiguration.java b/src/main/java/no/nibio/vips/logic/entity/ForecastConfiguration.java index de0e9041..730f6a7c 100755 --- a/src/main/java/no/nibio/vips/logic/entity/ForecastConfiguration.java +++ b/src/main/java/no/nibio/vips/logic/entity/ForecastConfiguration.java @@ -81,7 +81,9 @@ import org.hibernate.annotations.TypeDefs; @NamedQuery(name = "ForecastConfiguration.findPrivateByVipsLogicUserIdAndCropOrganismIdsAndDate", query = "SELECT f FROM ForecastConfiguration f WHERE f.vipsLogicUserId = :vipsLogicUserId AND f.cropOrganismId.organismId IN (:cropOrganismIds) AND f.dateStart <= :to AND f.dateEnd >= :from AND f.isPrivate = TRUE"), @NamedQuery(name = "ForecastConfiguration.findByVipsLogicUserIds", query = "SELECT f FROM ForecastConfiguration f WHERE f.vipsLogicUserId IN (:vipsLogicUserIds) AND f.isPrivate = FALSE"), @NamedQuery(name = "ForecastConfiguration.findByVipsLogicUserIdsAndModelIds", query = "SELECT f FROM ForecastConfiguration f WHERE f.vipsLogicUserId IN (:vipsLogicUserIds) AND f.modelId IN (:modelIds) AND f.isPrivate = FALSE"), - @NamedQuery(name = "ForecastConfiguration.findByVipsLogicUserIdsAndModelIdsAndDate", query = "SELECT f FROM ForecastConfiguration f WHERE f.vipsLogicUserId IN (:vipsLogicUserIds) AND f.modelId IN (:modelIds) AND f.dateStart <= :to AND f.dateEnd >= :from AND f.isPrivate = FALSE")}) + @NamedQuery(name = "ForecastConfiguration.findByVipsLogicUserIdsAndModelIdsAndDate", query = "SELECT f FROM ForecastConfiguration f WHERE f.vipsLogicUserId IN (:vipsLogicUserIds) AND f.modelId IN (:modelIds) AND f.dateStart <= :to AND f.dateEnd >= :from AND f.isPrivate = FALSE"), + @NamedQuery(name = "ForecastConfiguration.findByVipsLogicUserIdsAndDate", query = "SELECT f FROM ForecastConfiguration f WHERE f.vipsLogicUserId IN (:vipsLogicUserIds) AND f.dateStart <= :to AND f.dateEnd >= :from AND f.isPrivate = FALSE") +}) public class ForecastConfiguration implements Serializable, Comparable { @OneToMany(cascade = CascadeType.ALL, mappedBy = "forecastConfiguration", fetch = FetchType.EAGER) private Set<ForecastModelConfiguration> forecastModelConfigurationSet; diff --git a/src/main/java/no/nibio/vips/logic/service/LogicService.java b/src/main/java/no/nibio/vips/logic/service/LogicService.java index 5140cd36..43dadd4e 100755 --- a/src/main/java/no/nibio/vips/logic/service/LogicService.java +++ b/src/main/java/no/nibio/vips/logic/service/LogicService.java @@ -245,6 +245,10 @@ public class LogicService { try { Integer includeOrgId = Integer.valueOf(orgId); + if(includeOrgId.equals(organizationId)) + { + continue; + } summaries.addAll(SessionControllerGetter.getForecastBean().getForecastConfigurationSummaries(includeOrgId)); } catch(NumberFormatException ex){} @@ -335,6 +339,46 @@ public class LogicService { } + @GET + @Path("forecastconfigurationsincludeorgs/{organizationId}") + @GZIP + @Produces("application/json;charset=UTF-8") + public Response getActiveForecastConfigurationsWithIncludeOrganizations( + @PathParam("organizationId") Integer organizationId, + @QueryParam("includeOrganizationIds") String includeOrganizationIds + ) + { + Calendar cal = Calendar.getInstance(); + cal.setTime(SystemTime.getSystemTime()); + cal.set(cal.get(Calendar.YEAR), Calendar.JANUARY, 1, 0, 0, 0); + Date from = cal.getTime(); + cal.set(cal.get(Calendar.YEAR), Calendar.DECEMBER, 31, 23, 0, 0); + Date to = cal.getTime(); + + + List<Integer> orgIds = new ArrayList<>(); + orgIds.add(organizationId); + if(includeOrganizationIds != null) + { + String[] includeOrgIdStrs = includeOrganizationIds.split(","); + for(String orgIdStr:includeOrgIdStrs) + { + try + { + Integer includeOrgId = Integer.valueOf(orgIdStr.trim()); + if(includeOrgId.equals(organizationId)) + { + continue; + } + orgIds.add(includeOrgId); + } + catch(NumberFormatException ex){} + } + } + List<ForecastConfiguration> forecastConfigs = SessionControllerGetter.getForecastBean().getForecastConfigurations(orgIds, from, to); + return Response.ok().entity(forecastConfigs).build(); + } + /** * Returns a list of forecasts for given organization * @param organizationId -- GitLab