interface ExampleState {
    val field1: Boolean
    val field2: String
    val field3: Int
}

public interface MutableExampleState : ExampleState, StateFlowable<ExampleState> {
    // Override fields

    fun update(mutate: MutableExampleState.() -> Unit)
}

private class MutableExampleStateImpl(
    field1: Boolean,
    field2: String,
    field3: Int,
) : MutableExampleState {
    // 🔴 1. To track transaction status of transaction with reactivity
    private val transactionInProgress = MutableStateFlow<Boolean>(false)
    //...

    // 🔴 2. Implement function to update states in transaction
    override fun update(mutate: MutableExampleState.() -> Unit) {
        while (true) {
            // If other transaction is started in between, wait for its completion and
            // then only proceed for the current.
            if (transactionInProgress.compareAndSet(expect = false, update = true)) {
                mutate()
                transactionInProgress.update { false }
                return
            }
        }
    }

    private val _immutableStateFlowImpl : StateFlow<ExampleState> = object : StateFlow<ExampleState> {
        public override val `value`: ExampleState
            get() = ImmutableExampleState(
                field1 = _field1.value,
                field2 = _field2.value,
                field3 = _field3.value,
            )
        
        // 🔴 3. Change the implementation of collect and take care of collection while in transaction
        public override suspend fun collect(collector: FlowCollector<ExampleState>): Nothing = coroutineScope {
            combine(transactionInProgress, _field1, _field2, _field3) { params ->
                val isTransactionRunning = params[0] as Boolean
                if (!isTransactionRunning) {
                    value
                } else {
                    null
                }
            }.filterNotNull().stateIn(this).collect(collector)
         }
    }

val _state = MutableExampleState(false, "String", 0)

// 🔴 4. Use this method only to update values in transaction
_state.update {
    field1 = false
    field2 = "Example"
    field3 = 100
}