CarbStore.swift 60 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496
  1. //
  2. // CarbStore.swift
  3. // CarbKit
  4. //
  5. // Created by Nathan Racklyeft on 1/3/16.
  6. // Copyright © 2016 Nathan Racklyeft. All rights reserved.
  7. //
  8. import Foundation
  9. import CoreData
  10. import HealthKit
  11. import os.log
  12. public enum CarbStoreResult<T> {
  13. case success(T)
  14. case failure(CarbStore.CarbStoreError)
  15. }
  16. public enum CarbAbsorptionModel {
  17. case linear
  18. case nonlinear
  19. case adaptiveRateNonlinear
  20. }
  21. public protocol CarbStoreDelegate: AnyObject {
  22. /**
  23. Informs the delegate that the carb store has updated carb data.
  24. - Parameter carbStore: The carb store that has updated carb data.
  25. */
  26. func carbStoreHasUpdatedCarbData(_ carbStore: CarbStore)
  27. /**
  28. Informs the delegate that an internal error occurred.
  29. - parameter carbStore: The carb store
  30. - parameter error: The error describing the issue
  31. */
  32. func carbStore(_ carbStore: CarbStore, didError error: CarbStore.CarbStoreError)
  33. }
  34. /**
  35. Manages storage, retrieval, and calculation of carbohydrate data.
  36. There are two tiers of storage:
  37. * Persistant cache, stored in Core Data, used to ensure access if the app is suspended and re-launched while the Health database
  38. is protected and to provide data for upload to remote data services. Backfilled from HealthKit data up to observation interval.
  39. ```
  40. 0 [max(cacheLength, observationInterval, defaultAbsorptionTimes.slow * 2)]
  41. |––––––––––––|
  42. ```
  43. * HealthKit data, managed by the current application and persisted indefinitely
  44. ```
  45. 0
  46. |––––––––––––––––––--->
  47. ```
  48. */
  49. public final class CarbStore: HealthKitSampleStore {
  50. /// Notification posted when carb entries were changed, either via add/replace/delete methods or from HealthKit
  51. public static let carbEntriesDidChange = NSNotification.Name(rawValue: "com.loopkit.CarbStore.carbEntriesDidChange")
  52. public typealias DefaultAbsorptionTimes = (fast: TimeInterval, medium: TimeInterval, slow: TimeInterval)
  53. public enum CarbStoreError: Error {
  54. // The store isn't correctly configured for the requested operation
  55. case notConfigured
  56. // The health store request returned an error
  57. case healthStoreError(Error)
  58. // The core data request returned an error
  59. case coreDataError(Error)
  60. // The requested sample can't be modified by this store
  61. case unauthorized
  62. // No data was found to match the specified request
  63. case noData
  64. init?(error: PersistenceController.PersistenceControllerError?) {
  65. guard let error = error, case .coreDataError(let coreDataError) = error else {
  66. return nil
  67. }
  68. self = .coreDataError(coreDataError as Error)
  69. }
  70. }
  71. private let carbType = HKQuantityType.quantityType(forIdentifier: HKQuantityTypeIdentifier.dietaryCarbohydrates)!
  72. /// The preferred unit. iOS currently only supports grams for dietary carbohydrates.
  73. public override var preferredUnit: HKUnit! {
  74. return super.preferredUnit
  75. }
  76. /// A history of recently applied schedule overrides.
  77. private let overrideHistory: TemporaryScheduleOverrideHistory?
  78. /// Carbohydrate-to-insulin ratio
  79. public var carbRatioSchedule: CarbRatioSchedule? {
  80. get {
  81. return lockedCarbRatioSchedule.value
  82. }
  83. set {
  84. lockedCarbRatioSchedule.value = newValue
  85. }
  86. }
  87. private let lockedCarbRatioSchedule: Locked<CarbRatioSchedule?>
  88. /// The carb ratio schedule, applying recent overrides relative to the current moment in time.
  89. public var carbRatioScheduleApplyingOverrideHistory: CarbRatioSchedule? {
  90. if let carbRatioSchedule = carbRatioSchedule {
  91. return overrideHistory?.resolvingRecentCarbRatioSchedule(carbRatioSchedule)
  92. } else {
  93. return nil
  94. }
  95. }
  96. /// A trio of default carbohydrate absorption times. Defaults to 2, 3, and 4 hours.
  97. public let defaultAbsorptionTimes: DefaultAbsorptionTimes
  98. /// Insulin-to-glucose sensitivity
  99. public var insulinSensitivitySchedule: InsulinSensitivitySchedule? {
  100. get {
  101. return lockedInsulinSensitivitySchedule.value
  102. }
  103. set {
  104. lockedInsulinSensitivitySchedule.value = newValue
  105. }
  106. }
  107. private let lockedInsulinSensitivitySchedule: Locked<InsulinSensitivitySchedule?>
  108. /// The insulin sensitivity schedule, applying recent overrides relative to the current moment in time.
  109. public var insulinSensitivityScheduleApplyingOverrideHistory: InsulinSensitivitySchedule? {
  110. if let insulinSensitivitySchedule = insulinSensitivitySchedule {
  111. return overrideHistory?.resolvingRecentInsulinSensitivitySchedule(insulinSensitivitySchedule)
  112. } else {
  113. return nil
  114. }
  115. }
  116. /// The computed carbohydrate sensitivity schedule based on the insulin sensitivity and carb ratio schedules.
  117. public var carbSensitivitySchedule: CarbSensitivitySchedule? {
  118. guard let insulinSensitivitySchedule = insulinSensitivitySchedule, let carbRatioSchedule = carbRatioSchedule else {
  119. return nil
  120. }
  121. return .carbSensitivitySchedule(insulinSensitivitySchedule: insulinSensitivitySchedule, carbRatioSchedule: carbRatioSchedule)
  122. }
  123. /// The expected delay in the appearance of glucose effects, accounting for both digestion and sensor lag
  124. public let delay: TimeInterval
  125. /// The interval between effect values to use for the calculated timelines.
  126. public let delta: TimeInterval
  127. /// The factor by which the entered absorption time can be extended to accomodate slower-than-expected absorption
  128. public let absorptionTimeOverrun: Double
  129. /// Carb absorption model
  130. public let carbAbsorptionModel: CarbAbsorptionModel
  131. /// The interval of carb data to keep in cache
  132. public let cacheLength: TimeInterval
  133. /// The interval to observe HealthKit data to populate the cache
  134. public let observationInterval: TimeInterval
  135. private let storeEntriesToHealthKit: Bool
  136. private let cacheStore: PersistenceController
  137. /// The sync version used for new samples written to HealthKit
  138. /// Choose a lower or higher sync version if the same sample might be written twice (e.g. from an extension and from an app) for deterministic conflict resolution
  139. public let syncVersion: Int
  140. public weak var delegate: CarbStoreDelegate?
  141. private let queue = DispatchQueue(label: "com.loopkit.CarbStore.queue", qos: .utility)
  142. private let log = OSLog(category: "CarbStore")
  143. static let healthKitQueryAnchorMetadataKey = "com.loopkit.CarbStore.hkQueryAnchor"
  144. var settings = CarbModelSettings(absorptionModel: PiecewiseLinearAbsorption(), initialAbsorptionTimeOverrun: 1.5, adaptiveAbsorptionRateEnabled: false)
  145. private let provenanceIdentifier: String
  146. /**
  147. Initializes a new instance of the store.
  148. - returns: A new instance of the store
  149. */
  150. public init(
  151. healthStore: HKHealthStore,
  152. observeHealthKitSamplesFromOtherApps: Bool = true,
  153. storeEntriesToHealthKit: Bool = true,
  154. cacheStore: PersistenceController,
  155. cacheLength: TimeInterval,
  156. defaultAbsorptionTimes: DefaultAbsorptionTimes,
  157. observationInterval: TimeInterval,
  158. carbRatioSchedule: CarbRatioSchedule? = nil,
  159. insulinSensitivitySchedule: InsulinSensitivitySchedule? = nil,
  160. overrideHistory: TemporaryScheduleOverrideHistory? = nil,
  161. syncVersion: Int = 1,
  162. absorptionTimeOverrun: Double = 1.5,
  163. calculationDelta: TimeInterval = 5 /* minutes */ * 60,
  164. effectDelay: TimeInterval = 10 /* minutes */ * 60,
  165. carbAbsorptionModel: CarbAbsorptionModel = .nonlinear,
  166. provenanceIdentifier: String
  167. ) {
  168. self.storeEntriesToHealthKit = storeEntriesToHealthKit
  169. self.cacheStore = cacheStore
  170. self.defaultAbsorptionTimes = defaultAbsorptionTimes
  171. self.lockedCarbRatioSchedule = Locked(carbRatioSchedule)
  172. self.lockedInsulinSensitivitySchedule = Locked(insulinSensitivitySchedule)
  173. self.overrideHistory = overrideHistory
  174. self.syncVersion = syncVersion
  175. self.absorptionTimeOverrun = absorptionTimeOverrun
  176. self.delta = calculationDelta
  177. self.delay = effectDelay
  178. self.cacheLength = cacheLength
  179. self.observationInterval = observationInterval
  180. self.carbAbsorptionModel = carbAbsorptionModel
  181. self.provenanceIdentifier = provenanceIdentifier
  182. let observationEnabled = observationInterval > 0
  183. super.init(healthStore: healthStore,
  184. observeHealthKitSamplesFromCurrentApp: true,
  185. observeHealthKitSamplesFromOtherApps: observeHealthKitSamplesFromOtherApps,
  186. type: carbType,
  187. observationStart: Date(timeIntervalSinceNow: -self.observationInterval),
  188. observationEnabled: observationEnabled)
  189. // Carb model settings based on the selected absorption model
  190. switch self.carbAbsorptionModel {
  191. case .linear:
  192. self.settings = CarbModelSettings(absorptionModel: LinearAbsorption(), initialAbsorptionTimeOverrun: absorptionTimeOverrun, adaptiveAbsorptionRateEnabled: false)
  193. case .nonlinear:
  194. self.settings = CarbModelSettings(absorptionModel: PiecewiseLinearAbsorption(), initialAbsorptionTimeOverrun: absorptionTimeOverrun, adaptiveAbsorptionRateEnabled: false)
  195. case .adaptiveRateNonlinear:
  196. self.settings = CarbModelSettings(absorptionModel: PiecewiseLinearAbsorption(), initialAbsorptionTimeOverrun: 1.0, adaptiveAbsorptionRateEnabled: true, adaptiveRateStandbyIntervalFraction: 0.2)
  197. }
  198. let semaphore = DispatchSemaphore(value: 0)
  199. cacheStore.onReady { (error) in
  200. guard error == nil else {
  201. semaphore.signal()
  202. return
  203. }
  204. cacheStore.fetchAnchor(key: CarbStore.healthKitQueryAnchorMetadataKey) { (anchor) in
  205. self.queue.async {
  206. self.queryAnchor = anchor
  207. if !self.authorizationRequired {
  208. self.createQuery()
  209. }
  210. self.migrateLegacyCarbEntryKeys()
  211. semaphore.signal()
  212. }
  213. }
  214. }
  215. semaphore.wait()
  216. }
  217. // Migrate modifiedCarbEntries and deletedCarbEntryIDs
  218. private func migrateLegacyCarbEntryKeys() {
  219. cacheStore.managedObjectContext.performAndWait {
  220. var changed = false
  221. for entry in UserDefaults.standard.modifiedCarbEntries ?? [] {
  222. let object = CachedCarbObject(context: self.cacheStore.managedObjectContext)
  223. object.create(from: entry)
  224. changed = true
  225. }
  226. // Note: We no longer migrate UserDefaults.standard.deletedCarbEntryIds since we don't have a startDate (only
  227. // external ID) and CachedCarbObject requires a starDate. This only prevents a deleted carb entry that was previously
  228. // uploaded to Nightscout, but not yet deleted from Nightscout, from being deleted in Nightscout.
  229. if changed {
  230. self.cacheStore.save()
  231. }
  232. }
  233. UserDefaults.standard.purgeLegacyCarbEntryKeys()
  234. }
  235. // MARK: - HealthKitSampleStore
  236. override func queryAnchorDidChange() {
  237. cacheStore.storeAnchor(queryAnchor, key: CarbStore.healthKitQueryAnchorMetadataKey)
  238. }
  239. override func processResults(from query: HKAnchoredObjectQuery, added: [HKSample], deleted: [HKDeletedObject], anchor: HKQueryAnchor, completion: @escaping (Bool) -> Void) {
  240. queue.async {
  241. guard anchor != self.queryAnchor else {
  242. self.log.default("Skipping processing results from anchored object query, as anchor was already processed")
  243. completion(true)
  244. return
  245. }
  246. var changed = false
  247. var error: CarbStoreError?
  248. self.cacheStore.managedObjectContext.performAndWait {
  249. do {
  250. let date = Date()
  251. // Add new samples
  252. if let samples = added as? [HKQuantitySample] {
  253. for sample in samples {
  254. if try self.addCarbEntry(for: sample, on: date) {
  255. self.log.debug("Saved sample %@ into cache from HKAnchoredObjectQuery", sample.uuid.uuidString)
  256. changed = true
  257. } else {
  258. self.log.default("Sample %@ from HKAnchoredObjectQuery already present in cache", sample.uuid.uuidString)
  259. }
  260. }
  261. }
  262. // Delete deleted samples
  263. for sample in deleted {
  264. if try self.deleteCarbEntry(for: sample.uuid, on: date) {
  265. self.log.debug("Deleted sample %@ from cache from HKAnchoredObjectQuery", sample.uuid.uuidString)
  266. changed = true
  267. }
  268. }
  269. guard changed else {
  270. return
  271. }
  272. error = CarbStoreError(error: self.cacheStore.save())
  273. } catch let coreDataError {
  274. error = .coreDataError(coreDataError)
  275. }
  276. }
  277. if let error = error {
  278. self.delegate?.carbStore(self, didError: error)
  279. completion(false)
  280. return
  281. }
  282. if !changed {
  283. completion(true)
  284. return
  285. }
  286. self.handleUpdatedCarbData()
  287. completion(true)
  288. }
  289. }
  290. }
  291. // MARK: - Fetching
  292. extension CarbStore {
  293. /// Retrieves carb entries within the specified date range
  294. ///
  295. /// - Parameters:
  296. /// - start: The earliest date of values to retrieve
  297. /// - end: The latest date of values to retrieve, if provided
  298. /// - completion: A closure called once the values have been retrieved
  299. /// - result: An array of carb entries, in chronological order by startDate, or error
  300. public func getCarbEntries(start: Date? = nil, end: Date? = nil, completion: @escaping (_ result: CarbStoreResult<[StoredCarbEntry]>) -> Void) {
  301. queue.async {
  302. completion(self.getCarbEntries(start: start, end: end))
  303. }
  304. }
  305. /// Retrieves carb entries within the specified date range
  306. ///
  307. /// - Parameters:
  308. /// - start: The earliest date of values to retrieve
  309. /// - end: The latest date of values to retrieve, if provided
  310. /// - Returns: An array of carb entries, in chronological order by startDate, or error
  311. private func getCarbEntries(start: Date? = nil, end: Date? = nil) -> CarbStoreResult<[StoredCarbEntry]> {
  312. dispatchPrecondition(condition: .onQueue(queue))
  313. var entries: [StoredCarbEntry] = []
  314. var error: CarbStoreError?
  315. cacheStore.managedObjectContext.performAndWait {
  316. do {
  317. entries = try self.getActiveCachedCarbObjects(start: start, end: end).map { StoredCarbEntry(managedObject: $0) }
  318. } catch let coreDataError {
  319. error = .coreDataError(coreDataError)
  320. }
  321. }
  322. if let error = error {
  323. return .failure(error)
  324. }
  325. return .success(entries)
  326. }
  327. /// Retrieves active (not superceded, non-delete operation) cached carb objects within the specified date range
  328. ///
  329. /// - Parameters:
  330. /// - start: The earliest date of values to retrieve
  331. /// - end: The latest date of values to retrieve, if provided
  332. /// - Returns: An array of cached carb objects
  333. private func getActiveCachedCarbObjects(start: Date? = nil, end: Date? = nil) throws -> [CachedCarbObject] {
  334. dispatchPrecondition(condition: .onQueue(queue))
  335. var predicates = [NSPredicate(format: "operation != %d", Operation.delete.rawValue),
  336. NSPredicate(format: "supercededDate == NIL")]
  337. if let start = start {
  338. predicates.append(NSPredicate(format: "startDate >= %@", start as NSDate))
  339. }
  340. if let end = end {
  341. predicates.append(NSPredicate(format: "startDate < %@", end as NSDate))
  342. }
  343. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  344. request.predicate = NSCompoundPredicate(andPredicateWithSubpredicates: predicates)
  345. request.sortDescriptors = [NSSortDescriptor(key: "startDate", ascending: true)]
  346. return try self.cacheStore.managedObjectContext.fetch(request)
  347. }
  348. /// Retrieves carb entries from HealthKit within the specified date range and interprets their
  349. /// absorption status based on the provided glucose effect
  350. ///
  351. /// - Parameters:
  352. /// - start: The earliest date of values to retrieve
  353. /// - end: The latest date of values to retrieve, if provided
  354. /// - effectVelocities: A timeline of glucose effect velocities, ordered by start date
  355. /// - completion: A closure calld once the values have been retrieved
  356. /// - result: An array of carb entries, in chronological order by startDate
  357. public func getCarbStatus(
  358. start: Date,
  359. end: Date? = nil,
  360. effectVelocities: [GlucoseEffectVelocity]? = nil,
  361. completion: @escaping (_ result: CarbStoreResult<[CarbStatus<StoredCarbEntry>]>) -> Void
  362. ) {
  363. getCarbEntries(start: start, end: end) { (result) in
  364. switch result {
  365. case .success(let entries):
  366. let status = entries.map(
  367. to: effectVelocities ?? [],
  368. carbRatio: self.carbRatioScheduleApplyingOverrideHistory,
  369. insulinSensitivity: self.insulinSensitivityScheduleApplyingOverrideHistory,
  370. absorptionTimeOverrun: self.absorptionTimeOverrun,
  371. defaultAbsorptionTime: self.defaultAbsorptionTimes.medium,
  372. delay: self.delay,
  373. initialAbsorptionTimeOverrun: self.settings.initialAbsorptionTimeOverrun,
  374. absorptionModel: self.settings.absorptionModel,
  375. adaptiveAbsorptionRateEnabled: self.settings.adaptiveAbsorptionRateEnabled,
  376. adaptiveRateStandbyIntervalFraction: self.settings.adaptiveRateStandbyIntervalFraction
  377. )
  378. completion(.success(status))
  379. case .failure(let error):
  380. completion(.failure(error))
  381. }
  382. }
  383. }
  384. }
  385. // MARK: - Modification
  386. extension CarbStore {
  387. public func addCarbEntry(_ entry: NewCarbEntry, completion: @escaping (_ result: CarbStoreResult<StoredCarbEntry>) -> Void) {
  388. queue.async {
  389. var storedEntry: StoredCarbEntry?
  390. var error: CarbStoreError?
  391. self.cacheStore.managedObjectContext.performAndWait {
  392. do {
  393. let syncIdentifier = try self.cacheStore.managedObjectContext.generateUniqueSyncIdentifier()
  394. let newObject = CachedCarbObject(context: self.cacheStore.managedObjectContext)
  395. newObject.create(from: entry,
  396. provenanceIdentifier: self.provenanceIdentifier,
  397. syncIdentifier: syncIdentifier,
  398. syncVersion: self.syncVersion)
  399. if let saveError = CarbStoreError(error: self.cacheStore.save()) {
  400. error = saveError
  401. return
  402. }
  403. self.saveEntryToHealthKit(newObject)
  404. storedEntry = StoredCarbEntry(managedObject: newObject)
  405. } catch let coreDataError {
  406. error = .coreDataError(coreDataError)
  407. }
  408. }
  409. if let error = error {
  410. completion(.failure(error))
  411. return
  412. }
  413. completion(.success(storedEntry!))
  414. self.handleUpdatedCarbData()
  415. }
  416. }
  417. public func replaceCarbEntry(_ oldEntry: StoredCarbEntry, withEntry newEntry: NewCarbEntry, completion: @escaping (_ result: CarbStoreResult<StoredCarbEntry>) -> Void) {
  418. guard oldEntry.createdByCurrentApp else {
  419. completion(.failure(.unauthorized))
  420. return
  421. }
  422. queue.async {
  423. var storedEntry: StoredCarbEntry?
  424. var error: CarbStoreError?
  425. self.cacheStore.managedObjectContext.performAndWait {
  426. do {
  427. guard let oldObject = try self.cacheStore.managedObjectContext.cachedCarbObjectFromStoredCarbEntry(oldEntry) else {
  428. error = .noData
  429. return
  430. }
  431. // Use same date for superceding old object and adding new object
  432. let date = Date()
  433. oldObject.supercededDate = date
  434. let newObject = CachedCarbObject(context: self.cacheStore.managedObjectContext)
  435. newObject.update(from: newEntry, replacing: oldObject, on: date)
  436. if let saveError = CarbStoreError(error: self.cacheStore.save()) {
  437. error = saveError
  438. return
  439. }
  440. self.saveEntryToHealthKit(newObject)
  441. storedEntry = StoredCarbEntry(managedObject: newObject)
  442. } catch let coreDataError {
  443. error = .coreDataError(coreDataError)
  444. }
  445. }
  446. if let error = error {
  447. completion(.failure(error))
  448. return
  449. }
  450. completion(.success(storedEntry!))
  451. self.handleUpdatedCarbData()
  452. }
  453. }
  454. private func saveEntryToHealthKit(_ object: CachedCarbObject) {
  455. dispatchPrecondition(condition: .onQueue(queue))
  456. guard storeEntriesToHealthKit else {
  457. return
  458. }
  459. let quantitySample = object.quantitySample
  460. var error: Error?
  461. // Save object to HealthKit, log any errors, but do not fail
  462. let dispatchGroup = DispatchGroup()
  463. dispatchGroup.enter()
  464. self.healthStore.save(quantitySample) { (_, healthKitError) in
  465. error = healthKitError
  466. dispatchGroup.leave()
  467. }
  468. dispatchGroup.wait()
  469. if let error = error {
  470. self.log.error("Error saving HealthKit object: %@", String(describing: error))
  471. return
  472. }
  473. // Update Core Data with the change, log any errors, but do not fail
  474. object.uuid = quantitySample.uuid
  475. if let error = self.cacheStore.save() {
  476. self.log.error("Error updating CachedCarbObject after saving HealthKit object: %@", String(describing: error))
  477. object.uuid = nil
  478. }
  479. }
  480. public func deleteCarbEntry(_ oldEntry: StoredCarbEntry, completion: @escaping (_ result: CarbStoreResult<Bool>) -> Void) {
  481. guard oldEntry.createdByCurrentApp else {
  482. completion(.failure(.unauthorized))
  483. return
  484. }
  485. queue.async {
  486. var error: CarbStoreError?
  487. self.cacheStore.managedObjectContext.performAndWait {
  488. do {
  489. guard let oldObject = try self.cacheStore.managedObjectContext.cachedCarbObjectFromStoredCarbEntry(oldEntry) else {
  490. error = .noData
  491. return
  492. }
  493. // Use same date for superceding old object and adding new object; also used for userDeletedDate
  494. let date = Date()
  495. oldObject.supercededDate = date
  496. let newObject = CachedCarbObject(context: self.cacheStore.managedObjectContext)
  497. newObject.delete(from: oldObject, on: date)
  498. if let saveError = CarbStoreError(error: self.cacheStore.save()) {
  499. error = saveError
  500. return
  501. }
  502. self.deleteObjectFromHealthKit(newObject)
  503. } catch let coreDataError {
  504. error = .coreDataError(coreDataError)
  505. }
  506. }
  507. if let error = error {
  508. completion(.failure(error))
  509. return
  510. }
  511. completion(.success(true))
  512. self.handleUpdatedCarbData()
  513. }
  514. }
  515. private func deleteObjectFromHealthKit(_ object: CachedCarbObject) {
  516. dispatchPrecondition(condition: .onQueue(queue))
  517. // If the object does not have a UUID, then it was never saved to HealthKit, so no need to delete
  518. guard object.uuid != nil else {
  519. return
  520. }
  521. var error: Error?
  522. // Delete object from HealthKit, log any errors, but do not fail
  523. let dispatchGroup = DispatchGroup()
  524. dispatchGroup.enter()
  525. self.healthStore.deleteObjects(of: self.carbType, predicate: HKQuery.predicateForObject(with: object.uuid!)) { (_, _, healthKitError) in
  526. error = healthKitError
  527. dispatchGroup.leave()
  528. }
  529. dispatchGroup.wait()
  530. if let error = error {
  531. self.log.error("Error deleting HealthKit object: %@", String(describing: error))
  532. return
  533. }
  534. // Update Core Data with the change, log any errors, but do not fail
  535. object.uuid = nil
  536. if let error = self.cacheStore.save() {
  537. self.log.error("Error updating CachedCarbObject after deleting HealthKit object: %@", String(describing: error))
  538. }
  539. }
  540. private func addCarbEntry(for sample: HKQuantitySample, on date: Date) throws -> Bool {
  541. dispatchPrecondition(condition: .onQueue(queue))
  542. // Are there any objects matching the UUID?
  543. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  544. request.predicate = NSPredicate(format: "uuid == %@", sample.uuid as NSUUID)
  545. request.fetchLimit = 1
  546. let count = try cacheStore.managedObjectContext.count(for: request)
  547. guard count == 0 else {
  548. return false
  549. }
  550. // Find all objects being replaced
  551. let replacedObjects = try fetchRelatedCarbObjects(for: sample)
  552. // Mark all objects as superceded, as necessary
  553. replacedObjects.filter({ $0.supercededDate == nil }).forEach({ $0.supercededDate = date })
  554. // Add an object (create or update) for this UUID
  555. let object = CachedCarbObject(context: cacheStore.managedObjectContext)
  556. if let replacedObject = replacedObjects.last {
  557. object.update(from: sample, replacing: replacedObject, on: date)
  558. } else {
  559. object.create(from: sample, on: date)
  560. }
  561. return true
  562. }
  563. private func deleteCarbEntry(for uuid: UUID, on date: Date) throws -> Bool {
  564. dispatchPrecondition(condition: .onQueue(queue))
  565. // Fetch objects matching the UUID, if none found, then nothing to delete, sorted by last seen anchor key
  566. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  567. request.predicate = NSPredicate(format: "uuid == %@", uuid as NSUUID)
  568. request.sortDescriptors = [NSSortDescriptor(key: "anchorKey", ascending: true)]
  569. let objects = try cacheStore.managedObjectContext.fetch(request)
  570. guard !objects.isEmpty else {
  571. return false
  572. }
  573. // Find all unsuperceded create/update objects, if none found, then nothing to delete
  574. let supercededObjects = objects.filter { $0.operation != .delete && $0.supercededDate == nil }
  575. guard !supercededObjects.isEmpty else {
  576. return false
  577. }
  578. // Mark as superceded
  579. supercededObjects.forEach { $0.supercededDate = date }
  580. // If we don't yet have a delete object, then add one
  581. if !objects.contains(where: { $0.operation == .delete }), let supercededObject = supercededObjects.last {
  582. let object = CachedCarbObject(context: cacheStore.managedObjectContext)
  583. object.delete(from: supercededObject, on: date)
  584. }
  585. return true
  586. }
  587. // Fetch all objects that are different versions of the specified sample, using sync identifier
  588. private func fetchRelatedCarbObjects(for sample: HKQuantitySample) throws -> [CachedCarbObject] {
  589. dispatchPrecondition(condition: .onQueue(queue))
  590. guard let syncIdentifier = sample.syncIdentifier else {
  591. return []
  592. }
  593. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  594. request.predicate = NSCompoundPredicate(andPredicateWithSubpredicates: [NSPredicate(format: "provenanceIdentifier == %@", sample.provenanceIdentifier),
  595. NSPredicate(format: "syncIdentifier == %@", syncIdentifier)])
  596. request.sortDescriptors = [NSSortDescriptor(key: "anchorKey", ascending: true)]
  597. return try cacheStore.managedObjectContext.fetch(request)
  598. }
  599. }
  600. // MARK: - Watch Synchronization
  601. extension CarbStore {
  602. /// Get carb objects in main app to deliver to Watch extension
  603. public func getSyncCarbObjects(start: Date? = nil, end: Date? = nil, completion: @escaping (_ result: CarbStoreResult<[SyncCarbObject]>) -> Void) {
  604. queue.async {
  605. var objects: [SyncCarbObject] = []
  606. var error: CarbStoreError?
  607. self.cacheStore.managedObjectContext.performAndWait {
  608. do {
  609. objects = try self.getActiveCachedCarbObjects(start: start, end: end).map { SyncCarbObject(managedObject: $0) }
  610. } catch let coreDataError {
  611. error = .coreDataError(coreDataError)
  612. }
  613. }
  614. if let error = error {
  615. completion(.failure(error))
  616. return
  617. }
  618. completion(.success(objects))
  619. }
  620. }
  621. /// Store carb objects in Watch extension
  622. public func setSyncCarbObjects(_ objects: [SyncCarbObject], completion: @escaping (CarbStoreError?) -> Void) {
  623. queue.async {
  624. if let error = self.purgeCachedCarbObjectsUnconditionally() {
  625. completion(error)
  626. return
  627. }
  628. var error: CarbStoreError?
  629. self.cacheStore.managedObjectContext.performAndWait {
  630. guard !objects.isEmpty else {
  631. return
  632. }
  633. objects.forEach {
  634. let object = CachedCarbObject(context: self.cacheStore.managedObjectContext)
  635. object.update(from: $0)
  636. }
  637. error = CarbStoreError(error: self.cacheStore.save())
  638. }
  639. completion(error)
  640. self.handleUpdatedCarbData()
  641. }
  642. }
  643. }
  644. // MARK: - Cache management
  645. extension CarbStore {
  646. public var earliestCacheDate: Date {
  647. return Date(timeIntervalSinceNow: -cacheLength)
  648. }
  649. private func purgeExpiredCachedCarbObjects() {
  650. purgeCachedCarbObjects(before: earliestCacheDate)
  651. }
  652. @discardableResult
  653. private func purgeCachedCarbObjects(before date: Date) -> CarbStoreError? {
  654. dispatchPrecondition(condition: .onQueue(queue))
  655. var error: CarbStoreError?
  656. cacheStore.managedObjectContext.performAndWait {
  657. do {
  658. // Fetch all candidate objects for purge
  659. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  660. request.predicate = NSPredicate(format: "startDate < %@", date as NSDate)
  661. let objects = try self.cacheStore.managedObjectContext.fetch(request)
  662. // Objects can only be purged if all related objects can be purged
  663. let purgedObjects = try objects.filter { try self.areAllRelatedObjectsPurgable(to: $0, before: date) }
  664. guard !purgedObjects.isEmpty else {
  665. return
  666. }
  667. // Actually purge
  668. purgedObjects.forEach { self.cacheStore.managedObjectContext.delete($0) }
  669. if let saveError = CarbStoreError(error: self.cacheStore.save()) {
  670. error = saveError
  671. return
  672. }
  673. self.log.info("Purged %d CachedCarbObjects", purgedObjects.count)
  674. } catch let coreDataError {
  675. error = .coreDataError(coreDataError)
  676. }
  677. }
  678. if let error = error {
  679. self.log.error("Unable to purge CachedCarbObjects: %{public}@", String(describing: error))
  680. return error
  681. }
  682. return nil
  683. }
  684. public func purgeCachedCarbObjectsUnconditionally(before date: Date, completion: @escaping (CarbStoreError?) -> Void) {
  685. queue.async {
  686. if let error = self.purgeCachedCarbObjectsUnconditionally(before: date) {
  687. completion(error)
  688. return
  689. }
  690. self.handleUpdatedCarbData()
  691. completion(nil)
  692. }
  693. }
  694. private func purgeCachedCarbObjectsUnconditionally(before date: Date? = nil) -> CarbStoreError? {
  695. dispatchPrecondition(condition: .onQueue(queue))
  696. var error: CarbStoreError?
  697. cacheStore.managedObjectContext.performAndWait {
  698. do {
  699. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  700. if let date = date {
  701. request.predicate = NSPredicate(format: "startDate < %@", date as NSDate)
  702. }
  703. let count = try self.cacheStore.managedObjectContext.deleteObjects(matching: request)
  704. self.log.info("Purged all %d CachedCarbObjects", count)
  705. } catch let coreDataError {
  706. self.log.error("Unable to purge all CachedCarbObjects: %{public}@", String(describing: coreDataError))
  707. error = .coreDataError(coreDataError)
  708. }
  709. }
  710. return error
  711. }
  712. private func handleUpdatedCarbData() {
  713. dispatchPrecondition(condition: .onQueue(queue))
  714. purgeExpiredCachedCarbObjects()
  715. NotificationCenter.default.post(name: CarbStore.carbEntriesDidChange, object: self)
  716. delegate?.carbStoreHasUpdatedCarbData(self)
  717. }
  718. private func areAllRelatedObjectsPurgable(to object: CachedCarbObject, before date: Date) throws -> Bool {
  719. dispatchPrecondition(condition: .onQueue(queue))
  720. // If no sync identifier, then there are no related objects
  721. guard let syncIdentifier = object.syncIdentifier else {
  722. return true
  723. }
  724. // Count any that are NOT purgable
  725. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  726. request.predicate = NSCompoundPredicate(andPredicateWithSubpredicates: [NSPredicate(format: "provenanceIdentifier == %@", object.provenanceIdentifier),
  727. NSPredicate(format: "syncIdentifier == %@", syncIdentifier),
  728. NSPredicate(format: "startDate >= %@", date as NSDate)])
  729. request.fetchLimit = 1
  730. return try cacheStore.managedObjectContext.count(for: request) == 0
  731. }
  732. }
  733. // MARK: - Math
  734. extension CarbStore {
  735. /// The longest expected absorption time interval for carbohydrates. Defaults to 8 hours.
  736. public var maximumAbsorptionTimeInterval: TimeInterval {
  737. return defaultAbsorptionTimes.slow * 2
  738. }
  739. /// Retrieves the single carbs on-board value occuring just prior or equal to the specified date
  740. ///
  741. /// This operation is performed asynchronously and the completion will be executed on an arbitrary background queue.
  742. ///
  743. /// - Parameters:
  744. /// - date: The date of the value to retrieve
  745. /// - effectVelocities: A timeline of glucose effect velocities, ordered by start date
  746. /// - completion: A closure called once the value has been retrieved
  747. /// - result: The carbs on-board value
  748. public func carbsOnBoard(at date: Date, effectVelocities: [GlucoseEffectVelocity]? = nil, completion: @escaping (_ result: CarbStoreResult<CarbValue>) -> Void) {
  749. getCarbsOnBoardValues(start: date.addingTimeInterval(-delta), end: date, effectVelocities: effectVelocities) { (result) in
  750. switch result {
  751. case .failure(let error):
  752. completion(.failure(error))
  753. case .success(let values):
  754. guard let value = values.closestPrior(to: date) else {
  755. // If we have no cob values in the store, and did not encounter an error, return 0
  756. completion(.success(CarbValue(startDate: date, quantity: HKQuantity(unit: .gram(), doubleValue: 0))))
  757. return
  758. }
  759. completion(.success(value))
  760. }
  761. }
  762. }
  763. /// Retrieves a timeline of unabsorbed carbohydrates
  764. ///
  765. /// This operation is performed asynchronously and the completion will be executed on an arbitrary background queue.
  766. ///
  767. /// - Parameters:
  768. /// - start: The earliest date of values to retrieve
  769. /// - end: The latest date of values to retrieve, if provided
  770. /// - effectVelocities: A timeline of glucose effect velocities, ordered by start date
  771. /// - completion: A closure called once the values have been retrieved
  772. /// - values: A timeline of carb values, in chronological order
  773. public func getCarbsOnBoardValues(start: Date, end: Date? = nil, effectVelocities: [GlucoseEffectVelocity]? = nil, completion: @escaping (_ result: CarbStoreResult<[CarbValue]>) -> Void) {
  774. // To know COB at the requested start date, we need to fetch samples that might still be absorbing
  775. let foodStart = start.addingTimeInterval(-maximumAbsorptionTimeInterval)
  776. getCarbEntries(start: foodStart, end: end) { (result) in
  777. switch result {
  778. case .failure(let error):
  779. completion(.failure(error))
  780. case .success(let entries):
  781. let carbsOnBoard = self.carbsOnBoard(from: entries, startingAt: start, endingAt: end, effectVelocities: effectVelocities)
  782. completion(.success(carbsOnBoard))
  783. }
  784. }
  785. }
  786. /// Computes a timeline of unabsorbed carbohydrates
  787. /// - Parameters:
  788. /// - start: The earliest date of values to retrieve
  789. /// - end: The latest date of values to retrieve, if provided
  790. /// - effectVelocities: A timeline of glucose effect velocities, ordered by start date
  791. /// - Returns: A timeline of unabsorbed carbohydrates
  792. public func carbsOnBoard<Sample: CarbEntry>(
  793. from samples: [Sample],
  794. startingAt start: Date,
  795. endingAt end: Date? = nil,
  796. effectVelocities: [GlucoseEffectVelocity]? = nil
  797. ) -> [CarbValue] {
  798. if let velocities = effectVelocities,
  799. let carbRatioSchedule = carbRatioScheduleApplyingOverrideHistory,
  800. let insulinSensitivitySchedule = insulinSensitivityScheduleApplyingOverrideHistory
  801. {
  802. return samples.map(
  803. to: velocities,
  804. carbRatio: carbRatioSchedule,
  805. insulinSensitivity: insulinSensitivitySchedule,
  806. absorptionTimeOverrun: absorptionTimeOverrun,
  807. defaultAbsorptionTime: defaultAbsorptionTimes.medium,
  808. delay: delay,
  809. initialAbsorptionTimeOverrun: settings.initialAbsorptionTimeOverrun,
  810. absorptionModel: settings.absorptionModel,
  811. adaptiveAbsorptionRateEnabled: settings.adaptiveAbsorptionRateEnabled,
  812. adaptiveRateStandbyIntervalFraction: settings.adaptiveRateStandbyIntervalFraction
  813. ).dynamicCarbsOnBoard(
  814. from: start,
  815. to: end,
  816. defaultAbsorptionTime: defaultAbsorptionTimes.medium,
  817. absorptionModel: settings.absorptionModel,
  818. delay: delay,
  819. delta: delta
  820. )
  821. } else {
  822. return samples.carbsOnBoard(
  823. from: start,
  824. to: end,
  825. defaultAbsorptionTime: defaultAbsorptionTimes.medium,
  826. absorptionModel: settings.absorptionModel,
  827. delay: delay,
  828. delta: delta
  829. )
  830. }
  831. }
  832. /// Computes the single carbs on-board value occuring just prior or equal to the specified date
  833. /// - Parameters:
  834. /// - date: The date of the value to retrieve
  835. /// - effectVelocities: A timeline of glucose effect velocities, ordered by start date
  836. /// - Returns: The carbs on-board value
  837. public func carbsOnBoard<Sample: CarbEntry>(
  838. from samples: [Sample],
  839. at date: Date,
  840. effectVelocities: [GlucoseEffectVelocity]? = nil
  841. ) throws -> CarbValue {
  842. let values = carbsOnBoard(from: samples, startingAt: date.addingTimeInterval(-delta), endingAt: date, effectVelocities: effectVelocities)
  843. guard let value = values.closestPrior(to: date) else {
  844. throw CarbStoreError.noData
  845. }
  846. return value
  847. }
  848. /// Retrieves a timeline of effect on blood glucose from carbohydrates
  849. ///
  850. /// This operation is performed asynchronously and the completion will be executed on an arbitrary background queue.
  851. ///
  852. /// - Parameters:
  853. /// - start: The earliest date of effects to retrieve
  854. /// - end: The latest date of effects to retrieve, if provided
  855. /// - effectVelocities: A timeline of glucose effect velocities, ordered by start date
  856. /// - completion: A closure called once the effects have been retrieved
  857. /// - result: An array of effects, in chronological order
  858. public func getGlucoseEffects(start: Date, end: Date? = nil, effectVelocities: [GlucoseEffectVelocity]? = nil, completion: @escaping(_ result: CarbStoreResult<(entries: [StoredCarbEntry], effects: [GlucoseEffect])>) -> Void) {
  859. queue.async {
  860. guard self.carbRatioSchedule != nil, self.insulinSensitivitySchedule != nil else {
  861. completion(.failure(.notConfigured))
  862. return
  863. }
  864. // To know glucose effects at the requested start date, we need to fetch samples that might still be absorbing
  865. let foodStart = start.addingTimeInterval(-self.maximumAbsorptionTimeInterval)
  866. self.getCarbEntries(start: foodStart, end: end) { (result) in
  867. switch result {
  868. case .failure(let error):
  869. completion(.failure(error))
  870. case .success(let entries):
  871. do {
  872. let effects = try self.glucoseEffects(of: entries, startingAt: start, endingAt: end, effectVelocities: effectVelocities)
  873. completion(.success((entries: entries, effects: effects)))
  874. } catch let error as CarbStoreError {
  875. completion(.failure(error))
  876. } catch {
  877. fatalError()
  878. }
  879. }
  880. }
  881. }
  882. }
  883. /// Computes a timeline of effects on blood glucose from carbohydrates
  884. /// - Parameters:
  885. /// - start: The earliest date of effects to retrieve
  886. /// - end: The latest date of effects to retrieve, if provided
  887. /// - effectVelocities: A timeline of glucose effect velocities, ordered by start date
  888. public func glucoseEffects<Sample: CarbEntry>(
  889. of samples: [Sample],
  890. startingAt start: Date,
  891. endingAt end: Date? = nil,
  892. effectVelocities: [GlucoseEffectVelocity]? = nil
  893. ) throws -> [GlucoseEffect] {
  894. guard
  895. let carbRatioSchedule = carbRatioScheduleApplyingOverrideHistory,
  896. let insulinSensitivitySchedule = insulinSensitivityScheduleApplyingOverrideHistory
  897. else {
  898. throw CarbStoreError.notConfigured
  899. }
  900. if let effectVelocities = effectVelocities {
  901. return samples.map(
  902. to: effectVelocities,
  903. carbRatio: carbRatioSchedule,
  904. insulinSensitivity: insulinSensitivitySchedule,
  905. absorptionTimeOverrun: absorptionTimeOverrun,
  906. defaultAbsorptionTime: defaultAbsorptionTimes.medium,
  907. delay: delay,
  908. initialAbsorptionTimeOverrun: settings.initialAbsorptionTimeOverrun,
  909. absorptionModel: settings.absorptionModel,
  910. adaptiveAbsorptionRateEnabled: settings.adaptiveAbsorptionRateEnabled,
  911. adaptiveRateStandbyIntervalFraction: settings.adaptiveRateStandbyIntervalFraction
  912. ).dynamicGlucoseEffects(
  913. from: start,
  914. to: end,
  915. carbRatios: carbRatioSchedule,
  916. insulinSensitivities: insulinSensitivitySchedule,
  917. defaultAbsorptionTime: defaultAbsorptionTimes.medium,
  918. absorptionModel: settings.absorptionModel,
  919. delay: delay,
  920. delta: delta
  921. )
  922. } else {
  923. return samples.glucoseEffects(
  924. from: start,
  925. to: end,
  926. carbRatios: carbRatioSchedule,
  927. insulinSensitivities: insulinSensitivitySchedule,
  928. defaultAbsorptionTime: defaultAbsorptionTimes.medium,
  929. absorptionModel: settings.absorptionModel,
  930. delay: delay,
  931. delta: delta
  932. )
  933. }
  934. }
  935. /// Retrieves the total number of recorded carbohydrates for the specified period.
  936. ///
  937. /// This operation is performed asynchronously and the completion will be executed on an arbitrary background queue.
  938. ///
  939. /// - Parameters:
  940. /// - start: The earliest date of samples to include.
  941. /// - completion: A closure called once the value has been retrieved.
  942. /// - result: The total carbs recorded and the date of the first sample
  943. public func getTotalCarbs(since start: Date, completion: @escaping (_ result: CarbStoreResult<CarbValue>) -> Void) {
  944. getCarbEntries(start: start) { (result) in
  945. switch result {
  946. case .success(let samples):
  947. let total = samples.totalCarbs ?? CarbValue(
  948. startDate: start,
  949. quantity: HKQuantity(unit: .gram(), doubleValue: 0)
  950. )
  951. completion(.success(total))
  952. case .failure(let error):
  953. completion(.failure(error))
  954. }
  955. }
  956. }
  957. }
  958. // MARK: - Remote Data Service Query
  959. extension CarbStore {
  960. public struct QueryAnchor: Equatable, RawRepresentable {
  961. public typealias RawValue = [String: Any]
  962. internal var anchorKey: Int64
  963. public init() {
  964. self.anchorKey = 0
  965. }
  966. public init?(rawValue: RawValue) {
  967. guard let anchorKey = (rawValue["anchorKey"] ?? rawValue["storedModificationCounter"]) as? Int64 else { // Backwards compatibility with storedModificationCounter
  968. return nil
  969. }
  970. self.anchorKey = anchorKey
  971. }
  972. public var rawValue: RawValue {
  973. var rawValue: RawValue = [:]
  974. rawValue["anchorKey"] = anchorKey
  975. return rawValue
  976. }
  977. }
  978. public enum CarbQueryResult {
  979. case success(QueryAnchor, [SyncCarbObject], [SyncCarbObject], [SyncCarbObject])
  980. case failure(Error)
  981. }
  982. public func executeCarbQuery(fromQueryAnchor queryAnchor: QueryAnchor?, limit: Int, completion: @escaping (CarbQueryResult) -> Void) {
  983. queue.async {
  984. var queryAnchor = queryAnchor ?? QueryAnchor()
  985. var queryCreatedResult = [SyncCarbObject]()
  986. var queryUpdatedResult = [SyncCarbObject]()
  987. var queryDeletedResult = [SyncCarbObject]()
  988. var queryError: Error?
  989. guard limit > 0 else {
  990. completion(.success(queryAnchor, [], [], []))
  991. return
  992. }
  993. self.cacheStore.managedObjectContext.performAndWait {
  994. let storedRequest: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  995. storedRequest.predicate = NSPredicate(format: "anchorKey > %d", queryAnchor.anchorKey)
  996. storedRequest.sortDescriptors = [NSSortDescriptor(key: "anchorKey", ascending: true)]
  997. storedRequest.fetchLimit = limit
  998. do {
  999. let stored = try self.cacheStore.managedObjectContext.fetch(storedRequest)
  1000. if let anchorKey = stored.max(by: { $0.anchorKey < $1.anchorKey })?.anchorKey {
  1001. queryAnchor.anchorKey = anchorKey
  1002. }
  1003. stored.map({ SyncCarbObject(managedObject: $0) }).forEach {
  1004. switch $0.operation {
  1005. case .create:
  1006. queryCreatedResult.append($0)
  1007. case .update:
  1008. queryUpdatedResult.append($0)
  1009. case .delete:
  1010. queryDeletedResult.append($0)
  1011. }
  1012. }
  1013. } catch let coreDataError {
  1014. queryError = coreDataError
  1015. return
  1016. }
  1017. }
  1018. if let queryError = queryError {
  1019. completion(.failure(queryError))
  1020. return
  1021. }
  1022. completion(.success(queryAnchor, queryCreatedResult, queryUpdatedResult, queryDeletedResult))
  1023. }
  1024. }
  1025. }
  1026. // MARK: - Critical Event Log Export
  1027. extension CarbStore: CriticalEventLog {
  1028. private var exportProgressUnitCountPerObject: Int64 { 1 }
  1029. private var exportFetchLimit: Int { Int(criticalEventLogExportProgressUnitCountPerFetch / exportProgressUnitCountPerObject) }
  1030. public var exportName: String { "Carbs.json" }
  1031. public func exportProgressTotalUnitCount(startDate: Date, endDate: Date? = nil) -> Result<Int64, Error> {
  1032. var result: Result<Int64, Error>?
  1033. self.cacheStore.managedObjectContext.performAndWait {
  1034. do {
  1035. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  1036. request.predicate = self.exportDatePredicate(startDate: startDate, endDate: endDate)
  1037. let objectCount = try self.cacheStore.managedObjectContext.count(for: request)
  1038. result = .success(Int64(objectCount) * exportProgressUnitCountPerObject)
  1039. } catch let error {
  1040. result = .failure(error)
  1041. }
  1042. }
  1043. return result!
  1044. }
  1045. public func export(startDate: Date, endDate: Date, to stream: OutputStream, progress: Progress) -> Error? {
  1046. let encoder = JSONStreamEncoder(stream: stream)
  1047. var anchorKey: Int64 = 0
  1048. var fetching = true
  1049. var error: Error?
  1050. while fetching && error == nil {
  1051. self.cacheStore.managedObjectContext.performAndWait {
  1052. do {
  1053. guard !progress.isCancelled else {
  1054. throw CriticalEventLogError.cancelled
  1055. }
  1056. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  1057. request.predicate = NSCompoundPredicate(andPredicateWithSubpredicates: [NSPredicate(format: "anchorKey > %d", anchorKey),
  1058. self.exportDatePredicate(startDate: startDate, endDate: endDate)])
  1059. request.sortDescriptors = [NSSortDescriptor(key: "anchorKey", ascending: true)]
  1060. request.fetchLimit = self.exportFetchLimit
  1061. let objects = try self.cacheStore.managedObjectContext.fetch(request)
  1062. if objects.isEmpty {
  1063. fetching = false
  1064. return
  1065. }
  1066. try encoder.encode(objects)
  1067. anchorKey = objects.last!.anchorKey
  1068. progress.completedUnitCount += Int64(objects.count) * exportProgressUnitCountPerObject
  1069. } catch let fetchError {
  1070. error = fetchError
  1071. }
  1072. }
  1073. }
  1074. if let closeError = encoder.close(), error == nil {
  1075. error = closeError
  1076. }
  1077. return error
  1078. }
  1079. private func exportDatePredicate(startDate: Date, endDate: Date? = nil) -> NSPredicate {
  1080. var addedDatePredicate = NSPredicate(format: "addedDate >= %@", startDate as NSDate)
  1081. var supercededDatePredicate = NSPredicate(format: "supercededDate >= %@", startDate as NSDate)
  1082. if let endDate = endDate {
  1083. addedDatePredicate = NSCompoundPredicate(andPredicateWithSubpredicates: [addedDatePredicate, NSPredicate(format: "addedDate < %@", endDate as NSDate)])
  1084. supercededDatePredicate = NSCompoundPredicate(andPredicateWithSubpredicates: [supercededDatePredicate, NSPredicate(format: "supercededDate < %@", endDate as NSDate)])
  1085. }
  1086. return NSCompoundPredicate(orPredicateWithSubpredicates: [addedDatePredicate, supercededDatePredicate])
  1087. }
  1088. }
  1089. // MARK: - Core Data (Bulk) - TEST ONLY
  1090. extension CarbStore {
  1091. public func addNewCarbEntries(entries: [NewCarbEntry], completion: @escaping (Error?) -> Void) {
  1092. guard !entries.isEmpty else {
  1093. completion(nil)
  1094. return
  1095. }
  1096. queue.async {
  1097. var error: Error?
  1098. self.cacheStore.managedObjectContext.performAndWait {
  1099. do {
  1100. for entry in entries {
  1101. let syncIdentifier = try self.cacheStore.managedObjectContext.generateUniqueSyncIdentifier()
  1102. let object = CachedCarbObject(context: self.cacheStore.managedObjectContext)
  1103. object.create(from: entry,
  1104. provenanceIdentifier: self.provenanceIdentifier,
  1105. syncIdentifier: syncIdentifier,
  1106. on: entry.date)
  1107. }
  1108. error = self.cacheStore.save()
  1109. } catch let coreDataError {
  1110. error = coreDataError
  1111. }
  1112. }
  1113. guard error == nil else {
  1114. completion(error)
  1115. return
  1116. }
  1117. self.log.info("Added %d CachedCarbObjects", entries.count)
  1118. self.delegate?.carbStoreHasUpdatedCarbData(self)
  1119. completion(nil)
  1120. }
  1121. }
  1122. }
  1123. // MARK: - Issue Report
  1124. extension CarbStore {
  1125. /// Generates a diagnostic report about the current state
  1126. ///
  1127. /// This operation is performed asynchronously and the completion will be executed on an arbitrary background queue.
  1128. ///
  1129. /// - parameter completionHandler: A closure called once the report has been generated. The closure takes a single argument of the report string.
  1130. public func generateDiagnosticReport(_ completionHandler: @escaping (_ report: String) -> Void) {
  1131. queue.async {
  1132. var carbAbsorptionModel: String
  1133. switch self.carbAbsorptionModel {
  1134. case .linear:
  1135. carbAbsorptionModel = "Linear"
  1136. case .nonlinear:
  1137. carbAbsorptionModel = "Nonlinear"
  1138. case .adaptiveRateNonlinear:
  1139. carbAbsorptionModel = "Nonlinear with Adaptive Rate for Remaining Carbs"
  1140. }
  1141. var report: [String] = [
  1142. "## CarbStore",
  1143. "",
  1144. "* carbRatioSchedule: \(self.carbRatioSchedule?.debugDescription ?? "")",
  1145. "* carbRatioScheduleApplyingOverrideHistory: \(self.carbRatioScheduleApplyingOverrideHistory?.debugDescription ?? "nil")",
  1146. "* cacheLength: \(self.cacheLength)",
  1147. "* defaultAbsorptionTimes: \(self.defaultAbsorptionTimes)",
  1148. "* observationInterval: \(self.observationInterval)",
  1149. "* insulinSensitivitySchedule: \(self.insulinSensitivitySchedule?.debugDescription ?? "")",
  1150. "* insulinSensitivityScheduleApplyingOverrideHistory: \(self.insulinSensitivityScheduleApplyingOverrideHistory?.debugDescription ?? "nil")",
  1151. "* overrideHistory: \(self.overrideHistory.map(String.init(describing:)) ?? "nil")",
  1152. "* carbSensitivitySchedule: \(self.carbSensitivitySchedule?.debugDescription ?? "nil")",
  1153. "* delay: \(self.delay)",
  1154. "* delta: \(self.delta)",
  1155. "* absorptionTimeOverrun: \(self.absorptionTimeOverrun)",
  1156. "* carbAbsorptionModel: \(carbAbsorptionModel)",
  1157. "* Carb absorption model settings: \(self.settings)",
  1158. super.debugDescription,
  1159. "",
  1160. "cachedCarbEntries:"
  1161. ]
  1162. switch self.getCarbEntries() {
  1163. case .failure(let error):
  1164. report.append("Error: \(error)")
  1165. case .success(let entries):
  1166. report.append("[")
  1167. report.append("\tStoredCarbEntry(uuid, provenanceIdentifier, syncIdentifier, syncVersion, startDate, quantity, foodType, absorptionTime, createdByCurrentApp, userCreatedDate, userUpdatedDate)")
  1168. report.append(entries.map({ (entry) -> String in
  1169. return [
  1170. "\t",
  1171. entry.uuid?.uuidString ?? "",
  1172. entry.provenanceIdentifier,
  1173. entry.syncIdentifier ?? "",
  1174. entry.syncVersion != nil ? String(describing: entry.syncVersion) : "",
  1175. String(describing: entry.startDate),
  1176. String(describing: entry.quantity),
  1177. entry.foodType ?? "",
  1178. String(describing: entry.absorptionTime ?? self.defaultAbsorptionTimes.medium),
  1179. String(describing: entry.createdByCurrentApp),
  1180. entry.userCreatedDate != nil ? String(describing: entry.userCreatedDate) : "",
  1181. entry.userUpdatedDate != nil ? String(describing: entry.userUpdatedDate) : "",
  1182. ].joined(separator: ", ")
  1183. }).joined(separator: "\n"))
  1184. report.append("]")
  1185. report.append("")
  1186. }
  1187. completionHandler(report.joined(separator: "\n"))
  1188. }
  1189. }
  1190. }
  1191. // MARK: - NSManagedObjectContext
  1192. fileprivate extension NSManagedObjectContext {
  1193. func generateUniqueSyncIdentifier() throws -> String {
  1194. while true {
  1195. let syncIdentifier = UUID().uuidString
  1196. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  1197. request.predicate = NSPredicate(format: "syncIdentifier == %@", syncIdentifier)
  1198. request.fetchLimit = 1
  1199. if try count(for: request) == 0 {
  1200. return syncIdentifier
  1201. }
  1202. }
  1203. }
  1204. func cachedCarbObjectFromStoredCarbEntry(_ entry: StoredCarbEntry) throws -> CachedCarbObject? {
  1205. guard entry.createdByCurrentApp, let syncIdentifier = entry.syncIdentifier, let syncVersion = entry.syncVersion else {
  1206. return nil
  1207. }
  1208. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  1209. request.predicate = NSCompoundPredicate(andPredicateWithSubpredicates: [
  1210. NSPredicate(format: "createdByCurrentApp == YES"),
  1211. NSPredicate(format: "syncIdentifier == %@", syncIdentifier),
  1212. NSPredicate(format: "syncVersion == %d", syncVersion),
  1213. NSPredicate(format: "operation != %d", Operation.delete.rawValue),
  1214. NSPredicate(format: "supercededDate == NIL")
  1215. ])
  1216. request.fetchLimit = 1
  1217. if let object = try fetch(request).first {
  1218. return object
  1219. }
  1220. return try cachedCarbObjectFromStoredCarbEntryDEPRECATED(entry)
  1221. }
  1222. // DEPRECATED: Fallback for pre-syncIdentifier entries, just has UUID from HealthKit
  1223. func cachedCarbObjectFromStoredCarbEntryDEPRECATED(_ entry: StoredCarbEntry) throws -> CachedCarbObject? {
  1224. guard entry.createdByCurrentApp, let uuid = entry.uuid else {
  1225. return nil
  1226. }
  1227. let request: NSFetchRequest<CachedCarbObject> = CachedCarbObject.fetchRequest()
  1228. request.predicate = NSCompoundPredicate(andPredicateWithSubpredicates: [
  1229. NSPredicate(format: "createdByCurrentApp == YES"),
  1230. NSPredicate(format: "uuid == %@", uuid as NSUUID),
  1231. NSPredicate(format: "operation != %d", Operation.delete.rawValue),
  1232. NSPredicate(format: "supercededDate == NIL")
  1233. ])
  1234. request.fetchLimit = 1
  1235. if let object = try fetch(request).first {
  1236. return object
  1237. }
  1238. return nil
  1239. }
  1240. }