goldsmith/context.go

97 lines
2.4 KiB
Go
Raw Normal View History

2015-12-18 08:03:30 +00:00
/*
* Copyright (c) 2015 Alex Yatskov <alex@foosoft.net>
* Author: Alex Yatskov <alex@foosoft.net>
*
* Permission is hereby granted, free of charge, to any person obtaining a copy of
* this software and associated documentation files (the "Software"), to deal in
* the Software without restriction, including without limitation the rights to
* use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
* the Software, and to permit persons to whom the Software is furnished to do so,
* subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package goldsmith
import (
2015-12-20 14:18:58 +00:00
"os"
2015-12-20 08:38:53 +00:00
"runtime"
2015-12-18 08:03:30 +00:00
"sync"
)
2015-12-20 08:58:50 +00:00
type context struct {
2015-12-18 08:03:30 +00:00
gs *goldsmith
2016-01-10 11:17:35 +00:00
plug Plugin
2015-12-18 08:03:30 +00:00
input, output chan *file
}
2016-01-10 11:17:35 +00:00
func (ctx *context) chain() {
2015-12-20 08:58:50 +00:00
defer close(ctx.output)
2015-12-18 08:03:30 +00:00
2016-01-10 11:17:35 +00:00
init, _ := ctx.plug.(Initializer)
accept, _ := ctx.plug.(Accepter)
proc, _ := ctx.plug.(Processor)
fin, _ := ctx.plug.(Finalizer)
2015-12-18 08:03:30 +00:00
2015-12-20 08:38:53 +00:00
if init != nil {
2015-12-20 08:58:50 +00:00
if err := init.Initialize(ctx); err != nil {
ctx.gs.fault(nil, err)
2015-12-20 08:38:53 +00:00
return
2015-12-18 08:03:30 +00:00
}
}
2016-01-10 11:17:35 +00:00
if ctx.input != nil {
var wg sync.WaitGroup
for i := 0; i < runtime.NumCPU(); i++ {
wg.Add(1)
go func() {
defer wg.Done()
for f := range ctx.input {
if proc == nil || accept != nil && !accept.Accept(ctx, f) {
ctx.output <- f
} else {
if _, err := f.Seek(0, os.SEEK_SET); err != nil {
ctx.gs.fault(f, err)
}
if err := proc.Process(ctx, f); err != nil {
ctx.gs.fault(f, err)
}
2015-12-20 08:38:53 +00:00
}
2015-12-18 08:03:30 +00:00
}
2016-01-10 11:17:35 +00:00
}()
}
wg.Wait()
2015-12-18 08:03:30 +00:00
}
if fin != nil {
2015-12-20 08:58:50 +00:00
if err := fin.Finalize(ctx); err != nil {
ctx.gs.fault(nil, err)
2015-12-18 08:03:30 +00:00
}
2015-12-19 11:51:23 +00:00
}
2015-12-18 08:03:30 +00:00
}
//
// Context Implementation
//
2015-12-20 08:58:50 +00:00
func (ctx *context) DispatchFile(f File) {
ctx.output <- f.(*file)
2015-12-18 08:03:30 +00:00
}
2015-12-20 08:58:50 +00:00
func (ctx *context) SrcDir() string {
return ctx.gs.srcDir
2015-12-18 08:03:30 +00:00
}
2015-12-20 08:58:50 +00:00
func (ctx *context) DstDir() string {
return ctx.gs.dstDir
2015-12-18 08:03:30 +00:00
}