Added options and tests for exporters.
This commit is contained in:
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user