chore: minor tweaks

Author: edison1105

packages/runtime-core/src/helpers/useTemplateRef.ts

@@ -10,14 +10,15 @@ export function useTemplateRef<T = unknown, Keys extends string = string>(
 ): Readonly<ShallowRef<T | null>> {
   const i = getCurrentInstance()
   const r = shallowRef(null)
+  const ret = __DEV__ ? readonly(r) : r
   if (i) {
     const refs = i.refs === EMPTY_OBJ ? (i.refs = {}) : i.refs
     const refsCache =
       i.refsCache || (i.refsCache = new Map<string, ShallowRef>())
     if (refsCache.has(key)) {
       return refsCache.get(key)!
     } else {
-      refsCache.set(key, r)
+      refsCache.set(key, ret)
       Object.defineProperty(refs, key, {
         enumerable: true,
         get: () => r.value,
@@ -30,7 +31,6 @@ export function useTemplateRef<T = unknown, Keys extends string = string>(
         `instance to be associated with.`,
     )
   }
-  const ret = __DEV__ ? readonly(r) : r
   if (__DEV__) {
     knownTemplateRefs.add(ret)
   }