1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
@file:Suppress("PropertyName", "LocalVariableName", "NOTHING_TO_INLINE")
package io.dico.parcels2.storage.exposed
import io.dico.parcels2.*
import io.dico.parcels2.Privilege.DEFAULT
import io.dico.parcels2.util.ext.alsoIfTrue
import kotlinx.coroutines.channels.SendChannel
import org.jetbrains.exposed.sql.*
object PrivilegesLocalT : PrivilegesTable<ParcelId>("parcels_privilege_local", ParcelsT)
object PrivilegesGlobalT : PrivilegesTable<PlayerProfile>("parcels_privilege_global", ProfilesT)
object ParcelOptionsT : Table("parcels_options") {
val parcel_id = integer("parcel_id").primaryKey().references(ParcelsT.id, ReferenceOption.CASCADE)
val interact_bitmask = binary("interact_bitmask", 4)
}
typealias PrivilegesSendChannel<AttachT> = SendChannel<Pair<AttachT, PrivilegesHolder>>
sealed class PrivilegesTable<AttachT>(name: String, val idTable: IdTransactionsTable<*, AttachT>) : Table(name) {
val attach_id = integer("attach_id").references(idTable.id, ReferenceOption.CASCADE)
val profile_id = integer("profile_id").references(ProfilesT.id, ReferenceOption.CASCADE)
val privilege = integer("privilege")
val index_pair = uniqueIndexR("index_pair", attach_id, profile_id)
fun setPrivilege(attachedOn: AttachT, player: PlayerProfile.Real, privilege: Privilege) {
privilege.requireNonTransient()
if (privilege == DEFAULT) {
val player_id = ProfilesT.getId(player) ?: return
idTable.getId(attachedOn)?.let { holder ->
deleteWhere { (attach_id eq holder) and (profile_id eq player_id) }
}
return
}
val holder = idTable.getOrInitId(attachedOn)
val player_id = ProfilesT.getOrInitId(player)
upsert(conflictIndex = index_pair) {
it[attach_id] = holder
it[profile_id] = player_id
it[this.privilege] = privilege.number
}
}
fun readPrivileges(id: Int): PrivilegesHolder? {
val list = slice(profile_id, privilege).select { attach_id eq id }
val result = PrivilegesHolder()
for (row in list) {
val profile = ProfilesT.getRealItem(row[profile_id]) ?: continue
result.setRawStoredPrivilege(profile, Privilege.getByNumber(row[privilege]) ?: continue)
}
return result
}
fun sendAllPrivilegesH(channel: PrivilegesSendChannel<AttachT>) {
val iterator = selectAll().orderBy(attach_id).iterator()
if (iterator.hasNext()) {
var row = iterator.next()
var id: Int = row[attach_id]
var attach: AttachT? = null
var map: PrivilegesHolder? = null
fun initAttachAndMap() {
attach = idTable.getItem(id)
map = attach?.let { PrivilegesHolder() }
}
fun sendIfPresent() {
if (attach != null && map != null) {
channel.offer(attach!! to map!!)
}
attach = null
map = null
}
initAttachAndMap()
do {
val rowId = row[attach_id]
if (rowId != id) {
sendIfPresent()
id = rowId
initAttachAndMap()
}
if (attach == null) {
continue // owner not found for this owner id
}
val profile = ProfilesT.getRealItem(row[profile_id])
if (profile == null) {
logger.error("Privilege from database is null, id ${row[profile_id]}")
continue
}
val privilege = Privilege.getByNumber(row[privilege])
if (privilege == null) {
logger.error("Privilege from database is null, number ${row[this.privilege]}")
continue
}
map!!.setRawStoredPrivilege(profile, privilege)
} while (iterator.hasNext().alsoIfTrue { row = iterator.next() })
sendIfPresent()
}
}
}
|