Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add query and controller endpoint for usage data #102

Merged
merged 1 commit into from
Oct 16, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,14 +1,18 @@
package nz.govt.eop.plan_limits

import java.time.LocalDate
import java.time.temporal.ChronoUnit.DAYS
import java.util.concurrent.TimeUnit
import org.jooq.*
import org.springframework.http.CacheControl
import org.springframework.http.HttpStatus
import org.springframework.http.MediaType
import org.springframework.http.ResponseEntity
import org.springframework.stereotype.Controller
import org.springframework.web.bind.annotation.RequestMapping
import org.springframework.web.bind.annotation.RequestParam
import org.springframework.web.bind.annotation.ResponseBody
import org.springframework.web.server.ResponseStatusException

@Controller
class Controller(val context: DSLContext, val queries: Queries, val manifest: Manifest) {
Expand Down Expand Up @@ -79,4 +83,28 @@ class Controller(val context: DSLContext, val queries: Queries, val manifest: Ma
.cacheControl(CacheControl.maxAge(365, TimeUnit.DAYS))
.body(queries.flowLimits(councilId))
}

@RequestMapping("/plan-limits/weekly-usage", produces = [MediaType.APPLICATION_JSON_VALUE])
@ResponseBody
fun weeklyUsage(
@RequestParam(name = "councilId") councilId: Int,
@RequestParam("from") from: LocalDate,
@RequestParam("to") to: LocalDate,
@RequestParam("areaId") areaId: String?
): ResponseEntity<String> {
if (from >= to) {
throw ResponseStatusException(
HttpStatus.BAD_REQUEST, "The parameter \"to\" must be after \"from\"")
}
val difference = DAYS.between(from, to)
if (difference < 1 || difference >= 366) {
throw ResponseStatusException(
HttpStatus.BAD_REQUEST,
"The duration between \"from\" and \"to\" should be more than one day and at most one year")
}

return ResponseEntity.ok()
.cacheControl(CacheControl.maxAge(365, TimeUnit.DAYS))
.body(queries.weeklyUsage(councilId, from, to, areaId))
}
}
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
package nz.govt.eop.plan_limits

import java.time.LocalDate
import nz.govt.eop.si.jooq.tables.Councils.Companion.COUNCILS
import nz.govt.eop.si.jooq.tables.FlowLimits.Companion.FLOW_LIMITS
import nz.govt.eop.si.jooq.tables.FlowMeasurementSites.Companion.FLOW_MEASUREMENT_SITES
Expand All @@ -8,8 +9,10 @@ import nz.govt.eop.si.jooq.tables.GroundwaterLimits.Companion.GROUNDWATER_LIMITS
import nz.govt.eop.si.jooq.tables.PlanRegions.Companion.PLAN_REGIONS
import nz.govt.eop.si.jooq.tables.Plans.Companion.PLANS
import nz.govt.eop.si.jooq.tables.SurfaceWaterLimits.Companion.SURFACE_WATER_LIMITS
import nz.govt.eop.si.jooq.tables.WaterAllocationAndUsageByArea.Companion.WATER_ALLOCATION_AND_USAGE_BY_AREA
import nz.govt.eop.si.jooq.tables.WaterAllocationsByArea.Companion.WATER_ALLOCATIONS_BY_AREA
import org.jooq.*
import org.jooq.impl.DSL
import org.jooq.impl.DSL.*
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.stereotype.Component
Expand Down Expand Up @@ -152,6 +155,34 @@ class Queries(@Autowired val context: DSLContext) {
return buildFeatureCollection(context, innerQuery)
}

fun weeklyUsage(councilId: Int, from: LocalDate, to: LocalDate, areaId: String? = null): String {
var whereCondition = DSL.noCondition()
if (areaId != null) {
whereCondition = whereCondition.and(WATER_ALLOCATION_AND_USAGE_BY_AREA.AREA_ID.eq(areaId))
}

val innerQuery =
select(
WATER_ALLOCATION_AND_USAGE_BY_AREA.AREA_ID,
sum(WATER_ALLOCATION_AND_USAGE_BY_AREA.ALLOCATION).`as`("total_allocation"),
sum(WATER_ALLOCATION_AND_USAGE_BY_AREA.ALLOCATION_DAILY)
.`as`("metered_daily_allocation"),
sum(WATER_ALLOCATION_AND_USAGE_BY_AREA.METERED_ALLOCATION_YEARLY)
.`as`("metered_yearly_allocation"),
sum(WATER_ALLOCATION_AND_USAGE_BY_AREA.DAILY_USAGE).`as`("daily_usage"))
.from(WATER_ALLOCATION_AND_USAGE_BY_AREA)
.where(whereCondition)
.and(WATER_ALLOCATION_AND_USAGE_BY_AREA.DATE.ge(from))
.and(WATER_ALLOCATION_AND_USAGE_BY_AREA.DATE.le(to))
.groupBy(WATER_ALLOCATION_AND_USAGE_BY_AREA.AREA_ID)

val featureCollection: Field<JSONB> =
coalesce(function("json_agg", JSONB::class.java, field("inputs")), jsonArray())

val result = context.select(featureCollection).from(innerQuery.asTable("inputs")).fetch()
return result.firstNotNullOf { it.value1().toString() }
}

private fun <R : Record> buildFeatureCollection(
dslContext: DSLContext,
innerQuery: Select<R>
Expand Down