49 lines
965 B
Go
49 lines
965 B
Go
|
package table
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
type CSVPrinter struct {
|
||
|
|
writer io.Writer
|
||
|
|
}
|
||
|
|
|
||
|
|
// Headers specify the table headers
|
||
|
|
func (p *CSVPrinter) Headers(headers ...string) error {
|
||
|
|
for _, header := range headers {
|
||
|
|
if _, err := fmt.Fprintf(p.writer, "%s,", header); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
_, err := fmt.Fprintf(p.writer, "\n")
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fields add another row
|
||
|
|
func (p *CSVPrinter) Fields(fields ...interface{}) error {
|
||
|
|
for _, field := range fields {
|
||
|
|
if _, err := fmt.Fprintf(p.writer, "%v,", field); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
_, err := fmt.Fprintf(p.writer, "\n")
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *CSVPrinter) StringFields(fields ...string) error {
|
||
|
|
for _, field := range fields {
|
||
|
|
if _, err := fmt.Fprintf(p.writer, "%v,", field); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
_, err := fmt.Fprintf(p.writer, "\n")
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Flush implements the flush interface but for CVS printer is a noop
|
||
|
|
func (p *CSVPrinter) Flush() error {
|
||
|
|
return nil
|
||
|
|
}
|