Skip to content

构建时数据加载

VitePress 提供了一个叫做数据加载器的功能,它允许你加载任意数据并从页面或组件中导入它。数据加载只在构建时执行:最终的数据将被序列化为 JavaScript 包中的 JSON。

数据加载器可以被用于获取远程数据,也可以基于本地文件生成元数据。例如,你可以使用数据加载器来解析所有本地 API 页面并自动生成所有 API 入口的索引。

基本用法

一个数据加载器文件必须以 .data.js.data.ts 结尾。该文件应该提供一个默认导出的对象,该对象具有 load() 方法:

js
// example.data.js
export default {
  load() {
    return {
      data: 'hello'
    }
  }
}
// example.data.js
export default {
  load() {
    return {
      data: 'hello'
    }
  }
}

数据加载器模块只在 Node.js 中执行,因此你可以按需导入 Node API 和 npm 依赖。

然后,你可以在 .md 页面和 .vue 组件中使用 data 命名导出从该文件中导入数据:

html
<script setup>
import { data } from './example.data.js'
</script>

<pre>{{ data }}</pre>
<script setup>
import { data } from './example.data.js'
</script>

<pre>{{ data }}</pre>

输出:

json
{
  "data": "hello"
}
{
  "data": "hello"
}

你会注意到数据加载器本身并没有导出 data。这是因为 VitePress 在后台调用了 load() 方法,并通过名为 data 的命名导出隐式地暴露了结果。

即使 loader 是异步的,这也是有效的:

js
export default {
  async load() {
    // fetch remote data
    return (await fetch('...')).json()
  }
}
export default {
  async load() {
    // fetch remote data
    return (await fetch('...')).json()
  }
}

使用本地文件生成数据

当你需要基于本地文件生成数据时,你应该在数据加载器中使用 watch 选项,以便这些文件改动时可以触发热更新。

watch 选项也很方便,因为你可以使用 glob 模式 匹配多个文件。模式可以相对于加载器文件本身,load() 函数将接收匹配文件的绝对路径。

下面的例子展示了如何使用 csv-parse 加载 CSV 文件并将其转换为 JSON。因为此文件仅在构建时执行,因此不会将 CSV 解析器发送到客户端!

js
import fs from 'node:fs'
import { parse } from 'csv-parse/sync'

export default {
  watch: ['./data/*.csv'],
  load(watchedFiles) {
    // watchedFiles will be an array of absolute paths of the matched files.
    // generate an array of blog post metadata that can be used to render
    // a list in the theme layout
    return watchedFiles.map(file => {
      return parse(fs.readFileSync(file, 'utf-8'), {
        columns: true,
        skip_empty_lines: true
      })
    })
  }
}
import fs from 'node:fs'
import { parse } from 'csv-parse/sync'

export default {
  watch: ['./data/*.csv'],
  load(watchedFiles) {
    // watchedFiles will be an array of absolute paths of the matched files.
    // generate an array of blog post metadata that can be used to render
    // a list in the theme layout
    return watchedFiles.map(file => {
      return parse(fs.readFileSync(file, 'utf-8'), {
        columns: true,
        skip_empty_lines: true
      })
    })
  }
}

createContentLoader

当构建一个内容为主的网站时,我们经常需要创建一个“档案”或“索引”页面:一个我们可以列出内容中的所有可用条目的页面,例如博客文章或 API 页面。我们可以直接使用数据加载器 API 实现这一点,但由于这是一个常见的用例,VitePress 还提供了一个 createContentLoader 辅助函数来简化这个过程:

js
// posts.data.js
import { createContentLoader } from 'vitepress'

export default createContentLoader('posts/*.md', /* options */)
// posts.data.js
import { createContentLoader } from 'vitepress'

export default createContentLoader('posts/*.md', /* options */)

该辅助函数接受一个相对于 项目根目录 的 glob 模式,并返回一个 { watch, load } 数据加载器对象,该对象可以用作数据加载器文件中的默认导出。它还基于文件修改时间戳实现了缓存以提高开发性能。

请注意,加载器仅适用于 Markdown 文件 - 匹配的非 Markdown 文件将被跳过。

加载的数据将是一个类型为 ContentData[] 数组:

