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
```
51 lines
1.3 KiB
Go
51 lines
1.3 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"
|
|
)
|
|
|
|
// DDLHandler executes CREATE, ALTER, DROP, RENAME, TRUNCATE or ANALYZE statement.
|
|
type DDLHandler struct {
|
|
api *pilosa.API
|
|
}
|
|
|
|
// NewDDLHandler constructor
|
|
func NewDDLHandler(api *pilosa.API) *DDLHandler {
|
|
return &DDLHandler{
|
|
api: api,
|
|
}
|
|
}
|
|
|
|
// Handle executes mapped SQL
|
|
func (h *DDLHandler) Handle(ctx context.Context, mapped *MappedSQL) (pproto.ToRowser, error) {
|
|
stmt, ok := mapped.Statement.(*sqlparser.DDL)
|
|
if !ok {
|
|
return nil, fmt.Errorf("statement is not type DDL: %T", mapped.Statement)
|
|
}
|
|
|
|
switch stmt.Action {
|
|
case sqlparser.DropStr:
|
|
return h.execDropTable(ctx, stmt)
|
|
|
|
default:
|
|
return nil, errors.Errorf("unsupported DDL action: %s", stmt.Action)
|
|
}
|
|
}
|
|
|
|
func (h *DDLHandler) execDropTable(ctx context.Context, stmt *sqlparser.DDL) (pproto.ToRowser, error) {
|
|
if n := len(stmt.FromTables); n != 1 {
|
|
return nil, fmt.Errorf("statement can only contain a single drop table, but got: %d", n)
|
|
}
|
|
|
|
indexName := stmt.FromTables[0].ToViewName().Name.String()
|
|
if err := h.api.DeleteIndex(ctx, indexName); err != nil {
|
|
return nil, errors.Wrapf(err, "deleting index %s", indexName)
|
|
}
|
|
return pproto.ConstRowser{}, nil
|
|
}
|