/
qcaaNapoItems.go
72 lines (59 loc) · 1.41 KB
/
qcaaNapoItems.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package reports
import (
"encoding/csv"
"fmt"
"github.com/nsip/dev-nrt/records"
)
type QcaaNapoItems struct {
baseReport // embed common setup capability
}
//
// Detailed breakdown of each TestITem
//
func QcaaNapoItemsReport() *QcaaNapoItems {
r := QcaaNapoItems{}
r.initialise("./config/QcaaNapoItems.toml")
r.printStatus()
return &r
}
//
// implement the EventPipe interface, core work of the
// report engine.
//
func (r *QcaaNapoItems) ProcessObjectRecords(in chan *records.ObjectRecord) chan *records.ObjectRecord {
out := make(chan *records.ObjectRecord)
go func() {
defer close(out)
// open the csv file writer, and set the header
w := csv.NewWriter(r.outF)
defer r.outF.Close()
w.Write(r.config.header)
defer w.Flush()
for cfr := range in {
if !r.config.activated { // only process if activated
out <- cfr
continue
}
if cfr.RecordType != "NAPTestItem" {
out <- cfr
continue
}
//
// now loop through the ouput definitions to create a
// row of results
//
var result string
var row []string = make([]string, 0, len(r.config.queries))
for _, query := range r.config.queries {
result = cfr.GetValueString(query)
row = append(row, result)
}
// write the row to the output file
if err := w.Write(row); err != nil {
fmt.Println("Warning: error writing record to csv:", r.config.name, err)
}
out <- cfr
}
}()
return out
}