ts
interface ContentData {
  // mapped absolute URL for the page. e.g. /posts/hello.html
  url: string
  // frontmatter data of the page
  frontmatter: Record<string, any>

  // the following are only present if relevant options are enabled
  // we will discuss them below
  src: string | undefined
  html: string | undefined
  excerpt: string | undefined
}
interface ContentData {
  // mapped absolute URL for the page. e.g. /posts/hello.html
  url: string
  // frontmatter data of the page
  frontmatter: Record<string, any>

  // the following are only present if relevant options are enabled
  // we will discuss them below
  src: string | undefined
  html: string | undefined
  excerpt: string | undefined
}

默认情况下只提供 urlfrontmatter。这是因为加载的数据将作为 JSON 内联在客户端包中,我们需要谨慎考虑其大小。下面的例子展示了如何使用数据构建最小博客索引页面:

vue
<script setup>
import { data as posts } from './posts.data.js'
</script>

<template>
  <h1>All Blog Posts</h1>
  <ul>
    <li v-for="post of posts">
      <a :href="post.url">{{ post.frontmatter.title }}</a>
      <span>by {{ post.frontmatter.author }}</span>
    </li>
  </ul>
</template>
<script setup>
import { data as posts } from './posts.data.js'
</script>

<template>
  <h1>All Blog Posts</h1>
  <ul>
    <li v-for="post of posts">
      <a :href="post.url">{{ post.frontmatter.title }}</a>
      <span>by {{ post.frontmatter.author }}</span>
    </li>
  </ul>
</template>

Options

默认数据可能不适合所有需求 - 你可以选择使用选项转换数据:

js
// posts.data.js
import { createContentLoader } from 'vitepress'

export default createContentLoader('posts/*.md', {
  includeSrc: true, // include raw markdown source?
  render: true,     // include rendered full page HTML?
  excerpt: true,    // include excerpt?
  transform(rawData) {
    // map, sort, or filter the raw data as you wish.
    // the final result is what will be shipped to the client.
    return rawData.sort((a, b) => {
      return +new Date(b.frontmatter.date) - +new Date(a.frontmatter.date)
    }).map(page => {
      page.src  // raw markdown source
      page.html // rendered full page HTML
      page.excerpt // rendered excerpt HTML (content above first `---`)
      return {/* ... */}
    })
  }
})
// posts.data.js
import { createContentLoader } from 'vitepress'

export default createContentLoader('posts/*.md', {
  includeSrc: true, // include raw markdown source?
  render: true,     // include rendered full page HTML?
  excerpt: true,    // include excerpt?
  transform(rawData) {
    // map, sort, or filter the raw data as you wish.
    // the final result is what will be shipped to the client.
    return rawData.sort((a, b) => {
      return +new Date(b.frontmatter.date) - +new Date(a.frontmatter.date)
    }).map(page => {
      page.src  // raw markdown source
      page.html // rendered full page HTML
      page.excerpt // rendered excerpt HTML (content above first `---`)
      return {/* ... */}
    })
  }
})

查看它在 Vue.js 博客 中是如何使用的。

createContentLoader API 也可以在 构建钩子 中使用:

js
// .vitepress/config.js
export default {
  async buildEnd() {
    const posts = await createContentLoader('posts/*.md').load()
    // generate files based on posts metadata, e.g. RSS feed
  }
}
// .vitepress/config.js
export default {
  async buildEnd() {
    const posts = await createContentLoader('posts/*.md').load()
    // generate files based on posts metadata, e.g. RSS feed
  }
}

为数据加载器导出类型

当使用 TypeScript 时,你可以像这样为加载器和 data 导出类型:

ts
import { defineLoader } from 'vitepress'

export interface Data {
  // data type
}

declare const data: Data
export { data }

export default defineLoader({
  // type checked loader options
  glob: ['...'],
  async load(): Promise<Data> {
    // ...
  }
})
import { defineLoader } from 'vitepress'

export interface Data {
  // data type
}

declare const data: Data
export { data }

export default defineLoader({
  // type checked loader options
  glob: ['...'],
  async load(): Promise<Data> {
    // ...
  }
})

Released under the MIT License.