chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,108 @@
|
||||
// Copyright 2023 Dolthub, Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package ast
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/cockroachdb/errors"
|
||||
vitess "github.com/dolthub/vitess/go/vt/sqlparser"
|
||||
|
||||
"github.com/dolthub/doltgresql/postgres/parser/sem/tree"
|
||||
)
|
||||
|
||||
// nodeCreateView handles *tree.CreateView nodes.
|
||||
func nodeCreateView(ctx *Context, node *tree.CreateView) (*vitess.DDL, error) {
|
||||
if node == nil {
|
||||
return nil, nil
|
||||
}
|
||||
if node.Persistence.IsTemporary() {
|
||||
return nil, errors.Errorf("CREATE TEMPORARY VIEW is not yet supported")
|
||||
}
|
||||
if node.IsRecursive {
|
||||
return nil, errors.Errorf("CREATE RECURSIVE VIEW is not yet supported")
|
||||
}
|
||||
var checkOption = tree.ViewCheckOptionUnspecified
|
||||
var sqlSecurity string
|
||||
if node.Options != nil {
|
||||
for _, opt := range node.Options {
|
||||
switch strings.ToLower(opt.Name) {
|
||||
case "check_option":
|
||||
switch strings.ToLower(opt.CheckOpt) {
|
||||
case "local":
|
||||
checkOption = tree.ViewCheckOptionLocal
|
||||
case "cascaded":
|
||||
checkOption = tree.ViewCheckOptionCascaded
|
||||
default:
|
||||
return nil, errors.Errorf(`"ERROR: syntax error at or near "%s"`, opt.Name)
|
||||
}
|
||||
case "security_barrier":
|
||||
if opt.Security {
|
||||
return nil, errors.Errorf("CREATE VIEW '%s' = true option is not yet supported", opt.Name)
|
||||
}
|
||||
case "security_invoker":
|
||||
if opt.Security {
|
||||
sqlSecurity = "invoker"
|
||||
} else {
|
||||
sqlSecurity = "definer"
|
||||
}
|
||||
default:
|
||||
return nil, errors.Errorf(`"ERROR: syntax error at or near "%s"`, opt.Name)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if checkOption != tree.ViewCheckOptionUnspecified && node.CheckOption != tree.ViewCheckOptionUnspecified {
|
||||
return nil, errors.Errorf(`ERROR: parameter "check_option" specified more than once`)
|
||||
} else {
|
||||
checkOption = node.CheckOption
|
||||
}
|
||||
|
||||
vCheckOpt := vitess.ViewCheckOptionUnspecified
|
||||
switch checkOption {
|
||||
case tree.ViewCheckOptionCascaded:
|
||||
vCheckOpt = vitess.ViewCheckOptionCascaded
|
||||
case tree.ViewCheckOptionLocal:
|
||||
vCheckOpt = vitess.ViewCheckOptionLocal
|
||||
default:
|
||||
}
|
||||
|
||||
tableName, err := nodeTableName(ctx, &node.Name)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
selectStmt, err := nodeSelect(ctx, node.AsSource)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var cols = make(vitess.Columns, len(node.ColumnNames))
|
||||
for i, col := range node.ColumnNames {
|
||||
cols[i] = vitess.NewColIdent(col.String())
|
||||
}
|
||||
|
||||
stmt := &vitess.DDL{
|
||||
Action: vitess.CreateStr,
|
||||
OrReplace: node.Replace,
|
||||
ViewSpec: &vitess.ViewSpec{
|
||||
ViewName: tableName,
|
||||
ViewExpr: selectStmt,
|
||||
Columns: cols,
|
||||
Security: sqlSecurity,
|
||||
CheckOption: vCheckOpt,
|
||||
},
|
||||
SubStatementStr: node.AsSource.String(),
|
||||
}
|
||||
return stmt, nil
|
||||
}
|
||||
Reference in New Issue
Block a user