kylefang/cow
1
0
Fork 0
mirror of https://github.com/zhigang1992/cow.git synced 2026-04-30 18:32:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Simplify FieldsN.

Browse Source
This commit is contained in:
Chen Yufei
2013-02-21 20:09:46 +08:00
parent b8d98dacb1
commit 9e9dff4f2e
2 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
util.go
View File

@@ -129,18 +129,16 @@ func TrimSpace(s []byte) []byte {
return s[st : end+1]
}
// Simliar with bytes.Fields, but only consider space and '\t' as space, and
// will include all the content in the final slice with space trimmed.
// bytes.Split can't split on both space and '\t', and considers two separator
// as an empty item. bytes.FieldsFunc can't specify how much fields we need,
// which is required for parsing response status line.
// FieldsN is simliar with bytes.Fields, but only consider space and '\t' as
// space, and will include all content in the final slice with ending white
// space characters trimmed. bytes.Split can't split on both space and '\t',
// and considers two separator as an empty item. bytes.FieldsFunc can't
// specify how much fields we need, which is required for parsing response
// status line. Returns nil if n < 0.
func FieldsN(s []byte, n int) [][]byte {
if n == 0 {
if n <= 0 {
return nil
}
if n == 1 {
return [][]byte{TrimSpace(s)}
}
res := make([][]byte, n)
na := 0
fieldStart := -1
@@ -155,13 +153,13 @@ func FieldsN(s []byte, n int) [][]byte {
break
}
res[na] = s[fieldStart:i]
fieldStart = -1
na++
fieldStart = -1
}
}
if fieldStart >= 0 && na < n {
if fieldStart >= 0 { // must have na <= n-1 here
res[na] = TrimSpace(s[fieldStart:])
if len(res[na]) != 0 {
if len(res[na]) != 0 { // do not consider ending space as a field
na++
}
}