牛逼！Vue3.5的useTemplateRef让ref操作DOM更加丝滑
大家好，我是欧阳，又跟大家见面啦！
欧阳写了一本vue3编译原理揭秘开源电子书，看完后可以让你对vue编译的认知有质的提升。并且这本书初、中级前端都能看懂，完全免费，只求一个star，点击文末的阅读原文跳转到电子书。
前言
vue3中想要访问DOM和子组件可以使用ref进行模版引用，但是这个ref有一些让人迷惑的地方。比如定义的ref变量到底是一个响应式数据还是DOM元素？还有template中ref属性的值明明是一个字符串，比如ref="inputEl"，怎么就和script中同名的inputEl变量绑到一块了呢？所以Vue3.5推出了一个useTemplateRef函数，完美的解决了这些问题。
ref模版引用的问题
我们先来看一个react中使用ref访问DOM元素的例子，代码如下：
const inputEl = useRef<HTMLInputElement>(null);
<input type="text" ref={inputEl} />
使用useRef函数定义了一个名为inputEl的变量，然后将input元素的ref属性值设置为inputEl变量，这样就可以通过inputEl变量访问到input输入框了。
inputEl因为是一个.current属性的对象，由于inputEl变量赋值给了ref属性，所以他的.current属性的值被更新为了input DOM元素，这个做法很符合编程直觉。
再来看看vue3中的做法，相比之下就很不符合编程直觉了。
不知道有多少同学和欧阳一样，最开始接触vue3时总是在template中像react一样给ref属性绑定一个ref变量，而不是ref变量的名称。比如下面这样的代码：
<input type="text" :ref="inputEl" />
const inputEl = ref<HTMLInputElement>();
更加要命的是这样写还不会报错！！！！当我们使用inputEl变量去访问input输入框时始终拿到的都是undefined。
经过多次排查发现原来ref属性接收的不是一个ref变量，而是ref变量的名称。正确的代码应该是这样的：
<input type="text" ref="inputEl" />
const inputEl = ref<HTMLInputElement>();
还有就是如果我们将ref模版引用相关的逻辑抽成hooks后，那么必须将在vue组件中也要将ref属性对应的ref变量也定义才可以。
hooks代码如下：
export default function useRef() {
const inputEl = ref<HTMLInputElement>();
function setInputValue() {
if (inputEl.value) {
inputEl.value.value = "Hello, world!";
}
}
return {
inputEl,
setInputValue,
};
}
在hooks中定义了一个名为inputEl的变量，并且在setInputValue函数中会通过inputEl变量对input输入框进行操作。
vue组件代码如下：
<template>
<input type="text" ref="inputEl" />
<button @click="setInputValue">给input赋值</button>
</template>
<script setup lang="ts">
import useInput from "./useInput";
const { setInputValue, inputEl } = useInput();
</script>
虽然在vue组件中我们不会使用inputEl变量，但是还是需要从hooks中导入useInput变量。大家不觉得这很奇怪吗？导入了一个变量，又没有显式的去使用这个变量。
如果在这里不去从hooks中导入inputEl变量，那么inputEl变量中就不能绑定上input输入框了。
useTemplateRef函数
为了解决上面说的ref模版引用的问题，在Vue3.5中新增了一个useTemplateRef函数。
useTemplateRef函数的用法很简单：只接收一个参数key，是一个字符串。返回值是一个ref变量。
其中参数key字符串的值应该等于template中ref属性的值。
返回值是一个ref变量，变量的值指向模版引用的DOM元素或者子组件。
我们来看个例子，前面的demo改成useTemplateRef函数后代码如下：
<template>
<input type="text" ref="inputRef" />
<button @click="setInputValue">给input赋值</button>
</template>
<script setup lang="ts">
import { useTemplateRef } from "vue";
const inputEl = useTemplateRef<HTMLInputElement>("inputRef");
function setInputValue() {
if (inputEl.value) {
inputEl.value.value = "Hello, world!";
}
}
</script>
在template中ref属性的值为字符串"inputRef"。
在script中使用useTemplateRef函数，传入的第一个参数也是字符串"inputRef"。useTemplateRef函数的返回值就是指向input输入框的ref变量。
由于inputEl是一个ref变量，所以在click事件中想要访问到DOM元素input输入框就需要使用inputEl.value。
我们这里是要给输入框中塞一个字符串"Hello, world!"，所以使用inputEl.value.value = "Hello, world!"
使用了useTemplateRef函数后和之前比起来就很符合编程直觉了。template中ref属性值是一个字符串"inputRef"，使用useTemplateRef函数时也传入字符串"inputRef"就能拿到对应的模版引用了。
hooks中使用useTemplateRef
回到前面讲的hooks的例子，使用useTemplateRef后hooks代码如下：
export default function useInput(key) {
const inputEl = useTemplateRef<HTMLInputElement>(key);
function setInputValue() {
if (inputEl.value) {
inputEl.value.value = "Hello, world!";
}
}
return {
setInputValue,
};
}
现在我们在hooks中就不需要导出变量inputEl了，因为这个变量只需要在hooks内部使用。
vue组件代码如下：
<template>
<input type="text" ref="inputRef" />
<button @click="setInputValue">给input赋值</button>
</template>
<script setup lang="ts">
import useInput from "./useInput";
const { setInputValue } = useInput("inputRef");
</script>
由于在vue组件中我们不需要使用inputEl变量，所以在这里就不需要从useInput中引入变量inputEl了。而之前不使用useTemplateRef的方案中我们就不得不引入inputEl变量了。
动态切换ref绑定的变量
有的时候我们需要根据不同的场景去动态切换ref模版引用的变量，这时在template中ref属性的值就是动态的了，而不是一个写死的字符串。在这种场景中useTemplateRef也是支持的，代码如下：
<template>
<input type="text" :ref="refKey" />
<button @click="switchRef">切换ref绑定的变量</button>
<button @click="setInputValue">给input赋值</button>
</template>
<script setup lang="ts">
import { useTemplateRef, ref } from "vue";
const refKey = ref("inputEl1");
const inputEl1 = useTemplateRef<HTMLInputElement>("inputEl1");
const inputEl2 = useTemplateRef<HTMLInputElement>("inputEl2");
function switchRef() {
refKey.value = refKey.value === "inputEl1" ? "inputEl2" : "inputEl1";
}
function setInputValue() {
const curEl = refKey.value === "inputEl1" ? inputEl1 : inputEl2;
if (curEl.value) {
curEl.value.value = "Hello, world!";
}
}
</script>
在这个场景template中ref绑定的就是一个变量refKey，通过点击切换ref绑定的变量按钮可以切换refKey的值。相应的，绑定input输入框的变量也会从inputEl1变量切换成inputEl2变量。
useTemplateRef是如何实现的？
我们来看看useTemplateRef的源码，其实很简单，简化后的代码如下：
function useTemplateRef(key) {
const i = getCurrentInstance();
const r = shallowRef(null);
if (i) {
const refs = i.refs === EMPTY_OBJ ? (i.refs = {}) : i.refs;
Object.defineProperty(refs, key, {
enumerable: true,
get: () => r.value,
set: (val) => (r.value = val),
});
}
return r;
}
首先使用getCurrentInstance方法获取当前vue实例对象，赋值给变量i。
然后调用shallowRef函数生成一个浅层的ref对象，初始值为null。这个ref对象就是useTemplateRef函数返回的ref对象。
接着就是判断当前vue实例如果存在就读取实例上面的refs属性对象，如果实例对象上面没有refs属性，那么就初始化一个空对象到vue实例对象的refs属性。
vue实例对象上面的这个refs属性对象用过vue2的同学应该都很熟悉，里面存的是注册过ref属性的所有 DOM 元素和组件实例。
vue3虽然不像vue2一样将refs属性对象开放给开发者，但是他的内部依然还是用vue实例上面的refs属性对象来存储template中使用ref属性注册过的元素和组件实例。
这里使用了Object.defineProperty方法对refs属性对象进行拦截，拦截的字段是变量key的值，而这个key的值就是template中使用ref属性绑定的值。
以我们上面的demo举例，在template中的代码如下：
<input type="text" ref="inputRef" />
这里使用ref属性在vue实例的refs属性对象上面注册了一个input输入框，refs.inputRef的值就是指向DOM元素input输入框。
然后在script中是这样使用useTemplateRef的：
const inputEl = useTemplateRef<HTMLInputElement>("inputRef")
调用useTemplateRef函数时传入的是字符串"inputRef"，在useTemplateRef函数内部使用Object.defineProperty方法对refs属性对象进行拦截，拦截的字段为变量key的值，也就是调用useTemplateRef函数传入的字符串"inputRef"。
初始化时，vue处理input输入框上面的ref="inputRef"就会执行下面这样的代码：
refs[ref] = value
此时的value的值就是指向DOM元素input输入框，ref的值就是字符串"inputRef"。
那么这行代码就是将DOM元素input输入框赋值给refs对象上面的inputRef属性上。
由于这里对refs对象上面的inputRef属性进行写操作，所以会走到useTemplateRef函数中Object.defineProperty定义的set拦截。代码如下：
const r = shallowRef(null);
Object.defineProperty(refs, key, {
enumerable: true,
get: () => r.value,
set: (val) => (r.value = val),
});
在set拦截中会将DOM元素input输入框赋值给ref变量r，而这个r就是useTemplateRef函数返回的ref变量。
同样的当对象refs对象的inputRef属性进行读操作时，也会走到这里的get拦截中，返回useTemplateRef函数中定义的ref变量r的值。
总结
Vue3.5中新增的useTemplateRef函数解决了ref属性中存在的几个问题：
不符合编程直觉，template中ref属性的值是script中对应的ref变量的变量名。
在script中如果不使用ts，则不能直观的知道一个ref变量到底是响应式数据还是DOM元素？
将定义和访问DOM元素相关的逻辑抽到hooks中后，虽然vue组件中不会使用到存放DOM元素的变量，但是也必须在组件中从hooks中导入。
接着我们讲了useTemplateRef函数的实现。在useTemplateRef函数中会定义一个ref对象，在useTemplateRef函数最后就是return返回这个ref对象。
接着使用Object.defineProperty对vue实例上面的refs属性对象进行get和set拦截。
初始化时，处理template中的ref属性，会对vue实例上面的refs属性对象进行写操作。
然后就会被set拦截，在set拦截中会将useTemplateRef函数中定义的ref对象的值赋值为绑定的DOM元素或者组件实例。
而useTemplateRef函数就是将这个ref对象进行return返回，所以我们可以通过useTemplateRef函数的返回值拿到template中ref属性绑定的DOM元素或者组件实例。
长按图片加欧阳微信，拉你进欧阳的高质量vue源码交流群，群里可谓是藏龙卧虎。
关注公众号，发送消息：
666，领取欧阳研究vue源码过程中收集的源码资料。欧阳写文章有时也会参考这些资料，同时让你的朋友圈多一位对vue有深入理解的人。
因为微信公众号修改规则，如果不标星或点在看，你可能会收不到我公众号文章的推送，请大家将本公众号星标，看完文章后记得点下赞或者在看，谢谢各位！