Swiftgram/submodules/Postbox/Sources/MessageHistoryFailedTable.swift

import Foundation

final class MessageHistoryFailedTable: Table {
    static func tableSpec(_ id: Int32) -> ValueBoxTable {
        return ValueBoxTable(id: id, keyType: .binary, compactValuesOnCreation: true)
    }
    
    private let sharedKey = ValueBoxKey(length: 8 + 4 + 4)
    
    private(set) var updatedPeerIds = Set<PeerId>()
    private(set) var updatedMessageIds = Set<MessageId>()

    private func key(_ id: MessageId) -> ValueBoxKey {
        self.sharedKey.setInt64(0, value: id.peerId.toInt64())
        self.sharedKey.setInt32(8, value: id.namespace)
        self.sharedKey.setInt32(8 + 4, value: id.id)
        
        return self.sharedKey
    }
    
    private func lowerBound(peerId: PeerId) -> ValueBoxKey {
        let key = ValueBoxKey(length: 8)
        key.setInt64(0, value: peerId.toInt64())
        return key
    }
    
    private func upperBound(peerId: PeerId) -> ValueBoxKey {
        let key = ValueBoxKey(length: 8)
        key.setInt64(0, value: peerId.toInt64())
        return key.successor
    }
    
    func add(_ id: MessageId) {
        self.valueBox.set(self.table, key: self.key(id), value: MemoryBuffer())
        self.updatedPeerIds.insert(id.peerId)
        self.updatedMessageIds.insert(id)
    }
    
    func remove(_ id: MessageId) {
        self.valueBox.remove(self.table, key: self.key(id), secure: false)
        self.updatedPeerIds.insert(id.peerId)
        self.updatedMessageIds.remove(id)
    }
    
    func get(peerId: PeerId) -> [MessageId] {
        var ids:[MessageId] = []
        self.valueBox.range(self.table, start: self.lowerBound(peerId: peerId), end: self.upperBound(peerId: peerId), keys: { key in
            
            let peerId = PeerId(key.getInt64(0))
            let namespace = key.getInt32(8)
            let id = key.getInt32(8 + 4)
            ids.append(MessageId(peerId: peerId, namespace: namespace, id: id))
            
            return false
        }, limit: 100)
        
        self.updatedMessageIds = self.updatedMessageIds.union(ids)
        
        return ids
    }
    
    func contains(peerId: PeerId) -> Bool {
        var result = false
        self.valueBox.range(self.table, start: self.lowerBound(peerId: peerId), end: self.upperBound(peerId: peerId), keys: { _ in
            result = true
            return false
        }, limit: 1)
        return result
    }
    
    override func beforeCommit() {
        self.updatedPeerIds.removeAll()
    }
}