fix: add async-mutex to builds on large vaults
This commit is contained in:
		
							parent
							
								
									b99d4cd8ce
								
							
						
					
					
						commit
						e65ea48fae
					
				
					 5 changed files with 29 additions and 6 deletions
				
			
		| 
						 | 
				
			
			@ -16,6 +16,7 @@ import http from "http"
 | 
			
		|||
import serveHandler from "serve-handler"
 | 
			
		||||
import { WebSocketServer } from "ws"
 | 
			
		||||
import { randomUUID } from "crypto"
 | 
			
		||||
import { Mutex } from "async-mutex"
 | 
			
		||||
 | 
			
		||||
const ORIGIN_NAME = "origin"
 | 
			
		||||
const UPSTREAM_NAME = "upstream"
 | 
			
		||||
| 
						 | 
				
			
			@ -391,8 +392,10 @@ See the [documentation](https://quartz.jzhao.xyz) for how to get started.
 | 
			
		|||
      ],
 | 
			
		||||
    })
 | 
			
		||||
 | 
			
		||||
    const buildMutex = new Mutex()
 | 
			
		||||
    const timeoutIds = new Set()
 | 
			
		||||
    const build = async (clientRefresh) => {
 | 
			
		||||
      await buildMutex.acquire()
 | 
			
		||||
      const result = await ctx.rebuild().catch((err) => {
 | 
			
		||||
        console.error(`${chalk.red("Couldn't parse Quartz configuration:")} ${fp}`)
 | 
			
		||||
        console.log(`Reason: ${chalk.grey(err)}`)
 | 
			
		||||
| 
						 | 
				
			
			@ -415,6 +418,7 @@ See the [documentation](https://quartz.jzhao.xyz) for how to get started.
 | 
			
		|||
      const { default: buildQuartz } = await import(cacheFile + `?update=${randomUUID()}`)
 | 
			
		||||
      await buildQuartz(argv, clientRefresh)
 | 
			
		||||
      clientRefresh()
 | 
			
		||||
      buildMutex.release()
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    const rebuild = (clientRefresh) => {
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -16,6 +16,7 @@ import { Argv, BuildCtx } from "./util/ctx"
 | 
			
		|||
import { glob, toPosixPath } from "./util/glob"
 | 
			
		||||
import { trace } from "./util/trace"
 | 
			
		||||
import { options } from "./util/sourcemap"
 | 
			
		||||
import { Mutex } from "async-mutex"
 | 
			
		||||
 | 
			
		||||
async function buildQuartz(argv: Argv, clientRefresh: () => void) {
 | 
			
		||||
  const ctx: BuildCtx = {
 | 
			
		||||
| 
						 | 
				
			
			@ -77,10 +78,11 @@ async function startServing(
 | 
			
		|||
  }
 | 
			
		||||
 | 
			
		||||
  const initialSlugs = ctx.allSlugs
 | 
			
		||||
  let timeoutIds: Set<ReturnType<typeof setTimeout>> = new Set()
 | 
			
		||||
  let toRebuild: Set<FilePath> = new Set()
 | 
			
		||||
  let toRemove: Set<FilePath> = new Set()
 | 
			
		||||
  let trackedAssets: Set<FilePath> = new Set()
 | 
			
		||||
  const buildMutex = new Mutex()
 | 
			
		||||
  const timeoutIds: Set<ReturnType<typeof setTimeout>> = new Set()
 | 
			
		||||
  const toRebuild: Set<FilePath> = new Set()
 | 
			
		||||
  const toRemove: Set<FilePath> = new Set()
 | 
			
		||||
  const trackedAssets: Set<FilePath> = new Set()
 | 
			
		||||
  async function rebuild(fp: string, action: "add" | "change" | "delete") {
 | 
			
		||||
    // don't do anything for gitignored files
 | 
			
		||||
    if (ignored(fp)) {
 | 
			
		||||
| 
						 | 
				
			
			@ -111,6 +113,7 @@ async function startServing(
 | 
			
		|||
    // debounce rebuilds every 250ms
 | 
			
		||||
    timeoutIds.add(
 | 
			
		||||
      setTimeout(async () => {
 | 
			
		||||
        await buildMutex.acquire()
 | 
			
		||||
        const perf = new PerfTimer()
 | 
			
		||||
        console.log(chalk.yellow("Detected change, rebuilding..."))
 | 
			
		||||
        try {
 | 
			
		||||
| 
						 | 
				
			
			@ -143,6 +146,7 @@ async function startServing(
 | 
			
		|||
        clientRefresh()
 | 
			
		||||
        toRebuild.clear()
 | 
			
		||||
        toRemove.clear()
 | 
			
		||||
        buildMutex.release()
 | 
			
		||||
      }, 250),
 | 
			
		||||
    )
 | 
			
		||||
  }
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -25,13 +25,13 @@ export default ((userOpts?: Partial<Options>) => {
 | 
			
		|||
  const opts = { ...defaultOptions, ...userOpts }
 | 
			
		||||
  function RecentNotes(props: QuartzComponentProps) {
 | 
			
		||||
    const { allFiles, fileData, displayClass } = props
 | 
			
		||||
    const pages = allFiles.filter(opts.filter).sort(opts.sort).slice(0, opts.limit)
 | 
			
		||||
    const pages = allFiles.filter(opts.filter).sort(opts.sort)
 | 
			
		||||
    const remaining = Math.max(0, pages.length - opts.limit)
 | 
			
		||||
    return (
 | 
			
		||||
      <div class={`recent-notes ${displayClass}`}>
 | 
			
		||||
        <h3>{opts.title}</h3>
 | 
			
		||||
        <ul class="recent-ul">
 | 
			
		||||
          {pages.map((page) => {
 | 
			
		||||
          {pages.slice(0, opts.limit).map((page) => {
 | 
			
		||||
            const title = page.frontmatter?.title
 | 
			
		||||
            const tags = page.frontmatter?.tags ?? []
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue