wbrawner/twigs
1
0
Fork 0
Code Issues Pull requests 2 Projects Releases Packages 1 Wiki Activity Actions

Add paging and sorting to getCategories

Browse source 
Signed-off-by: Billy Brawner <billy@wbrawner.com>
This commit is contained in:
Billy Brawner 2019-10-07 20:34:10 -07:00
parent 5a41cacd67
commit 93be762dad
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
src/main/kotlin/com/wbrawner/budgetserver/category/CategoryController.kt
View file

@ -11,6 +11,7 @@ import io.swagger.annotations.Authorization
import org.hibernate.Hibernate
import org.springframework.beans.factory.annotation.Autowired
import org.springframework.data.domain.PageRequest
import org.springframework.data.domain.Sort
import org.springframework.http.MediaType
import org.springframework.http.ResponseEntity
import org.springframework.web.bind.annotation.*
@ -29,13 +30,22 @@ class CategoryController @Autowired constructor(
@Transactional
@GetMapping("", produces = [MediaType.APPLICATION_JSON_VALUE])
@ApiOperation(value = "getCategories", nickname = "getCategories", tags = ["Categories"])
fun getCategories(budgetId: Long? = null, count: Int?, page: Int?): ResponseEntity<List<CategoryResponse>> {
fun getCategories(budgetId: Long? = null,
count: Int?,
page: Int?,
sortBy: String?,
sortOrder: Sort.Direction?
): ResponseEntity<List<CategoryResponse>> {
val budgets = if (budgetId != null) {
budgetRepository.findAllById(listOf(budgetId))
} else {
budgetRepository.findAllByUsersContainsOrOwner(getCurrentUser()!!)
}.toList()
val pageRequest = PageRequest.of(min(0, page?.minus(1)?: 0), count?: 1000)
val pageRequest = PageRequest.of(
min(0, page?.minus(1)?: 0),
count?: 1000,
Sort(sortOrder?: Sort.Direction.ASC, sortBy?: "title")
)
return ResponseEntity.ok(categoryRepository.findAllByBudgetIn(budgets, pageRequest).map { CategoryResponse(it) })
}