Temp

submodules/TelegramUI/Sources/ChatMessageThrottledProcessingManager.swift

@@ -0,0 +1,112 @@
+import Foundation
+import UIKit
+import Postbox
+import SwiftSignalKit
+
+final class ChatMessageThrottledProcessingManager {
+    private let queue = Queue()
+    
+    private let delay: Double
+    
+    var process: ((Set<MessageId>) -> Void)?
+    
+    private var timer: SwiftSignalKit.Timer?
+    private var processedList: [MessageId] = []
+    private var processed = Set<MessageId>()
+    private var buffer = Set<MessageId>()
+    
+    init(delay: Double = 1.0) {
+        self.delay = delay
+    }
+    
+    func setProcess(process: @escaping (Set<MessageId>) -> Void) {
+        self.queue.async {
+            self.process = process
+        }
+    }
+    
+    func add(_ messageIds: [MessageId]) {
+        self.queue.async {
+            for id in messageIds {
+                if !self.processed.contains(id) {
+                    self.processed.insert(id)
+                    self.processedList.append(id)
+                    self.buffer.insert(id)
+                }
+            }
+            
+            if self.processedList.count > 1000 {
+                for i in 0 ..< 200 {
+                    self.processed.remove(self.processedList[i])
+                }
+                self.processedList.removeSubrange(0 ..< 200)
+            }
+            
+            if self.timer == nil {
+                var completionImpl: (() -> Void)?
+                let timer = SwiftSignalKit.Timer(timeout: self.delay, repeat: false, completion: {
+                    completionImpl?()
+                }, queue: self.queue)
+                completionImpl = { [weak self, weak timer] in
+                    if let strongSelf = self {
+                        if let timer = timer, strongSelf.timer === timer {
+                            strongSelf.timer = nil
+                        }
+                        let buffer = strongSelf.buffer
+                        strongSelf.buffer.removeAll()
+                        strongSelf.process?(buffer)
+                    }
+                }
+                self.timer = timer
+                timer.start()
+            }
+        }
+    }
+}
+
+
+final class ChatMessageVisibleThrottledProcessingManager {
+    private let queue = Queue()
+    
+    private let delay: Double
+    
+    private var currentIds = Set<MessageId>()
+    
+    var process: ((Set<MessageId>) -> Void)?
+    
+    private var timer: SwiftSignalKit.Timer?
+    
+    init(delay: Double = 1.0) {
+        self.delay = delay
+    }
+    
+    func setProcess(process: @escaping (Set<MessageId>) -> Void) {
+        self.queue.async {
+            self.process = process
+        }
+    }
+    
+    func update(_ ids: Set<MessageId>) {
+        self.queue.async {
+            if self.currentIds != ids {
+                self.currentIds = ids
+                if self.timer == nil {
+                    var completionImpl: (() -> Void)?
+                    let timer = SwiftSignalKit.Timer(timeout: self.delay, repeat: false, completion: {
+                        completionImpl?()
+                    }, queue: self.queue)
+                    completionImpl = { [weak self, weak timer] in
+                        if let strongSelf = self {
+                            if let timer = timer, strongSelf.timer === timer {
+                                strongSelf.timer = nil
+                            }
+                            strongSelf.process?(strongSelf.currentIds)
+                        }
+                    }
+                    self.timer = timer
+                    timer.start()
+                }
+            }
+        }
+    }
+}