-
Notifications
You must be signed in to change notification settings - Fork 11
/
element_extractor.go
42 lines (37 loc) · 1.04 KB
/
element_extractor.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
package segment
import (
"fmt"
"github.com/mitch000001/go-hbci/token"
)
// ExtractElements extracts the data elements from segment
func ExtractElements(segment []byte) ([][]byte, error) {
extractor := &elementExtractor{
rawSegment: segment,
}
return extractor.Extract()
}
type elementExtractor struct {
rawSegment []byte
elements [][]byte
}
func (e *elementExtractor) Extract() ([][]byte, error) {
var current []byte
lexer := token.NewLexer("ElementExtractor", e.rawSegment)
for lexer.HasNext() {
t := lexer.Next()
if t.Type() == token.ERROR {
return nil, fmt.Errorf("%T: SyntaxError at position %d: %q\n(%q)", e, t.Pos(), t.Value(), e.rawSegment)
}
if t.Type() == token.SEGMENT_END_MARKER || t.Type() == token.DATA_ELEMENT_SEPARATOR {
if len(current) != 0 || t.Type() != token.SEGMENT_END_MARKER {
e.elements = append(e.elements, []byte(current))
}
current = []byte{}
} else {
current = append(current, t.Value()...)
}
}
result := make([][]byte, len(e.elements))
copy(result, e.elements)
return result, nil
}