Added options and tests for exporters.

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

View File

@@ -11,20 +11,24 @@ import (
type JSONExporter struct {
FileName string
EscapeHTML bool
Prefix string
Indent string
}
// Export exports response data as JSON streaming file
func (e *JSONExporter) Export(exports chan interface{}) {
// Create file
newFile, err := os.OpenFile(internal.PreferFirst(e.FileName, "out.json"), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
// Create or append file
file, err := os.OpenFile(internal.PreferFirst(e.FileName, "out.json"), 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()
encoder := json.NewEncoder(newFile)
encoder := json.NewEncoder(file)
encoder.SetEscapeHTML(e.EscapeHTML)
encoder.SetIndent(e.Prefix, e.Indent)
// Export data as responses came
for res := range exports {