-
Notifications
You must be signed in to change notification settings - Fork 34
/
edi.go
79 lines (71 loc) · 2.02 KB
/
edi.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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
/*
Package edi utilizes Go's csv.Reader/csv.Writer
for generating Electronic Data Interchange Files
This helps us:
- Extend the current package if more complex EDI logic is necessary
- Provide a more descriptive name for usage
- Adhere to patterns of Go's stdlib csv.Writer/csv.Reader
*/
package edi
import (
"bytes"
"encoding/csv"
"io"
)
// Writer is just a wrapper for csv.Writer
type Writer struct {
*csv.Writer
}
// NewWriter returns a wrapped csv.Writer with `Comma = '*'`
func NewWriter(w io.Writer) *Writer {
csvWriter := csv.NewWriter(w)
csvWriter.Comma = '*'
return &Writer{
csvWriter,
}
}
// Reader is just a wrapper for csv.Reader
type Reader struct {
*csv.Reader
}
// NewReader returns a wrapped csv.Reader with `Comma = '*'`
func NewReader(r io.Reader) *Reader {
csvReader := csv.NewReader(r)
csvReader.Comma = '*'
return &Reader{
csvReader,
}
}
// dropCR drops a terminal \r from the data.
// pulled from bufio library
//
// See https://cs.opensource.google/go/go/+/refs/tags/go1.16.7:src/bufio/scan.go;l=336
func dropCR(data []byte) []byte {
if len(data) > 0 && data[len(data)-1] == '\r' {
return data[0 : len(data)-1]
}
return data
}
// SplitLines is a split function for BufIO library
// it borrows from the default function but add support for files with only
// carriage returns as the line delimiter. Sometimes EDI files have only carriage returns
//
// See https://cs.opensource.google/go/go/+/refs/tags/go1.16.7:src/bufio/scan.go;l=350
func SplitLines(data []byte, atEOF bool) (advance int, token []byte, err error) {
if atEOF && len(data) == 0 {
return 0, nil, nil
}
if i := bytes.IndexByte(data, '\n'); i >= 0 {
// We have a full newline-terminated line.
return i + 1, dropCR(data[0:i]), nil
} else if i := bytes.IndexByte(data, '\r'); i >= 0 {
// We have a full carriage return terminated line.
return i + 1, dropCR(data[0:i]), nil
}
//If we're at EOF, we have a final, non-terminated line. Return it.
if atEOF {
return len(data), dropCR(data), nil
}
// Request more data.
return 0, nil, nil
}