goldsmith/context.go

110 lines
2.7 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
2018-01-07 23:42:32 +00:00
filters []Filter
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
2018-01-07 23:42:32 +00:00
var err error
var filters []Filter
2016-07-10 20:05:23 +00:00
if initializer, ok := ctx.plug.(Initializer); ok {
filters, err = initializer.Initialize(ctx)
if err != nil {
2016-08-21 19:54:44 +00:00
ctx.gs.fault(ctx.plug.Name(), 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 {
2016-07-10 20:05:23 +00:00
processor, _ := ctx.plug.(Processor)
2016-01-10 11:17:35 +00:00
var wg sync.WaitGroup
for i := 0; i < runtime.NumCPU(); i++ {
wg.Add(1)
go func() {
defer wg.Done()
for f := range ctx.input {
2016-07-10 20:05:23 +00:00
accept := processor != nil
2018-01-07 23:42:32 +00:00
for _, filter := range append(ctx.filters, filters...) {
if accept, err = filter.Accept(ctx, f); err != nil {
ctx.gs.fault(filter.Name(), f, err)
return
}
2018-01-07 23:42:32 +00:00
if !accept {
break
}
}
2016-07-10 20:05:23 +00:00
if accept {
2016-01-10 11:17:35 +00:00
if _, err := f.Seek(0, os.SEEK_SET); err != nil {
2016-08-21 19:54:44 +00:00
ctx.gs.fault("core", f, err)
2016-01-10 11:17:35 +00:00
}
if err := processor.Process(ctx, f); err != nil {
2016-08-21 19:54:44 +00:00
ctx.gs.fault(ctx.plug.Name(), f, err)
2016-01-10 11:17:35 +00:00
}
} else {
ctx.output <- f
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
}
2016-07-10 20:05:23 +00:00
if finalizer, ok := ctx.plug.(Finalizer); ok {
if err := finalizer.Finalize(ctx); err != nil {
2016-08-21 19:54:44 +00:00
ctx.gs.fault(ctx.plug.Name(), 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
}