refactor and cleanup [skip ci]

2025-12-22 22:25:57 +00:00 · 2019-11-01 17:11:12 +04:00
parent 66191348f3
commit 8f66c10ac1
786 changed files with 983 additions and 69155 deletions
--- a/submodules/Postbox/Sources/MetadataTable.swift
+++ b/submodules/Postbox/Sources/MetadataTable.swift
@@ -0,0 +1,144 @@
+import Foundation
+
+private enum MetadataKey: Int32 {
+    case UserVersion = 1
+    case State = 2
+    case TransactionStateVersion = 3
+    case MasterClientId = 4
+    case AccessChallenge = 5
+    case RemoteContactCount = 6
+}
+
+final class MetadataTable: Table {
+    static func tableSpec(_ id: Int32) -> ValueBoxTable {
+        return ValueBoxTable(id: id, keyType: .int64, compactValuesOnCreation: false)
+    }
+    
+    private var cachedState: PostboxCoding?
+    private var cachedRemoteContactCount: Int32?
+    
+    private let sharedBuffer = WriteBuffer()
+    
+    override init(valueBox: ValueBox, table: ValueBoxTable) {
+        super.init(valueBox: valueBox, table: table)
+    }
+    
+    private func key(_ key: MetadataKey) -> ValueBoxKey {
+        let valueBoxKey = ValueBoxKey(length: 8)
+        valueBoxKey.setInt64(0, value: Int64(key.rawValue))
+        return valueBoxKey
+    }
+    
+    func userVersion() -> Int32? {
+        if let value = self.valueBox.get(self.table, key: self.key(.UserVersion)) {
+            var version: Int32 = 0
+            value.read(&version, offset: 0, length: 4)
+            return version
+        }
+        return nil
+    }
+    
+    func setUserVersion(_ version: Int32) {
+        sharedBuffer.reset()
+        let buffer = sharedBuffer
+        var varVersion: Int32 = version
+        buffer.write(&varVersion, offset: 0, length: 4)
+        self.valueBox.set(self.table, key: self.key(.UserVersion), value: buffer)
+    }
+    
+    func state() -> PostboxCoding? {
+        if let cachedState = self.cachedState {
+            return cachedState
+        } else {
+            if let value = self.valueBox.get(self.table, key: self.key(.State)) {
+                if let state = PostboxDecoder(buffer: value).decodeRootObject() {
+                    self.cachedState = state
+                    return state
+                }
+            }
+            return nil
+        }
+    }
+    
+    func setState(_ state: PostboxCoding) {
+        self.cachedState = state
+        
+        let encoder = PostboxEncoder()
+        encoder.encodeRootObject(state)
+        withExtendedLifetime(encoder, {
+            self.valueBox.set(self.table, key: self.key(.State), value: encoder.readBufferNoCopy())
+        })
+    }
+    
+    func transactionStateVersion() -> Int64 {
+        if let value = self.valueBox.get(self.table, key: self.key(.TransactionStateVersion)) {
+            var version: Int64 = 0
+            value.read(&version, offset: 0, length: 8)
+            return version
+        } else {
+            return 0
+        }
+    }
+    
+    func incrementTransactionStateVersion() -> Int64 {
+        var version = self.transactionStateVersion() + 1
+        sharedBuffer.reset()
+        let buffer = sharedBuffer
+        buffer.write(&version, offset: 0, length: 8)
+        self.valueBox.set(self.table, key: self.key(.TransactionStateVersion), value: buffer)
+        return version
+    }
+    
+    func masterClientId() -> Int64 {
+        if let value = self.valueBox.get(self.table, key: self.key(.MasterClientId)) {
+            var clientId: Int64 = 0
+            value.read(&clientId, offset: 0, length: 8)
+            return clientId
+        } else {
+            return 0
+        }
+    }
+    
+    func setMasterClientId(_ id: Int64) {
+        sharedBuffer.reset()
+        let buffer = sharedBuffer
+        var clientId = id
+        buffer.write(&clientId, offset: 0, length: 8)
+        self.valueBox.set(self.table, key: self.key(.MasterClientId), value: buffer)
+    }
+    
+    func accessChallengeData() -> PostboxAccessChallengeData {
+        if let value = self.valueBox.get(self.table, key: self.key(.AccessChallenge)) {
+            return PostboxAccessChallengeData(decoder: PostboxDecoder(buffer: value))
+        } else {
+            return .none
+        }
+    }
+    
+    func setRemoteContactCount(_ count: Int32) {
+        self.cachedRemoteContactCount = count
+        var mutableCount: Int32 = count
+        self.valueBox.set(self.table, key: self.key(.RemoteContactCount), value: MemoryBuffer(memory: &mutableCount, capacity: 4, length: 4, freeWhenDone: false))
+    }
+    
+    func getRemoteContactCount() -> Int32 {
+        if let cachedRemoteContactCount = self.cachedRemoteContactCount {
+            return cachedRemoteContactCount
+        } else {
+            if let value = self.valueBox.get(self.table, key: self.key(.RemoteContactCount)) {
+                var count: Int32 = 0
+                value.read(&count, offset: 0, length: 4)
+                self.cachedRemoteContactCount = count
+                return count
+            } else {
+                self.cachedRemoteContactCount = 0
+                return 0
+            }
+        }
+    }
+    
+    override func clearMemoryCache() {
+        self.cachedState = nil
+        self.cachedRemoteContactCount = nil
+    }
+}