Function extractChangedResources

backend/plugin/parser/pg/resource_change.go:20–214

(database string, currentSchema string, dbMetadata *model.DatabaseMetadata, asts []base.AST, _ string)

}

func extractChangedResources(database string, currentSchema string, dbMetadata *model.DatabaseMetadata, asts []base.AST, _ string) (*base.ChangeSummary, error) {
	changedResources := model.NewChangedResources(dbMetadata)
	searchPath := dbMetadata.GetSearchPath()
	if currentSchema != "" {
		searchPath = []string{currentSchema}
	}
	if len(searchPath) == 0 {
		searchPath = []string{"public"}
	}

	if len(asts) == 0 {
		return &base.ChangeSummary{
			ChangedResources: changedResources,
			DMLCount:         0,
			SampleDMLS:       []string{},
			InsertCount:      0,
		}, nil
	}

	var dmlCount, insertCount int
	var sampleDMLs []string

	for _, unifiedAST := range asts {
		omniAST, ok := unifiedAST.(*OmniAST)
		if !ok {
			return nil, errors.New("expected OmniAST for PostgreSQL")
		}
		if omniAST.Node == nil {
			continue
		}

		switch n := omniAST.Node.(type) {
		case *ast.VariableSetStmt:
			if strings.EqualFold(n.Name, "search_path") && n.Args != nil {
				var newSearchPath []string
				for _, arg := range n.Args.Items {
					if ac, ok := arg.(*ast.A_Const); ok {
						if s, ok := ac.Val.(*ast.String); ok {
							newSearchPath = append(newSearchPath, s.Str)
						}
					}
				}
				if len(newSearchPath) > 0 {
					searchPath = newSearchPath
				}
			}

		case *ast.CreateStmt:
			if n.Relation != nil {
				db, schema, table := extractRangeVarNames(n.Relation, database, searchPath)
				changedResources.AddTable(db, schema, &storepb.ChangedResourceTable{Name: table}, false)
			}

		case *ast.DropStmt:
			objType := ast.ObjectType(n.RemoveType)
			switch objType {
			case ast.OBJECT_INDEX:
				handleDropIndexOmni(n, database, searchPath, dbMetadata, changedResources)