hugolib: Prepare render per output format

See #3220
This commit is contained in:
Bjørn Erik Pedersen 2017-05-05 09:24:37 +02:00
parent 1e4d082cf5
commit 2bcbf10400
4 changed files with 50 additions and 34 deletions

View file

@ -492,7 +492,13 @@ func (h *HugoSites) setupTranslations() {
}
}
func (s *Site) preparePagesForRender(cfg *BuildCfg) {
func (s *Site) preparePagesForRender(outFormatIdx int, cfg *BuildCfg) {
if outFormatIdx > 0 {
// TODO(bep) for now
return
}
pageChan := make(chan *Page)
wg := &sync.WaitGroup{}
numWorkers := getGoMaxProcs() * 4

View file

@ -203,26 +203,31 @@ func (h *HugoSites) assemble(config *BuildCfg) error {
return err
}
for _, s := range h.Sites {
s.preparePagesForRender(config)
}
return nil
}
func (h *HugoSites) render(config *BuildCfg) error {
if !config.SkipRender {
for _, s := range h.Sites {
if err := s.render(); err != nil {
s.initRenderFormats()
for i, rf := range s.renderFormats {
s.rc = &siteRenderingContext{Format: rf}
s.preparePagesForRender(i, config)
if !config.SkipRender {
if err := s.render(i); err != nil {
return err
}
}
}
if config.PrintStats {
if !config.SkipRender && config.PrintStats {
s.Stats()
}
}
if !config.SkipRender {
if err := h.renderCrossSitesArtifacts(); err != nil {
return err
}

View file

@ -309,6 +309,7 @@ func executeShortcodeFuncMap(funcs map[string]func() (string, error)) (map[strin
}
func renderShortcodes(shortcodes map[string]shortcode, p *Page) map[string]func() (string, error) {
renderedShortcodes := make(map[string]func() (string, error))
for key, sc := range shortcodes {
@ -316,8 +317,8 @@ func renderShortcodes(shortcodes map[string]shortcode, p *Page) map[string]func(
// need to have something to replace with
renderedShortcodes[key] = emptyShortcodeFn
} else {
shorctode := sc
renderedShortcodes[key] = func() (string, error) { return renderShortcode(shorctode, nil, p), nil }
shortcode := sc
renderedShortcodes[key] = func() (string, error) { return renderShortcode(shortcode, nil, p), nil }
}
}

View file

@ -986,8 +986,9 @@ func (s *Site) setupSitePages() {
s.Info.LastChange = siteLastChange
}
func (s *Site) render() (err error) {
func (s *Site) render(outFormatIdx int) (err error) {
if outFormatIdx == 0 {
if err = s.preparePages(); err != nil {
return
}
@ -1002,16 +1003,19 @@ func (s *Site) render() (err error) {
}
s.timerStep("render and write aliases")
// TODO(bep) render consider this, ref. render404 etc.
s.initRenderFormats()
for _, rf := range s.renderFormats {
s.rc = &siteRenderingContext{Format: rf}
}
if err = s.renderPages(); err != nil {
return
}
}
s.timerStep("render and write pages")
// TODO(bep) render consider this, ref. render404 etc.
if outFormatIdx > 0 {
return
}
if err = s.renderSitemap(); err != nil {
return
}