Method RowIter

server/node/create_extension.go:81–171

RowIter implements the interface sql.ExecSourceRel.

(ctx *sql.Context, r sql.Row)

// RowIter implements the interface sql.ExecSourceRel.
func (c *CreateExtension) RowIter(ctx *sql.Context, r sql.Row) (sql.RowIter, error) {
	extCollection, err := core.GetExtensionsCollectionFromContext(ctx, "")
	if err != nil {
		return nil, err
	}
	if extCollection.HasLoadedExtension(ctx, id.NewExtension(c.Name)) {
		if c.IfNotExists {
			return sql.RowsToRowIter(), nil
		}
		return nil, errors.Errorf(`extension "%s" already exists`, c.Name)
	}
	ext, err := extensions.GetExtension(c.Name)
	if err != nil {
		return nil, err
	}
	// The returned files are in their proper order of execution, so we can iterate and execute
	sqlFiles, err := ext.LoadSQLFiles()
	if err != nil {
		return nil, err
	}
	// save the current search_path
	originalSchema, err := ctx.GetSessionVariable(ctx, "search_path")
	if err != nil {
		return nil, err
	}

	if c.SchemaName != "" {
		defer func() {
			_ = ctx.SetSessionVariable(ctx, "search_path", originalSchema)
		}()

		spErr := ctx.SetSessionVariable(ctx, "search_path", c.SchemaName)
		if spErr != nil {
			return nil, spErr
		}
	}

	for _, sqlFile := range sqlFiles {
		// Remove echo PSQL control statements
		for {
			echoStartIdx := strings.Index(sqlFile, `\echo`)
			if echoStartIdx == -1 {
				break
			}
			echoEndIdx := strings.Index(sqlFile[echoStartIdx:], "\n")
			if echoEndIdx != -1 {
				// Set the correct absolute position if there is a newline
				echoEndIdx += echoStartIdx
			} else {
				// Set the position at the end of the file if there's no newline (comment appears before EOF)
				echoEndIdx = len(sqlFile)
			}
			sqlFile = strings.Replace(sqlFile, sqlFile[echoStartIdx:echoEndIdx], "", 1)
		}
		statements, err := parser.Parse(sqlFile)
		if err != nil {
			return nil, err
		}