83 lines
2.0 KiB
Swift
83 lines
2.0 KiB
Swift
|
//
|
||
|
// FetchAllUnreadCountsOperation.swift
|
||
|
// ArticlesDatabase
|
||
|
//
|
||
|
// Created by Brent Simmons on 1/26/20.
|
||
|
// Copyright © 2020 Ranchero Software. All rights reserved.
|
||
|
//
|
||
|
|
||
|
import Foundation
|
||
|
import RSCore
|
||
|
import RSDatabase
|
||
|
|
||
|
final class FetchAllUnreadCountsOperation: MainThreadOperation {
|
||
|
|
||
|
public var unreadCountDictionary: UnreadCountDictionary?
|
||
|
|
||
|
// MainThreadOperation
|
||
|
var isCanceled = false
|
||
|
var id: Int?
|
||
|
weak var operationDelegate: MainThreadOperationDelegate?
|
||
|
var name: String? = "FetchAllUnreadCountsOperation"
|
||
|
var completionBlock: MainThreadOperation.MainThreadOperationCompletionBlock?
|
||
|
|
||
|
private let queue: DatabaseQueue
|
||
|
private let cutoffDate: Date
|
||
|
|
||
|
init(databaseQueue: DatabaseQueue, cutoffDate: Date) {
|
||
|
self.queue = databaseQueue
|
||
|
self.cutoffDate = cutoffDate
|
||
|
}
|
||
|
|
||
|
func run() {
|
||
|
queue.runInDatabase { databaseResult in
|
||
|
if self.isCanceled {
|
||
|
self.informOperationDelegateOfCompletion()
|
||
|
return
|
||
|
}
|
||
|
|
||
|
switch databaseResult {
|
||
|
case .success(let database):
|
||
|
self.fetchUnreadCounts(database)
|
||
|
case .failure:
|
||
|
self.informOperationDelegateOfCompletion()
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private extension FetchAllUnreadCountsOperation {
|
||
|
|
||
|
func informOperationDelegateOfCompletion() {
|
||
|
DispatchQueue.main.async {
|
||
|
if !self.isCanceled {
|
||
|
self.operationDelegate?.operationDidComplete(self)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func fetchUnreadCounts(_ database: FMDatabase) {
|
||
|
let sql = "select distinct feedID, count(*) from articles natural join statuses where read=0 and userDeleted=0 and (starred=1 or dateArrived>?) group by feedID;"
|
||
|
|
||
|
guard let resultSet = database.executeQuery(sql, withArgumentsIn: [cutoffDate]) else {
|
||
|
informOperationDelegateOfCompletion()
|
||
|
return
|
||
|
}
|
||
|
|
||
|
var d = UnreadCountDictionary()
|
||
|
while resultSet.next() {
|
||
|
if isCanceled {
|
||
|
informOperationDelegateOfCompletion()
|
||
|
return
|
||
|
}
|
||
|
let unreadCount = resultSet.long(forColumnIndex: 1)
|
||
|
if let webFeedID = resultSet.string(forColumnIndex: 0) {
|
||
|
d[webFeedID] = unreadCount
|
||
|
}
|
||
|
}
|
||
|
|
||
|
unreadCountDictionary = d
|
||
|
informOperationDelegateOfCompletion()
|
||
|
}
|
||
|
}
|