Vue 3 的 setup语法糖是怎么工作的?
共 11764字，需浏览 24分钟
·
2024-04-11 03:21
我们每天写vue3项目的时候都会使用setup语法糖，但是你有没有思考过下面几个问题。setup语法糖经过编译后是什么样子的？为什么在setup顶层定义的变量可以在template中可以直接使用？为什么import一个组件后就可以直接使用，无需使用components 选项来显式注册组件？
要回答上面的问题，我们先来了解一下从一个vue文件到渲染到浏览器这一过程经历了什么？
我们的vue代码一般都是写在后缀名为vue的文件上，显然浏览器是不认识vue文件的，浏览器只认识html、css、jss等文件。所以第一步就是通过webpack或者vite将一个vue文件编译为一个包含render函数的js文件。然后执行render函数生成虚拟DOM，再调用浏览器的DOM API根据虚拟DOM生成真实DOM挂载到浏览器上。
setup编译后的样子
在javascript标准中script标签是不支持setup属性的，浏览器根本就不认识setup属性。所以很明显setup是作用于编译时阶段，也就是从vue文件编译为js文件这一过程。
我们来看一个简单的demo，这个是index.vue源代码：
<template>
<h1>{{ title }}</h1>
<h1>{{ msg }}</h1>
<Child />
</template>
<script lang="ts" setup>
import { ref } from "vue";
import Child from "./child.vue";
const msg = ref("Hello World!");
const title = "title";
if (msg.value) {
const content = "content";
console.log(content);
}
</script>
这里我们定义了一个名为msg的ref响应式变量和非响应式的title变量，还有import了child.vue组件。
这个是child.vue的源代码
<template>
<div>i am child</div>
</template>
我们接下来看index.vue编译后的样子，代码我已经做过了简化：
import { ref } from "vue";
import Child from "./Child.vue";
const title = "title";
const __sfc__ = {
__name: "index",
setup() {
const msg = ref("Hello World!");
if (msg.value) {
const content = "content";
console.log(content);
}
const __returned__ = { title, msg, Child };
return __returned__;
},
};
import {
toDisplayString as _toDisplayString,
createElementVNode as _createElementVNode,
createVNode as _createVNode,
Fragment as _Fragment,
openBlock as _openBlock,
createElementBlock as _createElementBlock,
} from "vue";
function render(_ctx, _cache, $props, $setup, $data, $options) {
return (
_openBlock(),
_createElementBlock(
_Fragment,
null,
[
_createElementVNode("h1", null, _toDisplayString($setup.title)),
_createElementVNode(
"h1",
null,
_toDisplayString($setup.msg),
1 /* TEXT */
),
_createVNode($setup["Child"]),
],
64 /* STABLE_FRAGMENT */
)
);
}
__sfc__.render = render;
export default __sfc__;
我们可以看到index.vue编译后的代码中已经没有了template标签和script标签，取而代之是render函数和__sfc__对象。并且使用__sfc__.render = render将render函数挂到__sfc__对象上，然后将__sfc__对象export default出去。
看到这里你应该知道了其实一个vue组件就是一个普通的js对象，import一个vue组件，实际就是import这个js对象。这个js对象中包含render方法和setup方法。
编译后的setup方法
我们先来看看这个setup方法，是不是觉得和我们源代码中的setup语法糖中的代码很相似？没错，这个setup方法内的代码就是由setup语法糖中的代码编译后来的。
setup语法糖原始代码
<script lang="ts" setup>
import { ref } from "vue";
import Child from "./child.vue";
const msg = ref("Hello World!");
const title = "title";
if (msg.value) {
const content = "content";
console.log(content);
}
</script>
setup编译后的代码
import { ref } from "vue";
import Child from "./Child.vue";
const title = "title";
const __sfc__ = {
__name: "index",
setup() {
const msg = ref("Hello World!");
if (msg.value) {
const content = "content";
console.log(content);
}
const __returned__ = { title, msg, Child };
return __returned__;
},
};
经过分析我们发现title变量由于不是响应式变量，所以编译后title变量被提到了js文件的全局变量上面去了。而msg变量是响应式变量，所以依然还是在setup方法中。我们再来看看setup的返回值，返回值是一个对象，对象中包含title、msg、Child属性，非setup顶层中定义的content变量就不在返回值对象中。
看到这里，可以回答我们前面提的第一个问题。
setup语法糖经过编译后是什么样子的？
setup语法糖编译后会变成一个setup方法，编译后setup方法中的代码和script标签中的源代码很相似。方法会返回一个对象，对象由setup中定义的顶层变量和import导入的内容组成。
由template编译后的render函数
我们先来看看原本template中的代码：
<template>
<h1>{{ title }}</h1>
<h1>{{ msg }}</h1>
<Child />
</template>
我们再来看看由template编译成的render函数：
import {
toDisplayString as _toDisplayString,
createElementVNode as _createElementVNode,
createVNode as _createVNode,
Fragment as _Fragment,
openBlock as _openBlock,
createElementBlock as _createElementBlock,
} from "vue";
function render(_ctx, _cache, $props, $setup, $data, $options) {
return (
_openBlock(),
_createElementBlock(
_Fragment,
null,
[
_createElementVNode("h1", null, _toDisplayString($setup.title)),
_createElementVNode(
"h1",
null,
_toDisplayString($setup.msg),
1 /* TEXT */
),
_createVNode($setup["Child"]),
],
64 /* STABLE_FRAGMENT */
)
);
}
我们这次主要看在render函数中如何访问setup中定义的顶层变量title、msg，createElementBlock和createElementVNode等创建虚拟DOM的函数不在这篇文章的讨论范围内。你只需要知道createElementVNode("h1", null, _toDisplayString($setup.title))为创建一个h1标签的虚拟DOM就行了。
在render函数中我们发现读取title变量的值是通过$setup.title读取到的，读取msg变量的值是通过$setup.msg读取到的。这个$setup对象就是调用render函数时传入的第四个变量，我想你应该猜出来了，这个$setup对象就是我们前面的setup方法返回的对象。
那么问题来了，在执行render函数的时候是如何将setup方法的返回值作为第四个变量传递给render函数的呢？我在下一节会一步一步的带你通过debug源码的方式去搞清楚这个问题，我们带着问题去debug源码其实非常简单。
debug源码搞清楚是如何调用render函数
有的小伙伴看到这里需要看源码就觉得头大了，别着急，其实很简单，我会一步一步的带着你去debug源码。
首先我们将Enable JavaScript source maps给取消勾选了，不然在debug源码的时候断点就会走到vue文件中，而不是走到编译会的js文件中。
然后我们需要在设置里面的Ignore List看看node_modules文件夹是否被忽略。新版谷歌浏览器中会默认排除掉node_modules文件夹，所以我们需要将这个取消勾选。如果忽略了node_modules文件夹，那么debug的时候断点就不会走到node_modules中vue的源码中去了。
接下来我们需要在浏览器中找到vue文件编译后的js代码，我们只需要在network面板中找到这个vue文件的http请求，然后在Response下右键选择Open in Sources panel，就会自动在sources面板自动打开对应编译后的js文件代码。
找到编译后的js文件，我们想debug看看是如何调用render函数的，所以我们给render函数加一个断点。然后刷新页面，发现代码已经走到了断点的地方。我们再来看看右边的Call Stack调用栈，发现render函数是由一个vue源码中的renderComponentRoot函数调用的。
点击Call Stack中的renderComponentRoot函数就可以跳转到renderComponentRoot函数的源码，我们发现renderComponentRoot函数中调用render函数的代码主要是下面这样的：
function renderComponentRoot(instance) {
const {
props,
data,
setupState,
// 省略...
} = instance;
render2.call(
thisProxy,
proxyToUse,
renderCache,
props,
setupState,
data,
ctx
)
}
这里我们可以看到前面的$setup实际就是由setupState赋值的，而setupState是当前vue实例上面的一个属性。那么setupState属性是如何被赋值到vue实例上面的呢？
我们需要给setup函数加一个断点，然后刷新页面进入断点。通过分析Call Stack调用栈，我们发现setup函数是由vue中的一个setupStatefulComponent函数调用执行的。
点击Call Stack调用栈中的setupStatefulComponent，进入到setupStatefulComponent的源码。我们看到setupStatefulComponent中的代码主要是这样的：
function setupStatefulComponent(instance) {
const { setup } = Component;
// 省略
const setupResult = callWithErrorHandling(
setup,
instance
);
handleSetupResult(instance, setupResult);
}
setup函数是Component上面的一个属性，我们将鼠标放到Component上面，看看这个Component是什么东西？
看到这个Component对象中既有render方法也有setup方法是不是感觉很熟悉，没错这个Component对象实际就是我们的vue文件编译后的js对象。
const __sfc__ = {
__name: "index",
setup() {
const msg = ref("Hello World!");
if (msg.value) {
const content = "content";
console.log(content);
}
const __returned__ = { title, msg, Child };
return __returned__;
},
};
__sfc__.render = render;
从Component对象中拿到setup函数，然后执行setup函数得到setupResult对象。然后再调用handleSetupResult(instance, setupResult);
我们再来看看handleSetupResult函数是什么样的，下面是我简化后的代码：
function handleSetupResult(instance, setupResult) {
if (isFunction(setupResult)) {
// 省略
} else if (isObject(setupResult)) {
instance.setupState = proxyRefs(setupResult);
}
}
我们的setup的返回值是一个对象，所以这里会执行instance.setupState = proxyRefs(setupResult)，将setup执行会的返回值赋值到vue实例的setupState属性上。
看到这里我们整个流程已经可以串起来了，首先会执行由setup语法糖编译后的setup函数。然后将setup函数中由顶层变量和import导入组成的返回值对象赋值给vue实例的setupState属性，然后执行render函数的时候从vue实例中取出setupState属性也就是setup的返回值。这样在render函数也就是template模版就可以访问到setup中的顶层变量和import导入。
现在我们可以回答前面提的另外两个问题了：
为什么在setup顶层定义的变量可以在template中可以直接使用？
因为在setup语法糖顶层定义的变量经过编译后会被加入到setup函数返回值对象__returned__中，而非setup顶层定义的变量不会加入到__returned__对象中。setup函数返回值会被塞到vue实例的setupState属性上，执行render函数的时候会将vue实例上的setupState属性传递给render函数，所以在render函数中就可以访问到setup顶层定义的变量和import导入。而render函数实际就是由template编译得来的，所以说在template中可以访问到setup顶层定义的变量和import导入。。
为什么import一个组件后就可以直接使用，无需使用components 选项来显式注册组件？
因为在setup语法糖中import导入的组件对象经过编译后同样也会被加入到setup函数返回值对象__returned__中，同理在template中也可以访问到setup的返回值对象，也就可以直接使用这个导入的组件了。
setup语法糖经过编译后就变成了setup函数，而setup函数的返回值是一个对象，这个对象就是由在setup顶层定义的变量和import导入组成的。vue在初始化的时候会执行setup函数，然后将setup函数返回值塞到vue实例的setupState属性上。执行render函数的时候会将vue实例上的setupState属性（也就是setup函数的返回值）传递给render函数，所以在render函数中就可以访问到setup顶层定义的变量和import导入。而render函数实际就是由template编译得来的，所以说在template中就可以访问到setup顶层定义的变量和import导入。