Improved API
This commit is contained in:
parent
3413e5a45f
commit
fea7e02623
29
context.go
29
context.go
@ -40,35 +40,40 @@ type context struct {
|
|||||||
func (ctx *context) chain() {
|
func (ctx *context) chain() {
|
||||||
defer close(ctx.output)
|
defer close(ctx.output)
|
||||||
|
|
||||||
initializer, _ := ctx.plug.(Initializer)
|
var filters []string
|
||||||
accepter, _ := ctx.plug.(Accepter)
|
if initializer, ok := ctx.plug.(Initializer); ok {
|
||||||
processor, _ := ctx.plug.(Processor)
|
var err error
|
||||||
finalizer, _ := ctx.plug.(Finalizer)
|
filters, err = initializer.Initialize(ctx)
|
||||||
|
if err != nil {
|
||||||
if initializer != nil {
|
|
||||||
if err := initializer.Initialize(ctx); err != nil {
|
|
||||||
ctx.gs.fault(nil, err)
|
ctx.gs.fault(nil, err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if ctx.input != nil {
|
if ctx.input != nil {
|
||||||
|
processor, _ := ctx.plug.(Processor)
|
||||||
|
|
||||||
var wg sync.WaitGroup
|
var wg sync.WaitGroup
|
||||||
for i := 0; i < runtime.NumCPU(); i++ {
|
for i := 0; i < runtime.NumCPU(); i++ {
|
||||||
wg.Add(1)
|
wg.Add(1)
|
||||||
go func() {
|
go func() {
|
||||||
defer wg.Done()
|
defer wg.Done()
|
||||||
for f := range ctx.input {
|
for f := range ctx.input {
|
||||||
accept := processor != nil && (accepter == nil || accepter.Accept(ctx, f))
|
accept := processor != nil
|
||||||
if accept && len(ctx.filters) > 0 {
|
matcher := func(patterns []string) {
|
||||||
|
if accept && len(patterns) > 0 {
|
||||||
accept = false
|
accept = false
|
||||||
for _, filter := range ctx.filters {
|
for _, pattern := range patterns {
|
||||||
if match, _ := doublestar.PathMatch(filter, f.Path()); match {
|
if match, _ := doublestar.PathMatch(pattern, f.Path()); match {
|
||||||
accept = true
|
accept = true
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
matcher(filters)
|
||||||
|
matcher(ctx.filters)
|
||||||
|
|
||||||
if accept {
|
if accept {
|
||||||
if _, err := f.Seek(0, os.SEEK_SET); err != nil {
|
if _, err := f.Seek(0, os.SEEK_SET); err != nil {
|
||||||
@ -86,7 +91,7 @@ func (ctx *context) chain() {
|
|||||||
wg.Wait()
|
wg.Wait()
|
||||||
}
|
}
|
||||||
|
|
||||||
if finalizer != nil {
|
if finalizer, ok := ctx.plug.(Finalizer); ok {
|
||||||
if err := finalizer.Finalize(ctx); err != nil {
|
if err := finalizer.Finalize(ctx); err != nil {
|
||||||
ctx.gs.fault(nil, err)
|
ctx.gs.fault(nil, err)
|
||||||
}
|
}
|
||||||
|
@ -106,11 +106,7 @@ func (e Error) Error() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type Initializer interface {
|
type Initializer interface {
|
||||||
Initialize(ctx Context) error
|
Initialize(ctx Context) ([]string, error)
|
||||||
}
|
|
||||||
|
|
||||||
type Accepter interface {
|
|
||||||
Accept(ctx Context, f File) bool
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type Processor interface {
|
type Processor interface {
|
||||||
|
@ -26,7 +26,7 @@ import "path/filepath"
|
|||||||
|
|
||||||
type loader struct{}
|
type loader struct{}
|
||||||
|
|
||||||
func (*loader) Initialize(ctx Context) error {
|
func (*loader) Initialize(ctx Context) ([]string, error) {
|
||||||
infos := make(chan fileInfo)
|
infos := make(chan fileInfo)
|
||||||
go scanDir(ctx.SrcDir(), infos)
|
go scanDir(ctx.SrcDir(), infos)
|
||||||
|
|
||||||
@ -48,5 +48,5 @@ func (*loader) Initialize(ctx Context) error {
|
|||||||
ctx.DispatchFile(f)
|
ctx.DispatchFile(f)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil, nil
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user