mirror of
https://github.com/yannh/kubeconform.git
synced 2026-02-17 08:57:02 +00:00
do not buffer all of stdin before starting to process resources
This commit is contained in:
parent
4b16128b75
commit
d64a376779
1 changed files with 24 additions and 9 deletions
|
|
@ -1,12 +1,30 @@
|
||||||
package resource
|
package resource
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bufio"
|
||||||
"context"
|
"context"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func yamlSplit(data []byte, atEOF bool) (advance int, token []byte, err error) {
|
||||||
|
// Return nothing if at end of file and no data passed
|
||||||
|
if atEOF && len(data) == 0 {
|
||||||
|
return 0, nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if i := strings.Index(string(data), "---\n"); i >= 0 {
|
||||||
|
return i + 4, data[0:i], nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// If at end of file with data return the data
|
||||||
|
if atEOF {
|
||||||
|
return len(data), data, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
// FromStream reads resources from a byte stream, usually here stdin
|
// FromStream reads resources from a byte stream, usually here stdin
|
||||||
func FromStream(ctx context.Context, path string, r io.Reader) (<-chan Resource, <-chan error) {
|
func FromStream(ctx context.Context, path string, r io.Reader) (<-chan Resource, <-chan error) {
|
||||||
resources := make(chan Resource)
|
resources := make(chan Resource)
|
||||||
|
|
@ -19,17 +37,14 @@ func FromStream(ctx context.Context, path string, r io.Reader) (<-chan Resource,
|
||||||
}()
|
}()
|
||||||
|
|
||||||
go func() {
|
go func() {
|
||||||
data, err := ioutil.ReadAll(r)
|
scanner := bufio.NewScanner(r)
|
||||||
if err != nil {
|
scanner.Split(yamlSplit)
|
||||||
errors <- DiscoveryError{path, err}
|
|
||||||
}
|
|
||||||
|
|
||||||
rawResources := bytes.Split(data, []byte("---\n"))
|
for res := scanner.Scan(); res != false; res = scanner.Scan() {
|
||||||
for _, rawResource := range rawResources {
|
|
||||||
if stop == true {
|
if stop == true {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
resources <- Resource{Path: path, Bytes: rawResource}
|
resources <- Resource{Path: path, Bytes: scanner.Bytes()}
|
||||||
}
|
}
|
||||||
|
|
||||||
close(resources)
|
close(resources)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue