mirror of
https://github.com/Swiftgram/Telegram-iOS.git
synced 2025-12-23 22:55:00 +00:00
Support refreshDueToExternalTransaction in more views
This commit is contained in:
@@ -5,7 +5,7 @@ final class PeerChatStateTable: Table {
|
||||
return ValueBoxTable(id: id, keyType: .int64, compactValuesOnCreation: false)
|
||||
}
|
||||
|
||||
private var cachedPeerChatStates: [PeerId: PostboxCoding?] = [:]
|
||||
private var cachedPeerChatStates: [PeerId: CodableEntry?] = [:]
|
||||
private var updatedPeerIds = Set<PeerId>()
|
||||
|
||||
private let sharedKey = ValueBoxKey(length: 8)
|
||||
@@ -15,11 +15,12 @@ final class PeerChatStateTable: Table {
|
||||
return self.sharedKey
|
||||
}
|
||||
|
||||
func get(_ id: PeerId) -> PostboxCoding? {
|
||||
func get(_ id: PeerId) -> CodableEntry? {
|
||||
if let state = self.cachedPeerChatStates[id] {
|
||||
return state
|
||||
} else {
|
||||
if let value = self.valueBox.get(self.table, key: self.key(id)), let state = PostboxDecoder(buffer: value).decodeRootObject() {
|
||||
if let value = self.valueBox.get(self.table, key: self.key(id)) {
|
||||
let state = CodableEntry(data: value.makeData())
|
||||
self.cachedPeerChatStates[id] = state
|
||||
return state
|
||||
} else {
|
||||
@@ -29,7 +30,7 @@ final class PeerChatStateTable: Table {
|
||||
}
|
||||
}
|
||||
|
||||
func set(_ id: PeerId, state: PostboxCoding?) {
|
||||
func set(_ id: PeerId, state: CodableEntry?) {
|
||||
self.cachedPeerChatStates[id] = state
|
||||
self.updatedPeerIds.insert(id)
|
||||
}
|
||||
@@ -41,12 +42,9 @@ final class PeerChatStateTable: Table {
|
||||
|
||||
override func beforeCommit() {
|
||||
if !self.updatedPeerIds.isEmpty {
|
||||
let sharedEncoder = PostboxEncoder()
|
||||
for id in self.updatedPeerIds {
|
||||
if let wrappedState = self.cachedPeerChatStates[id], let state = wrappedState {
|
||||
sharedEncoder.reset()
|
||||
sharedEncoder.encodeRootObject(state)
|
||||
self.valueBox.set(self.table, key: self.key(id), value: sharedEncoder.readBufferNoCopy())
|
||||
self.valueBox.set(self.table, key: self.key(id), value: ReadBuffer(data: state.data))
|
||||
} else {
|
||||
self.valueBox.remove(self.table, key: self.key(id), secure: false)
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user