added support 'IF NOT EXISTS' for functions

AfterShip · Nov 2, 2023 · 2d3c6a8 · 2d3c6a8
1 parent 7b0f30e
commit 2d3c6a8
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/parser/ast.go b/parser/ast.go
@@ -863,6 +863,7 @@ func (c *CreateView) String(level int) string {
 
 type CreateFunction struct {
 	CreatePos    Pos
+	IfNotExists  bool
 	FunctionName *Ident
 	OnCluster    *OnClusterExpr
 	Params       *ParamExprList
@@ -884,6 +885,9 @@ func (c *CreateFunction) End() Pos {
 func (c *CreateFunction) String(level int) string {
 	var builder strings.Builder
 	builder.WriteString("CREATE FUNCTION ")
+	if c.IfNotExists {
+		builder.WriteString("IF NOT EXISTS ")
+	}
 	builder.WriteString(c.FunctionName.String(level))
 	if c.OnCluster != nil {
 		builder.WriteString(NewLine(level))

diff --git a/parser/parser_table.go b/parser/parser_table.go
@@ -1263,6 +1263,10 @@ func (p *Parser) parseCreateFunction(pos Pos) (*CreateFunction, error) {
 	if err := p.consumeKeyword(KeywordFunction); err != nil {
 		return nil, err
 	}
+	ifNotExists, err := p.tryParseIfNotExists()
+	if err != nil {
+		return nil, err
+	}
 	functionName, err := p.parseIdent()
 	if err != nil {
 		return nil, err
@@ -1287,6 +1291,7 @@ func (p *Parser) parseCreateFunction(pos Pos) (*CreateFunction, error) {
 	}
 	return &CreateFunction{
 		CreatePos:    pos,
+		IfNotExists:  ifNotExists,
 		FunctionName: functionName,
 		OnCluster:    onCluster,
 		Params:       params,

diff --git a/parser/testdata/ddl/output/create_function_simple.sql.golden.json b/parser/testdata/ddl/output/create_function_simple.sql.golden.json
@@ -1,6 +1,7 @@
 [
   {
     "CreatePos": 0,
+    "IfNotExists": false,
     "FunctionName": {
       "Name": "linear_equation",
       "Unquoted": false,