github.com/cockroachdb/cockroachdb-parser@v0.23.3-0.20240213214944-911057d40c9a/pkg/sql/sem/plpgsqltree/visitor.go (about)

     1  // Copyright 2023 The Cockroach Authors.
     2  //
     3  // Use of this software is governed by the Business Source License
     4  // included in the file licenses/BSL.txt.
     5  //
     6  // As of the Change Date specified in that file, in accordance with
     7  // the Business Source License, use of this software will be governed
     8  // by the Apache License, Version 2.0, included in the file
     9  // licenses/APL.txt.
    10  
    11  package plpgsqltree
    12  
    13  // StatementVisitor defines methods that are called plpgsql statements during
    14  // a statement walk.
    15  type StatementVisitor interface {
    16  	// Visit is called during a statement walk.
    17  	Visit(stmt Statement) (newStmt Statement, changed bool)
    18  }
    19  
    20  // Walk traverses the plpgsql statement.
    21  func Walk(v StatementVisitor, stmt Statement) Statement {
    22  	newStmt, _ := stmt.WalkStmt(v)
    23  	return newStmt
    24  }