Skip to content

Commit 19e16a2

Browse files
ra1028ra1028
authored
Strict main actor isolation (#138)
* Refactoring * Add a test case that unsubscription performs on background thread * Add more main actor isolation
1 parent 330123c commit 19e16a2

24 files changed

+231
-129
lines changed

‎Sources/Atoms/Context/AtomTransactionContext.swift‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,14 @@ public struct AtomTransactionContext: AtomWatchableContext {
77
@usableFromInline
88
internal let _store: StoreContext
99
@usableFromInline
10-
internal let _transaction: Transaction
10+
internal let _transactionState: TransactionState
1111

1212
internal init(
1313
store: StoreContext,
14-
transaction: Transaction
14+
transactionState: TransactionState
1515
) {
1616
self._store = store
17-
self._transaction = transaction
17+
self._transactionState = transactionState
1818
}
1919

2020
/// Accesses the value associated with the given atom without watching it.
@@ -191,6 +191,6 @@ public struct AtomTransactionContext: AtomWatchableContext {
191191
@inlinable
192192
@discardableResult
193193
public func watch<Node: Atom>(_ atom: Node) -> Node.Produced {
194-
_store.watch(atom, in: _transaction)
194+
_store.watch(atom, in: _transactionState)
195195
}
196196
}

‎Sources/Atoms/Core/AtomState.swift‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,13 @@ internal protocol AtomStateProtocol: AnyObject {
33
associatedtype Effect: AtomEffect
44

55
var effect: Effect { get }
6-
var transaction: Transaction? { get set }
6+
var transactionState: TransactionState? { get set }
77
}
88

9+
@MainActor
910
internal final class AtomState<Effect: AtomEffect>: AtomStateProtocol {
1011
let effect: Effect
11-
var transaction: Transaction?
12+
var transactionState: TransactionState?
1213

1314
init(effect: Effect) {
1415
self.effect = effect

‎Sources/Atoms/Core/Producer/AtomProducerContext.swift‎

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,43 +1,43 @@
11
@MainActor
22
internal struct AtomProducerContext<Value> {
33
private let store: StoreContext
4-
private let transaction: Transaction
4+
private let transactionState: TransactionState
55
private let update: @MainActor (Value) -> Void
66

77
init(
88
store: StoreContext,
9-
transaction: Transaction,
9+
transactionState: TransactionState,
1010
update: @escaping @MainActor (Value) -> Void
1111
) {
1212
self.store = store
13-
self.transaction = transaction
13+
self.transactionState = transactionState
1414
self.update = update
1515
}
1616

1717
var isTerminated: Bool {
18-
transaction.isTerminated
18+
transactionState.isTerminated
1919
}
2020

2121
var onTermination: (@MainActor () -> Void)? {
22-
get { transaction.onTermination }
23-
nonmutating set { transaction.onTermination = newValue }
22+
get { transactionState.onTermination }
23+
nonmutating set { transactionState.onTermination = newValue }
2424
}
2525

2626
func update(with value: Value) {
2727
update(value)
2828
}
2929

3030
func transaction<T>(_ body: @MainActor (AtomTransactionContext) -> T) -> T {
31-
transaction.begin()
32-
let context = AtomTransactionContext(store: store, transaction: transaction)
33-
defer { transaction.commit() }
31+
transactionState.begin()
32+
let context = AtomTransactionContext(store: store, transactionState: transactionState)
33+
defer { transactionState.commit() }
3434
return body(context)
3535
}
3636

3737
func transaction<T>(_ body: @MainActor (AtomTransactionContext) async throws -> T) async rethrows -> T {
38-
transaction.begin()
39-
let context = AtomTransactionContext(store: store, transaction: transaction)
40-
defer { transaction.commit() }
38+
transactionState.begin()
39+
let context = AtomTransactionContext(store: store, transactionState: transactionState)
40+
defer { transactionState.commit() }
4141
return try await body(context)
4242
}
4343
}

‎Sources/Atoms/Core/ScopeKey.swift‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
@usableFromInline
22
internal struct ScopeKey: Hashable, CustomStringConvertible {
3+
@MainActor
34
final class Token {}
45

56
private let identifier: ObjectIdentifier

‎Sources/Atoms/Core/StoreContext.swift‎

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,8 @@ internal struct StoreContext {
100100
}
101101

102102
@usableFromInline
103-
func watch<Node: Atom>(_ atom: Node, in transaction: Transaction) -> Node.Produced {
104-
guard !transaction.isTerminated else {
103+
func watch<Node: Atom>(_ atom: Node, in transactionState: TransactionState) -> Node.Produced {
104+
guard !transactionState.isTerminated else {
105105
return read(atom)
106106
}
107107

@@ -112,8 +112,8 @@ internal struct StoreContext {
112112
let value = cache?.value ?? initialize(of: atom, for: key, override: override)
113113

114114
// Add an `Edge` from the upstream to downstream.
115-
store.graph.dependencies[transaction.key, default: []].insert(key)
116-
store.graph.children[key, default: []].insert(transaction.key)
115+
store.graph.dependencies[transactionState.key, default: []].insert(key)
116+
store.graph.children[key, default: []].insert(transactionState.key)
117117

118118
return value
119119
}
@@ -189,13 +189,13 @@ internal struct StoreContext {
189189
// Restore dependencies when the refresh is completed.
190190
attachDependencies(dependencies, for: key)
191191

192-
guard let transaction = state.transaction, let cache = lookupCache(of: atom, for: key) else {
192+
guard let transactionState = state.transactionState, let cache = lookupCache(of: atom, for: key) else {
193193
checkAndRelease(for: key)
194194
return value
195195
}
196196

197197
// Notify update unless it's cancelled or terminated by other operations.
198-
if !Task.isCancelled && !transaction.isTerminated {
198+
if !Task.isCancelled && !transactionState.isTerminated {
199199
update(atom: atom, for: key, oldValue: cache.value, newValue: value)
200200
}
201201

@@ -428,7 +428,7 @@ private extension StoreContext {
428428
}
429429
}
430430

431-
state?.transaction?.terminate()
431+
state?.transactionState?.terminate()
432432

433433
let context = AtomCurrentContext(store: self)
434434
state?.effect.released(context: context)
@@ -486,7 +486,7 @@ private extension StoreContext {
486486
of atom: Node,
487487
for key: AtomKey
488488
) -> AtomProducerContext<Node.Produced> {
489-
let transaction = Transaction(key: key) {
489+
let transactionState = TransactionState(key: key) {
490490
let oldDependencies = detachDependencies(for: key)
491491

492492
return {
@@ -502,11 +502,11 @@ private extension StoreContext {
502502

503503
let state = getState(of: atom, for: key)
504504
// Terminate the ongoing transaction first.
505-
state.transaction?.terminate()
505+
state.transactionState?.terminate()
506506
// Register the transaction state so it can be terminated from anywhere.
507-
state.transaction = transaction
507+
state.transactionState = transactionState
508508

509-
return AtomProducerContext(store: self, transaction: transaction) { newValue in
509+
return AtomProducerContext(store: self, transactionState: transactionState) { newValue in
510510
if let cache = lookupCache(of: atom, for: key) {
511511
update(atom: atom, for: key, oldValue: cache.value, newValue: newValue)
512512
}

‎Sources/Atoms/Core/StoreState.swift‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
internal struct StoreState {
1+
@MainActor
2+
internal final class StoreState {
23
var caches = [AtomKey: any AtomCacheProtocol]()
34
var states = [AtomKey: any AtomStateProtocol]()
45
var subscriptions = [AtomKey: [SubscriberKey: Subscription]]()

‎Sources/Atoms/Core/SubscriberKey.swift‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
internal struct SubscriberKey: Hashable {
2+
@MainActor
23
final class Token {}
34

45
private let identifier: ObjectIdentifier

‎Sources/Atoms/Core/SubscriberState.swift‎

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,21 @@
1-
final class SubscriberState {
1+
import Foundation
2+
3+
@MainActor
4+
internal final class SubscriberState {
25
let token = SubscriberKey.Token()
36
var subscribing = Set<AtomKey>()
47
var unsubscribe: ((Set<AtomKey>) -> Void)?
58

6-
init() {}
7-
9+
// TODO: Use isolated synchronous deinit once it's available.
10+
// 0371-isolated-synchronous-deinit
811
deinit {
9-
unsubscribe?(subscribing)
12+
if Thread.isMainThread {
13+
unsubscribe?(subscribing)
14+
}
15+
else {
16+
Task(priority: .high) { @MainActor [unsubscribe, subscribing] in
17+
unsubscribe?(subscribing)
18+
}
19+
}
1020
}
1121
}

‎Sources/Atoms/Core/Subscription.swift‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
@usableFromInline
2-
@MainActor
32
internal struct Subscription {
43
let location: SourceLocation
5-
let update: () -> Void
4+
let update: @MainActor @Sendable () -> Void
65
}

‎Sources/Atoms/Core/Transaction.swift‎ renamed to ‎Sources/Atoms/Core/TransactionState.swift‎

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,18 @@
11
@usableFromInline
22
@MainActor
3-
internal final class Transaction {
4-
private var body: (() -> () -> Void)?
5-
private var cleanup: (() -> Void)?
3+
internal final class TransactionState {
4+
private var body: (@MainActor () -> @MainActor () -> Void)?
5+
private var cleanup: (@MainActor () -> Void)?
66

77
let key: AtomKey
88

99
private var termination: (@MainActor () -> Void)?
1010
private(set) var isTerminated = false
1111

12-
init(key: AtomKey, _ body: @escaping () -> () -> Void) {
12+
init(
13+
key: AtomKey,
14+
_ body: @MainActor @escaping () -> @MainActor () -> Void
15+
) {
1316
self.key = key
1417
self.body = body
1518
}

0 commit comments

Comments
 (0)