From 0a56f2af4e1969e76e94fdfb56d1bbed0e685625 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bj=C3=B8rn=20Erik=20Pedersen?= Date: Mon, 14 Mar 2022 16:02:04 +0100 Subject: [PATCH] Revert "Revert "Allow rendering static files to disk and dynamic to memory in server mode"" This reverts commit 64b7b7a89753a39661219b2fcb92d7f185a03f63. --- commands/commandeer.go | 14 +++++++++++++- commands/hugo.go | 3 +++ commands/server.go | 23 ++++++++++++++--------- commands/static_syncer.go | 9 ++++++++- hugofs/fs.go | 12 ++++++++---- hugolib/filesystems/basefs.go | 13 +++++++++---- hugolib/pages_process.go | 16 ++++++++++++---- hugolib/site.go | 4 ++-- 8 files changed, 69 insertions(+), 25 deletions(-) diff --git a/commands/commandeer.go b/commands/commandeer.go index 8a19258931e..6f74bf8640d 100644 --- a/commands/commandeer.go +++ b/commands/commandeer.go @@ -94,6 +94,7 @@ type commandeer struct { languagesConfigured bool languages langs.Languages doLiveReload bool + renderStaticToDisk bool fastRenderMode bool showErrorInBrowser bool wasError bool @@ -375,8 +376,9 @@ func (c *commandeer) loadConfig() error { } createMemFs := config.GetBool("renderToMemory") + c.renderStaticToDisk = config.GetBool("renderStaticToDisk") - if createMemFs { + if createMemFs && !c.renderStaticToDisk { // Rendering to memoryFS, publish to Root regardless of publishDir. config.Set("publishDir", "/") } @@ -387,6 +389,14 @@ func (c *commandeer) loadConfig() error { if c.destinationFs != nil { // Need to reuse the destination on server rebuilds. fs.Destination = c.destinationFs + } else if createMemFs && c.renderStaticToDisk { + // Writes the dynamic output on memory, + // while serve others directly from publishDir + publishDir := config.GetString("publishDir") + writableFs := afero.NewBasePathFs(afero.NewMemMapFs(), publishDir) + publicFs := afero.NewOsFs() + fs.Destination = afero.NewCopyOnWriteFs(afero.NewReadOnlyFs(publicFs), writableFs) + fs.DestinationStatic = publicFs } else if createMemFs { // Hugo writes the output to memory instead of the disk. fs.Destination = new(afero.MemMapFs) @@ -404,11 +414,13 @@ func (c *commandeer) loadConfig() error { changeDetector.PrepareNew() fs.Destination = hugofs.NewHashingFs(fs.Destination, changeDetector) + fs.DestinationStatic = hugofs.NewHashingFs(fs.DestinationStatic, changeDetector) c.changeDetector = changeDetector } if c.Cfg.GetBool("logPathWarnings") { fs.Destination = hugofs.NewCreateCountingFs(fs.Destination) + fs.DestinationStatic = hugofs.NewCreateCountingFs(fs.DestinationStatic) } // To debug hard-to-find path issues. diff --git a/commands/hugo.go b/commands/hugo.go index 8c5294f006d..21140fa4307 100644 --- a/commands/hugo.go +++ b/commands/hugo.go @@ -652,6 +652,9 @@ func (c *commandeer) copyStaticTo(sourceFs *filesystems.SourceFilesystem) (uint6 syncer.ChmodFilter = chmodFilter syncer.SrcFs = fs syncer.DestFs = c.Fs.Destination + if c.renderStaticToDisk { + syncer.DestFs = c.Fs.DestinationStatic + } // Now that we are using a unionFs for the static directories // We can effectively clean the publishDir on initial sync syncer.Delete = c.Cfg.GetBool("cleanDestinationDir") diff --git a/commands/server.go b/commands/server.go index bb6a4e15d45..23301d004f6 100644 --- a/commands/server.go +++ b/commands/server.go @@ -50,15 +50,16 @@ type serverCmd struct { // Can be used to stop the server. Useful in tests stop chan bool - disableLiveReload bool - navigateToChanged bool - renderToDisk bool - serverAppend bool - serverInterface string - serverPort int - liveReloadPort int - serverWatch bool - noHTTPCache bool + disableLiveReload bool + navigateToChanged bool + renderToDisk bool + renderStaticToDisk bool + serverAppend bool + serverInterface string + serverPort int + liveReloadPort int + serverWatch bool + noHTTPCache bool disableFastRender bool disableBrowserError bool @@ -109,6 +110,7 @@ of a second, you will be able to save and see your changes nearly instantly.`, cc.cmd.Flags().BoolVar(&cc.renderToDisk, "renderToDisk", false, "render to Destination path (default is render to memory & serve from there)") cc.cmd.Flags().BoolVar(&cc.disableFastRender, "disableFastRender", false, "enables full re-renders on changes") cc.cmd.Flags().BoolVar(&cc.disableBrowserError, "disableBrowserError", false, "do not show build errors in the browser") + cc.cmd.Flags().BoolVar(&cc.renderStaticToDisk, "renderStaticToDisk", false, "render static files to disk but dynamic files render to memory.") cc.cmd.Flags().String("memstats", "", "log memory usage to this file") cc.cmd.Flags().String("meminterval", "100ms", "interval to poll memory usage (requires --memstats), valid time units are \"ns\", \"us\" (or \"µs\"), \"ms\", \"s\", \"m\", \"h\".") @@ -147,6 +149,7 @@ func (sc *serverCmd) server(cmd *cobra.Command, args []string) error { cfgInit := func(c *commandeer) (rerr error) { c.Set("renderToMemory", !sc.renderToDisk) + c.Set("renderStaticToDisk", sc.renderStaticToDisk) if cmd.Flags().Changed("navigateToChanged") { c.Set("navigateToChanged", sc.navigateToChanged) } @@ -340,6 +343,8 @@ func (f *fileServer) createEndpoint(i int) (*http.ServeMux, net.Listener, string if i == 0 { if f.s.renderToDisk { jww.FEEDBACK.Println("Serving pages from " + absPublishDir) + } else if f.s.renderStaticToDisk { + jww.FEEDBACK.Println("Serving pages from memory and static files from " + absPublishDir) } else { jww.FEEDBACK.Println("Serving pages from memory") } diff --git a/commands/static_syncer.go b/commands/static_syncer.go index 5569d4de688..2eb2b666233 100644 --- a/commands/static_syncer.go +++ b/commands/static_syncer.go @@ -56,6 +56,9 @@ func (s *staticSyncer) syncsStaticEvents(staticEvents []fsnotify.Event) error { syncer.ChmodFilter = chmodFilter syncer.SrcFs = sourceFs.Fs syncer.DestFs = c.Fs.Destination + if c.renderStaticToDisk { + syncer.DestFs = c.Fs.DestinationStatic + } // prevent spamming the log on changes logger := helpers.NewDistinctErrorLogger() @@ -101,7 +104,11 @@ func (s *staticSyncer) syncsStaticEvents(staticEvents []fsnotify.Event) error { toRemove := filepath.Join(publishDir, relPath) logger.Println("File no longer exists in static dir, removing", toRemove) - _ = c.Fs.Destination.RemoveAll(toRemove) + if c.renderStaticToDisk { + _ = c.Fs.DestinationStatic.RemoveAll(toRemove) + } else { + _ = c.Fs.Destination.RemoveAll(toRemove) + } } else if err == nil { // If file still exists, sync it logger.Println("Syncing", relPath, "to", publishDir) diff --git a/hugofs/fs.go b/hugofs/fs.go index 54d962553e7..95645204e9f 100644 --- a/hugofs/fs.go +++ b/hugofs/fs.go @@ -35,6 +35,9 @@ type Fs struct { // Destination is Hugo's destination file system. Destination afero.Fs + // Destination used for `renderStaticToDisk` + DestinationStatic afero.Fs + // Os is an OS file system. // NOTE: Field is currently unused. Os afero.Fs @@ -69,10 +72,11 @@ func NewFrom(fs afero.Fs, cfg config.Provider) *Fs { func newFs(base afero.Fs, cfg config.Provider) *Fs { return &Fs{ - Source: base, - Destination: base, - Os: &afero.OsFs{}, - WorkingDir: getWorkingDirFs(base, cfg), + Source: base, + Destination: base, + DestinationStatic: base, + Os: &afero.OsFs{}, + WorkingDir: getWorkingDirFs(base, cfg), } } diff --git a/hugolib/filesystems/basefs.go b/hugolib/filesystems/basefs.go index 6e3f88a4be1..0290d2e1cf3 100644 --- a/hugolib/filesystems/basefs.go +++ b/hugolib/filesystems/basefs.go @@ -71,6 +71,9 @@ type BaseFs struct { // A read-only filesystem starting from the project workDir. WorkDir afero.Fs + // The filesystem used for renderStaticToDisk. + PublishFsStatic afero.Fs + theBigFs *filesystemsCollector // Locks. @@ -438,15 +441,17 @@ func NewBase(p *paths.Paths, logger loggers.Logger, options ...func(*BaseFs) err publishFs := hugofs.NewBaseFileDecorator(afero.NewBasePathFs(fs.Destination, p.AbsPublishDir)) sourceFs := hugofs.NewBaseFileDecorator(afero.NewBasePathFs(fs.Source, p.WorkingDir)) + publishFsStatic := afero.NewBasePathFs(fs.Source, p.AbsPublishDir) // Same as sourceFs, but no decoration. This is what's used by os.ReadDir etc. workDir := afero.NewBasePathFs(afero.NewReadOnlyFs(fs.Source), p.WorkingDir) b := &BaseFs{ - SourceFs: sourceFs, - WorkDir: workDir, - PublishFs: publishFs, - buildMu: lockedfile.MutexAt(filepath.Join(p.WorkingDir, lockFileBuild)), + SourceFs: sourceFs, + WorkDir: workDir, + PublishFs: publishFs, + PublishFsStatic: publishFsStatic, + buildMu: lockedfile.MutexAt(filepath.Join(p.WorkingDir, lockFileBuild)), } for _, opt := range options { diff --git a/hugolib/pages_process.go b/hugolib/pages_process.go index d33f70d8e67..47687eaad8a 100644 --- a/hugolib/pages_process.go +++ b/hugolib/pages_process.go @@ -33,9 +33,10 @@ func newPagesProcessor(h *HugoSites, sp *source.SourceSpec) *pagesProcessor { procs := make(map[string]pagesCollectorProcessorProvider) for _, s := range h.Sites { procs[s.Lang()] = &sitePagesProcessor{ - m: s.pageMap, - errorSender: s.h, - itemChan: make(chan any, config.GetNumWorkerMultiplier()*2), + m: s.pageMap, + errorSender: s.h, + itemChan: make(chan interface{}, config.GetNumWorkerMultiplier()*2), + renderStaticToDisk: h.Cfg.GetBool("renderStaticToDisk"), } } return &pagesProcessor{ @@ -118,6 +119,8 @@ type sitePagesProcessor struct { ctx context.Context itemChan chan any itemGroup *errgroup.Group + + renderStaticToDisk bool } func (p *sitePagesProcessor) Process(item any) error { @@ -162,7 +165,12 @@ func (p *sitePagesProcessor) copyFile(fim hugofs.FileMetaInfo) error { defer f.Close() - return s.publish(&s.PathSpec.ProcessingStats.Files, target, f) + fs := s.PublishFs + if p.renderStaticToDisk { + fs = s.PublishFsStatic + } + + return s.publish(&s.PathSpec.ProcessingStats.Files, target, f, fs) } func (p *sitePagesProcessor) doProcess(item any) error { diff --git a/hugolib/site.go b/hugolib/site.go index efa9368306a..d7b5cb64e55 100644 --- a/hugolib/site.go +++ b/hugolib/site.go @@ -1829,10 +1829,10 @@ func (s *Site) lookupTemplate(layouts ...string) (tpl.Template, bool) { return nil, false } -func (s *Site) publish(statCounter *uint64, path string, r io.Reader) (err error) { +func (s *Site) publish(statCounter *uint64, path string, r io.Reader, fs afero.Fs) (err error) { s.PathSpec.ProcessingStats.Incr(statCounter) - return helpers.WriteToDisk(filepath.Clean(path), r, s.BaseFs.PublishFs) + return helpers.WriteToDisk(filepath.Clean(path), r, fs) } func (s *Site) kindFromFileInfoOrSections(fi *fileInfo, sections []string) string {