From: Evan You <yyx990803@gmail.com>
Date: Tue, 20 Aug 2019 13:45:28 +0000 (-0400)
Subject: wip: make computed implementation consistent
X-Git-Tag: v3.0.0-alpha.0~913
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=0519e105189da6ba994d60da3107beca71e8745c;p=thirdparty%2Fvuejs%2Fcore.git

wip: make computed implementation consistent
---

diff --git a/packages/reactivity/src/computed.ts b/packages/reactivity/src/computed.ts
index 5cbd1b5a2c..ef6e17c138 100644
--- a/packages/reactivity/src/computed.ts
+++ b/packages/reactivity/src/computed.ts
@@ -1,17 +1,29 @@
 import { effect, ReactiveEffect, activeReactiveEffectStack } from './effect'
 import { UnwrapNestedRefs, knownRefs } from './ref'
+import { isFunction } from '@vue/shared'
 
 export interface ComputedRef<T> {
   readonly value: UnwrapNestedRefs<T>
   readonly effect: ReactiveEffect
 }
 
+export interface ComputedOptions<T> {
+  get: () => T
+  set: (v: T) => void
+}
+
 export function computed<T>(
-  getter: () => T,
-  setter?: (v: T) => void
+  getterOrOptions: (() => T) | ComputedOptions<T>
 ): ComputedRef<T> {
+  const isReadonly = isFunction(getterOrOptions)
+  const getter = isReadonly
+    ? (getterOrOptions as (() => T))
+    : (getterOrOptions as ComputedOptions<T>).get
+  const setter = isReadonly ? null : (getterOrOptions as ComputedOptions<T>).set
+
   let dirty: boolean = true
   let value: any = undefined
+
   const runner = effect(getter, {
     lazy: true,
     // mark effect as computed so that it gets priority during trigger
diff --git a/packages/reactivity/src/index.ts b/packages/reactivity/src/index.ts
index 99112a94ca..71893f6365 100644
--- a/packages/reactivity/src/index.ts
+++ b/packages/reactivity/src/index.ts
@@ -8,7 +8,7 @@ export {
   markImmutable,
   markNonReactive
 } from './reactive'
-export { computed, ComputedRef } from './computed'
+export { computed, ComputedRef, ComputedOptions } from './computed'
 export {
   effect,
   stop,
diff --git a/packages/runtime-core/src/apiReactivity.ts b/packages/runtime-core/src/apiReactivity.ts
index 327cc38527..fa67f54cbb 100644
--- a/packages/runtime-core/src/apiReactivity.ts
+++ b/packages/runtime-core/src/apiReactivity.ts
@@ -23,11 +23,11 @@ export {
 import {
   computed as _computed,
   ComputedRef,
+  ComputedOptions,
   ReactiveEffect
 } from '@vue/reactivity'
 
 import { currentInstance } from './component'
-import { isFunction } from '@vue/shared'
 
 // record effects created during a component's setup() so that they can be
 // stopped when the component unmounts
@@ -37,20 +37,10 @@ export function recordEffect(effect: ReactiveEffect) {
   }
 }
 
-interface ComputedOptions<T> {
-  get: () => T
-  set: (v: T) => void
-}
-
 export function computed<T>(
   getterOrOptions: (() => T) | ComputedOptions<T>
 ): ComputedRef<T> {
-  let c: ComputedRef<T>
-  if (isFunction(getterOrOptions)) {
-    c = _computed(getterOrOptions)
-  } else {
-    c = _computed(getterOrOptions.get, getterOrOptions.set)
-  }
+  const c = _computed(getterOrOptions)
   recordEffect(c.effect)
   return c
 }