gitea/geziyor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added options and tests for exporters.

Browse Source
This commit is contained in:
Musab Gültekin
2019-06-27 16:54:09 +03:00
parent d20ea47390
commit 8fe194bd10
5 changed files with 55 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
exporter/csv.go
View File

@@ -12,19 +12,24 @@ import (
// CSVExporter exports response data as CSV streaming file
type CSVExporter struct {
FileName string
Comma rune
UseCRLF bool
}
// Export exports response data as CSV streaming file
func (e *CSVExporter) Export(exports chan interface{}) {
// Create file
newFile, err := os.OpenFile(internal.PreferFirst(e.FileName, "out.csv"), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
// Create or append file
file, err := os.OpenFile(internal.PreferFirst(e.FileName, "out.csv"), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if err != nil {
log.Printf("Output file creation error: %v\n", err)
return
}
defer file.Close()
writer := csv.NewWriter(newFile)
writer := csv.NewWriter(file)
writer.Comma = internal.PreferFirstRune(e.Comma, ',')
writer.UseCRLF = e.UseCRLF
// Export data as responses came
for res := range exports {