Function convertSQLStmts

server/ast/create_function.go:221–254

convertSQLStmts takes parser.Statements and routine parameters and returns converted to string representation and vitess statements.

(stmts parser.Statements, params []pgnodes.RoutineParam)

// convertSQLStmts takes parser.Statements and routine parameters and
// returns converted to string representation and vitess statements.
func convertSQLStmts(stmts parser.Statements, params []pgnodes.RoutineParam) (string, []vitess.Statement, error) {
	paramMap := make(map[string]*framework.ParamTypAndValue, len(params))
	for i, param := range params {
		tv := &framework.ParamTypAndValue{
			Typ:        param.Type,
			Val:        nil,
			FromCreate: true,
		}
		// placeholder name is empty
		if param.Name == "\"\"" {
			n := fmt.Sprintf("$%d", i+1)
			paramMap[n] = tv
			params[i].Name = n
		} else {
			paramMap[param.Name] = tv
		}
	}

	var sqlDefs = make([]string, len(stmts))
	var vitessASTs = make([]vitess.Statement, len(stmts))
	for i, stmt := range stmts {
		sqlDefs[i] = stmt.AST.String()
		err := framework.ReplaceFunctionColumn(stmt.AST, paramMap)
		if err != nil {
			return "", nil, err
		}
		// stmt.AST is updated at this point with FunctionColumn
		vitessASTs[i], err = Convert(stmt)
		if err != nil {
			return "", nil, err
		}
	}
	return strings.Join(sqlDefs, ";"), vitessASTs, nil
}

// validateRoutineOptions ensures that each option is defined only once. Returns a map containing all options, or an
// error if an option is invalid or is defined multiple times.