Refactor - reduce code complexity

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")
+}