Add parser features and enable 8 more tests (#17)

Author: kyleconroy

ast/alter_server_configuration_statement.go

@@ -1,5 +1,22 @@
 package ast
 
+// AlterServerConfigurationStatement represents ALTER SERVER CONFIGURATION SET PROCESS AFFINITY statement
+type AlterServerConfigurationStatement struct {
+	ProcessAffinity       string                  // "CpuAuto", "Cpu", "NumaNode"
+	ProcessAffinityRanges []*ProcessAffinityRange // for Cpu or NumaNode
+}
+
+func (a *AlterServerConfigurationStatement) node()      {}
+func (a *AlterServerConfigurationStatement) statement() {}
+
+// ProcessAffinityRange represents a CPU or NUMA node range
+type ProcessAffinityRange struct {
+	From ScalarExpression // IntegerLiteral
+	To   ScalarExpression // IntegerLiteral (optional)
+}
+
+func (p *ProcessAffinityRange) node() {}
+
 // AlterServerConfigurationSetSoftNumaStatement represents ALTER SERVER CONFIGURATION SET SOFTNUMA statement
 type AlterServerConfigurationSetSoftNumaStatement struct {
 	Options []*AlterServerConfigurationSoftNumaOption

ast/create_table_statement.go

@@ -83,11 +83,12 @@ type TableConstraint interface {
 
 // IndexDefinition represents an index definition within CREATE TABLE
 type IndexDefinition struct {
-	Name         *Identifier
-	Columns      []*ColumnWithSortOrder
-	Unique       bool
-	IndexType    *IndexType
-	IndexOptions []*IndexExpressionOption
+	Name           *Identifier
+	Columns        []*ColumnWithSortOrder
+	Unique         bool
+	IndexType      *IndexType
+	IndexOptions   []*IndexExpressionOption
+	IncludeColumns []*ColumnReferenceExpression
 }
 
 func (i *IndexDefinition) node() {}
@@ -108,3 +109,41 @@ const (
 	SortOrderAscending
 	SortOrderDescending
 )
+
+// CheckConstraintDefinition represents a CHECK constraint
+type CheckConstraintDefinition struct {
+	ConstraintIdentifier *Identifier
+	CheckCondition       BooleanExpression
+	NotForReplication    bool
+}
+
+func (c *CheckConstraintDefinition) node()              {}
+func (c *CheckConstraintDefinition) tableConstraint()   {}
+func (c *CheckConstraintDefinition) constraintDefinition() {}
+
+// UniqueConstraintDefinition represents a UNIQUE or PRIMARY KEY constraint
+type UniqueConstraintDefinition struct {
+	ConstraintIdentifier *Identifier
+	Clustered            bool
+	IsPrimaryKey         bool
+	Columns              []*ColumnWithSortOrder
+	IndexType            *IndexType
+}
+
+func (u *UniqueConstraintDefinition) node()              {}
+func (u *UniqueConstraintDefinition) tableConstraint()   {}
+func (u *UniqueConstraintDefinition) constraintDefinition() {}
+
+// ForeignKeyConstraintDefinition represents a FOREIGN KEY constraint
+type ForeignKeyConstraintDefinition struct {
+	ConstraintIdentifier *Identifier
+	Columns              []*Identifier
+	ReferenceTableName   *SchemaObjectName
+	ReferencedColumns    []*Identifier
+	DeleteAction         string
+	UpdateAction         string
+	NotForReplication    bool
+}
+
+func (f *ForeignKeyConstraintDefinition) node()            {}
+func (f *ForeignKeyConstraintDefinition) tableConstraint() {}

ast/declare_variable_statement.go

@@ -16,6 +16,23 @@ type DeclareVariableElement struct {
 	Nullable     *NullableConstraintDefinition `json:"Nullable,omitempty"`
 }
 
+// DeclareTableVariableStatement represents a DECLARE @var TABLE statement.
+type DeclareTableVariableStatement struct {
+	Body *DeclareTableVariableBody `json:"Body,omitempty"`
+}
+
+func (d *DeclareTableVariableStatement) node()      {}
+func (d *DeclareTableVariableStatement) statement() {}
+
+// DeclareTableVariableBody represents the body of a table variable declaration.
+type DeclareTableVariableBody struct {
+	VariableName *Identifier      `json:"VariableName,omitempty"`
+	AsDefined    bool             `json:"AsDefined,omitempty"`
+	Definition   *TableDefinition `json:"Definition,omitempty"`
+}
+
+func (d *DeclareTableVariableBody) node() {}
+
 // SqlDataTypeReference represents a SQL data type.
 type SqlDataTypeReference struct {
 	SqlDataTypeOption string            `json:"SqlDataTypeOption,omitempty"`

ast/drop_statements.go

@@ -152,6 +152,15 @@ type DropExternalResourcePoolStatement struct {
 func (s *DropExternalResourcePoolStatement) statement() {}
 func (s *DropExternalResourcePoolStatement) node()      {}
 
+// DropExternalModelStatement represents a DROP EXTERNAL MODEL statement
+type DropExternalModelStatement struct {
+	IsIfExists bool
+	Name       *SchemaObjectName
+}
+
+func (s *DropExternalModelStatement) statement() {}
+func (s *DropExternalModelStatement) node()      {}
+
 // DropWorkloadGroupStatement represents a DROP WORKLOAD GROUP statement
 type DropWorkloadGroupStatement struct {
 	IsIfExists bool

ast/set_variable_statement.go

@@ -2,12 +2,14 @@ package ast
 
 // SetVariableStatement represents a SET @var = value statement.
 type SetVariableStatement struct {
-	Variable          *VariableReference  `json:"Variable,omitempty"`
-	Expression        ScalarExpression    `json:"Expression,omitempty"`
-	CursorDefinition  *CursorDefinition   `json:"CursorDefinition,omitempty"`
-	AssignmentKind    string              `json:"AssignmentKind,omitempty"`
-	SeparatorType     string              `json:"SeparatorType,omitempty"`
+	Variable           *VariableReference `json:"Variable,omitempty"`
+	Expression         ScalarExpression   `json:"Expression,omitempty"`
+	CursorDefinition   *CursorDefinition  `json:"CursorDefinition,omitempty"`
+	AssignmentKind     string             `json:"AssignmentKind,omitempty"`
+	SeparatorType      string             `json:"SeparatorType,omitempty"`
+	Identifier         *Identifier        `json:"Identifier,omitempty"`
 	FunctionCallExists bool               `json:"FunctionCallExists,omitempty"`
+	Parameters         []ScalarExpression `json:"Parameters,omitempty"`
 }
 
 func (s *SetVariableStatement) node()      {}