mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-10-15 07:18:04 +00:00
4c53f86e82
i used this script, a little clunky but it got the job done
```bash
for file in `find . -type f -print | grep '\.go'`; do
sed '1,/^\/\/ limitations under the License.$/d' $file > $file.tmp;
result=`cat $file.tmp`
if [[ result != "" ]]; then
gofmt $file.tmp &> /dev/null;
if [[ $? == 0 ]]; then
mv $file.tmp $file && gofmt -w $file;
else
rm $file.tmp;
fi
else
rm $file.tmp;
fi
done
```
103 lines
2.6 KiB
Go
103 lines
2.6 KiB
Go
package sql
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/molecula/featurebase/v2"
|
|
pproto "github.com/molecula/featurebase/v2/proto"
|
|
"github.com/pkg/errors"
|
|
"vitess.io/vitess/go/vt/sqlparser"
|
|
)
|
|
|
|
// ShowHandler executes SQL show table/field statements
|
|
type ShowHandler struct {
|
|
api *pilosa.API
|
|
}
|
|
|
|
// NewShowHandler constructor
|
|
func NewShowHandler(api *pilosa.API) *ShowHandler {
|
|
return &ShowHandler{
|
|
api: api,
|
|
}
|
|
}
|
|
|
|
// Handle executes mapped SQL
|
|
func (s *ShowHandler) Handle(ctx context.Context, mapped *MappedSQL) (pproto.ToRowser, error) {
|
|
stmt, ok := mapped.Statement.(*sqlparser.Show)
|
|
if !ok {
|
|
return nil, fmt.Errorf("statement is not type show: %T", mapped.Statement)
|
|
}
|
|
|
|
switch stmt.Type {
|
|
case "tables":
|
|
return s.execShowTables(ctx, stmt)
|
|
case "fields":
|
|
return s.execShowFields(ctx, stmt)
|
|
default:
|
|
return nil, fmt.Errorf("cannot show: %s", stmt.Type)
|
|
}
|
|
}
|
|
|
|
func (s *ShowHandler) execShowTables(ctx context.Context, showStmt *sqlparser.Show) (pproto.ToRowser, error) {
|
|
indexInfo, err := s.api.Schema(ctx, false)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "getting schema")
|
|
}
|
|
|
|
result := make(pproto.ConstRowser, len(indexInfo))
|
|
for i, ii := range indexInfo {
|
|
result[i] = pproto.RowResponse{
|
|
Headers: []*pproto.ColumnInfo{
|
|
{Name: "Table", Datatype: "string"},
|
|
},
|
|
Columns: []*pproto.ColumnResponse{
|
|
{ColumnVal: &pproto.ColumnResponse_StringVal{StringVal: ii.Name}},
|
|
},
|
|
}
|
|
}
|
|
|
|
// Sort the result.
|
|
return OrderBy(result, []string{"Table"}, []string{"asc"}), nil
|
|
}
|
|
|
|
func (s *ShowHandler) execShowFields(ctx context.Context, showStmt *sqlparser.Show) (pproto.ToRowser, error) {
|
|
indexName := showStmt.OnTable.ToViewName().Name.String()
|
|
index, err := s.api.Index(ctx, indexName)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "getting schema")
|
|
}
|
|
if index == nil {
|
|
return nil, errors.WithMessage(pilosa.ErrIndexNotFound, indexName)
|
|
}
|
|
fields := index.Fields()
|
|
|
|
result := make(pproto.ConstRowser, 0, len(fields))
|
|
for _, f := range fields {
|
|
if f.Name() == "_exists" {
|
|
continue
|
|
}
|
|
|
|
typeName := f.Type()
|
|
if f.Keys() {
|
|
typeName = "keyed-" + typeName
|
|
}
|
|
if f.ForeignIndex() != "" {
|
|
typeName = "foreign-" + typeName
|
|
}
|
|
|
|
result = append(result, pproto.RowResponse{
|
|
Headers: []*pproto.ColumnInfo{
|
|
{Name: "Field", Datatype: "string"},
|
|
{Name: "Type", Datatype: "string"},
|
|
},
|
|
Columns: []*pproto.ColumnResponse{
|
|
{ColumnVal: &pproto.ColumnResponse_StringVal{StringVal: f.Name()}},
|
|
{ColumnVal: &pproto.ColumnResponse_StringVal{StringVal: typeName}},
|
|
},
|
|
})
|
|
}
|
|
|
|
// Sort the result.
|
|
return OrderBy(result, []string{"Field"}, []string{"asc"}), nil
|
|
}
|