Function parseSQLFile

sql_migration.go:159–199

parseSQLFile parses the content of a SQL file into an SQLMigration struct It looks for the up and down annotations to split the file into up and down migrations example of file: -- migrate:up tx=true CREATE TABLE users (id INT PRIMARY KEY, name TEXT); -- migrate:down tx=false DROP TABLE users; In th

(fileContent []byte, config Configuration)

// DROP TABLE users;
// In this example, the up migration will be run in a transaction, while the down migration will not
func parseSQLFile(fileContent []byte, config Configuration) (SQLMigration, error) {
	file := SQLMigration{
		up:     "",
		down:   "",
		txUp:   true,
		txDown: true,
	}

	var upLines, downLines [][]byte
	var current *[][]byte // nil = before up, &upLines = in up, &downLines = in down

	txRegexp := regexp.MustCompile(`tx=(true|false)`)
	scanner := bufio.NewScanner(bytes.NewReader(fileContent))
	for scanner.Scan() {
		line := scanner.Bytes()

		if bytes.HasPrefix(line, []byte(config.SQLFileUpAnnotation)) {
			parseTxAnnotation(scanner.Text(), &file.txUp, txRegexp)
			current = &upLines
			continue
		}
		if bytes.HasPrefix(line, []byte(config.SQLFileDownAnnotation)) {
			parseTxAnnotation(scanner.Text(), &file.txDown, txRegexp)
			current = &downLines
			continue
		}

		if current != nil {
			*current = append(*current, bytes.Clone(line)) // Clone car scanner réutilise le buffer
		}
	}

	if err := scanner.Err(); err != nil {
		return file, fmt.Errorf("failed to scan file: %w", err)
	}

	file.up = string(bytes.Join(upLines, []byte("\n")))
	file.down = string(bytes.Join(downLines, []byte("\n")))

	return file, nil
}

// parseTxAnnotation parses the tx annotation from the given text and sets the value of b accordingly
// A typical refexp is regexp.MustCompile(`tx=(true|false)`)