141 lines
4.1 KiB
Kotlin
141 lines
4.1 KiB
Kotlin
package fr.dcproject.repository
|
|
|
|
import fr.dcproject.entity.*
|
|
import fr.postgresjson.connexion.Paginated
|
|
import fr.postgresjson.connexion.Requester
|
|
import fr.postgresjson.entity.immutable.UuidEntity
|
|
import fr.postgresjson.repository.RepositoryI
|
|
import kotlinx.coroutines.flow.Flow
|
|
import kotlinx.coroutines.flow.flow
|
|
import java.util.*
|
|
import fr.dcproject.entity.Article as ArticleEntity
|
|
import fr.dcproject.entity.Constitution as ConstitutionEntity
|
|
import fr.dcproject.entity.Follow as FollowEntity
|
|
|
|
sealed class Follow<IN : TargetRef, OUT : TargetRef>(override var requester: Requester) : RepositoryI {
|
|
open fun findByCitizen(
|
|
citizen: CitizenI,
|
|
page: Int = 1,
|
|
limit: Int = 50
|
|
): Paginated<FollowEntity<OUT>> =
|
|
findByCitizen(citizen.id, page, limit)
|
|
|
|
open fun findByCitizen(
|
|
citizenId: UUID,
|
|
page: Int = 1,
|
|
limit: Int = 50
|
|
): Paginated<FollowEntity<OUT>> {
|
|
return requester
|
|
.getFunction("find_follows_by_citizen")
|
|
.select(
|
|
page, limit,
|
|
"created_by_id" to citizenId
|
|
)
|
|
}
|
|
|
|
fun follow(follow: FollowEntity<IN>) {
|
|
requester
|
|
.getFunction("follow")
|
|
.sendQuery(
|
|
"reference" to follow.target.reference,
|
|
"target_id" to follow.target.id,
|
|
"created_by_id" to follow.createdBy.id
|
|
)
|
|
}
|
|
|
|
fun unfollow(follow: FollowEntity<IN>) {
|
|
requester
|
|
.getFunction("unfollow")
|
|
.sendQuery(
|
|
"reference" to follow.target.reference,
|
|
"target_id" to follow.target.id,
|
|
"created_by_id" to follow.createdBy.id
|
|
)
|
|
}
|
|
|
|
open fun findFollow(
|
|
citizen: CitizenI,
|
|
target: TargetRef
|
|
): FollowEntity<OUT>? =
|
|
requester
|
|
.getFunction("find_follow")
|
|
.selectOne(
|
|
"citizen_id" to citizen.id,
|
|
"target_id" to target.id,
|
|
"target_reference" to target.reference
|
|
)
|
|
|
|
fun findFollowsByTarget(
|
|
target: UuidEntity,
|
|
bulkSize: Int = 300
|
|
): Flow<FollowSimple<IN, CitizenRef>> = flow {
|
|
var nextPage = 1
|
|
do {
|
|
val paginate = findFollowsByTarget(target, nextPage, bulkSize)
|
|
paginate.result.forEach {
|
|
emit(it)
|
|
}
|
|
nextPage = paginate.currentPage + 1
|
|
} while (!paginate.isLastPage())
|
|
}
|
|
|
|
abstract fun findFollowsByTarget(
|
|
target: UuidEntity,
|
|
page: Int = 1,
|
|
limit: Int = 300
|
|
): Paginated<FollowSimple<IN, CitizenRef>>
|
|
}
|
|
|
|
class FollowArticle(requester: Requester) : Follow<ArticleRef, ArticleEntity>(requester) {
|
|
override fun findByCitizen(
|
|
citizenId: UUID,
|
|
page: Int,
|
|
limit: Int
|
|
): Paginated<FollowEntity<ArticleEntity>> {
|
|
return requester.run {
|
|
getFunction("find_follows_article_by_citizen")
|
|
.select(
|
|
page, limit,
|
|
"created_by_id" to citizenId
|
|
)
|
|
}
|
|
}
|
|
|
|
override fun findFollowsByTarget(
|
|
target: UuidEntity,
|
|
page: Int,
|
|
limit: Int
|
|
): Paginated<FollowSimple<ArticleRef, CitizenRef>> {
|
|
return requester
|
|
.getFunction("find_follows_article_by_target")
|
|
.select(
|
|
page, limit,
|
|
"target_id" to target.id
|
|
)
|
|
}
|
|
}
|
|
|
|
class FollowConstitution(requester: Requester) : Follow<ConstitutionRef, ConstitutionEntity>(requester) {
|
|
override fun findByCitizen(
|
|
citizenId: UUID,
|
|
page: Int,
|
|
limit: Int
|
|
): Paginated<FollowEntity<ConstitutionEntity>> {
|
|
return requester.run {
|
|
getFunction("find_follows_constitution_by_citizen")
|
|
.select(
|
|
page, limit,
|
|
"created_by_id" to citizenId
|
|
)
|
|
}
|
|
}
|
|
|
|
override fun findFollowsByTarget(
|
|
target: UuidEntity,
|
|
page: Int,
|
|
limit: Int
|
|
): Paginated<FollowSimple<ConstitutionRef, CitizenRef>> {
|
|
TODO("Not yet implemented")
|
|
}
|
|
}
|