diff options
| -rw-r--r-- | internal/mapr/groupsetresult.go | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/internal/mapr/groupsetresult.go b/internal/mapr/groupsetresult.go index 9c3c134..d01a3c9 100644 --- a/internal/mapr/groupsetresult.go +++ b/internal/mapr/groupsetresult.go @@ -221,14 +221,7 @@ func (g *GroupSet) resultWriteUnformatted(query *Query, rows []result, fd *os.Fi lastColumn := len(query.Select) - 1 if writeHeader { - for i, sc := range query.Select { - fd.WriteString(sc.FieldStorage) - if i == lastColumn { - continue - } - fd.WriteString(protocol.CSVDelimiter) - } - fd.WriteString("\n") + g.resultWriteUnformattedHeader(query, fd, lastColumn) } // And now write the data @@ -256,3 +249,14 @@ func (g *GroupSet) resultWriteUnformatted(query *Query, rows []result, fd *os.Fi return nil } + +func (g *GroupSet) resultWriteUnformattedHeader(query *Query, fd *os.File, lastColumn int) { + for i, sc := range query.Select { + fd.WriteString(sc.FieldStorage) + if i == lastColumn { + continue + } + fd.WriteString(protocol.CSVDelimiter) + } + fd.WriteString("\n") +} |