505cbd2591
* remove amazon from core * vendor amazon plugin * remove website content * Add amazon to docs-remote-plugins * update amazon reference links in the documentation * update amazon docs version to latest Co-authored-by: Adrien Delorme <adrien.delorme@icloud.com>
31 lines
779 B
Go
31 lines
779 B
Go
package textseg
|
|
|
|
import (
|
|
"bufio"
|
|
"bytes"
|
|
)
|
|
|
|
// AllTokens is a utility that uses a bufio.SplitFunc to produce a slice of
|
|
// all of the recognized tokens in the given buffer.
|
|
func AllTokens(buf []byte, splitFunc bufio.SplitFunc) ([][]byte, error) {
|
|
scanner := bufio.NewScanner(bytes.NewReader(buf))
|
|
scanner.Split(splitFunc)
|
|
var ret [][]byte
|
|
for scanner.Scan() {
|
|
ret = append(ret, scanner.Bytes())
|
|
}
|
|
return ret, scanner.Err()
|
|
}
|
|
|
|
// TokenCount is a utility that uses a bufio.SplitFunc to count the number of
|
|
// recognized tokens in the given buffer.
|
|
func TokenCount(buf []byte, splitFunc bufio.SplitFunc) (int, error) {
|
|
scanner := bufio.NewScanner(bytes.NewReader(buf))
|
|
scanner.Split(splitFunc)
|
|
var ret int
|
|
for scanner.Scan() {
|
|
ret++
|
|
}
|
|
return ret, scanner.Err()
|
|
}
|