mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-25 09:16:59 +03:00
2831211ae9
* wip: bunch o' linting * bunch more linting
192 lines
5.1 KiB
Go
192 lines
5.1 KiB
Go
package tablew
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/neilotoole/sq/cli/output"
|
|
"github.com/neilotoole/sq/libsq/core/stringz"
|
|
"github.com/neilotoole/sq/libsq/driver"
|
|
"github.com/neilotoole/sq/libsq/source"
|
|
)
|
|
|
|
type mdWriter struct {
|
|
tbl *table
|
|
}
|
|
|
|
// NewMetadataWriter returns a new output.MetadataWriter instance
|
|
// that outputs metadata in table format.
|
|
func NewMetadataWriter(out io.Writer, fm *output.Formatting) output.MetadataWriter {
|
|
tbl := &table{out: out, fm: fm, header: true}
|
|
w := &mdWriter{tbl: tbl}
|
|
w.tbl.reset()
|
|
return w
|
|
}
|
|
|
|
// DriverMetadata implements output.MetadataWriter.
|
|
func (w *mdWriter) DriverMetadata(drvrs []driver.Metadata) error {
|
|
headers := []string{"DRIVER", "DESCRIPTION", "USER-DEFINED", "DOC"}
|
|
w.tbl.tblImpl.SetHeader(headers)
|
|
w.tbl.tblImpl.SetColTrans(2, w.tbl.fm.Bool.SprintFunc())
|
|
|
|
var rows [][]string
|
|
for _, md := range drvrs {
|
|
row := []string{string(md.Type), md.Description, strconv.FormatBool(md.UserDefined), md.Doc}
|
|
rows = append(rows, row)
|
|
}
|
|
w.tbl.appendRowsAndRenderAll(rows)
|
|
return nil
|
|
}
|
|
|
|
// TableMetadata implements output.MetadataWriter.
|
|
func (w *mdWriter) TableMetadata(tblMeta *source.TableMetadata) error {
|
|
var headers []string
|
|
var rows [][]string
|
|
|
|
colNames := make([]string, len(tblMeta.Columns))
|
|
colTypes := make([]string, len(tblMeta.Columns))
|
|
|
|
for i, col := range tblMeta.Columns {
|
|
colNames[i] = col.Name
|
|
colTypes[i] = col.ColumnType
|
|
}
|
|
|
|
size := "-"
|
|
if tblMeta.Size != nil {
|
|
size = stringz.ByteSized(*tblMeta.Size, 1, "")
|
|
}
|
|
|
|
if w.tbl.fm.Verbose {
|
|
headers = []string{"TABLE", "ROWS", "TYPE", "SIZE", "NUM COLS", "COL NAMES", "COL TYPES"}
|
|
|
|
w.tbl.tblImpl.SetHeader(headers)
|
|
w.tbl.tblImpl.SetColTrans(0, w.tbl.fm.Handle.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(1, w.tbl.fm.Number.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(3, w.tbl.fm.Number.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(4, w.tbl.fm.Number.SprintFunc())
|
|
|
|
row := []string{
|
|
tblMeta.Name,
|
|
fmt.Sprintf("%d", tblMeta.RowCount),
|
|
tblMeta.TableType,
|
|
size,
|
|
fmt.Sprintf("%d", len(tblMeta.Columns)),
|
|
strings.Join(colNames, ", "),
|
|
strings.Join(colTypes, ", "),
|
|
}
|
|
rows = append(rows, row)
|
|
} else {
|
|
headers = []string{"TABLE", "ROWS", "COL NAMES"}
|
|
|
|
w.tbl.tblImpl.SetHeader(headers)
|
|
w.tbl.tblImpl.SetColTrans(0, w.tbl.fm.Handle.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(1, w.tbl.fm.Number.SprintFunc())
|
|
|
|
row := []string{
|
|
tblMeta.Name,
|
|
fmt.Sprintf("%d", tblMeta.RowCount),
|
|
strings.Join(colNames, ", "),
|
|
}
|
|
rows = append(rows, row)
|
|
}
|
|
|
|
w.tbl.appendRowsAndRenderAll(rows)
|
|
return nil
|
|
}
|
|
|
|
// SourceMetadata implements output.MetadataWriter.
|
|
func (w *mdWriter) SourceMetadata(meta *source.Metadata) error {
|
|
var headers []string
|
|
var row []string
|
|
|
|
if meta.Name == meta.FQName {
|
|
headers = []string{"HANDLE", "DRIVER", "NAME", "SIZE", "TABLES", "LOCATION"}
|
|
w.tbl.tblImpl.SetColTrans(0, w.tbl.fm.Handle.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(3, w.tbl.fm.Number.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(4, w.tbl.fm.Number.SprintFunc())
|
|
row = []string{
|
|
meta.Handle,
|
|
meta.SourceType.String(),
|
|
meta.Name,
|
|
stringz.ByteSized(meta.Size, 1, ""),
|
|
fmt.Sprintf("%d", len(meta.Tables)),
|
|
source.RedactLocation(meta.Location),
|
|
}
|
|
} else {
|
|
headers = []string{"HANDLE", "DRIVER", "NAME", "FQ NAME", "SIZE", "TABLES", "LOCATION"}
|
|
w.tbl.tblImpl.SetColTrans(0, w.tbl.fm.Handle.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(4, w.tbl.fm.Number.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(5, w.tbl.fm.Number.SprintFunc())
|
|
row = []string{
|
|
meta.Handle,
|
|
meta.SourceType.String(),
|
|
meta.Name,
|
|
meta.FQName,
|
|
stringz.ByteSized(meta.Size, 1, ""),
|
|
fmt.Sprintf("%d", len(meta.Tables)),
|
|
source.RedactLocation(meta.Location),
|
|
}
|
|
}
|
|
|
|
w.tbl.tblImpl.SetHeader(headers)
|
|
w.tbl.renderRow(row)
|
|
w.tbl.reset()
|
|
fmt.Fprintln(w.tbl.out)
|
|
|
|
if w.tbl.fm.Verbose {
|
|
headers = []string{"TABLE", "ROWS", "TYPE", "SIZE", "NUM COLS", "COL NAMES", "COL TYPES"}
|
|
w.tbl.tblImpl.SetHeader(headers)
|
|
w.tbl.tblImpl.SetColTrans(0, w.tbl.fm.Handle.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(1, w.tbl.fm.Number.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(3, w.tbl.fm.Number.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(4, w.tbl.fm.Number.SprintFunc())
|
|
} else {
|
|
headers = []string{"TABLE", "ROWS", "COL NAMES"}
|
|
w.tbl.tblImpl.SetHeader(headers)
|
|
w.tbl.tblImpl.SetColTrans(0, w.tbl.fm.Handle.SprintFunc())
|
|
w.tbl.tblImpl.SetColTrans(1, w.tbl.fm.Number.SprintFunc())
|
|
}
|
|
|
|
var rows [][]string
|
|
|
|
for _, tbl := range meta.Tables {
|
|
colNames := make([]string, len(tbl.Columns))
|
|
colTypes := make([]string, len(tbl.Columns))
|
|
|
|
for i, col := range tbl.Columns {
|
|
colNames[i] = col.Name
|
|
colTypes[i] = col.ColumnType
|
|
}
|
|
|
|
size := "-"
|
|
if tbl.Size != nil {
|
|
size = stringz.ByteSized(*tbl.Size, 1, "")
|
|
}
|
|
|
|
if w.tbl.fm.Verbose {
|
|
row = []string{
|
|
tbl.Name,
|
|
fmt.Sprintf("%d", tbl.RowCount),
|
|
tbl.TableType,
|
|
size,
|
|
fmt.Sprintf("%d", len(tbl.Columns)),
|
|
strings.Join(colNames, ", "),
|
|
strings.Join(colTypes, ", "),
|
|
}
|
|
} else {
|
|
row = []string{
|
|
tbl.Name,
|
|
fmt.Sprintf("%d", tbl.RowCount),
|
|
strings.Join(colNames, ", "),
|
|
}
|
|
}
|
|
|
|
rows = append(rows, row)
|
|
}
|
|
|
|
w.tbl.appendRowsAndRenderAll(rows)
|
|
return nil
|
|
}
|