836d0fe119
- Parsing `alter` statement to catch `change old_name new_name ...` statements - Auto deducing renamed columns - When suspecting renamed columns, requesting explicit `--approve-renamed-columns` or `--skip-renamed-columns` - updated tests
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
/*
|
|
Copyright 2016 GitHub Inc.
|
|
See https://github.com/github/gh-ost/blob/master/LICENSE
|
|
*/
|
|
|
|
package sql
|
|
|
|
import (
|
|
"regexp"
|
|
"strconv"
|
|
)
|
|
|
|
var (
|
|
renameColumnRegexp = regexp.MustCompile(`(?i)CHANGE\s+(column\s+|)([\S]+)\s+([\S]+)\s+`)
|
|
)
|
|
|
|
type Parser struct {
|
|
columnRenameMap map[string]string
|
|
}
|
|
|
|
func NewParser() *Parser {
|
|
return &Parser{
|
|
columnRenameMap: make(map[string]string),
|
|
}
|
|
}
|
|
|
|
func (this *Parser) ParseAlterStatement(alterStatement string) (err error) {
|
|
allStringSubmatch := renameColumnRegexp.FindAllStringSubmatch(alterStatement, -1)
|
|
for _, submatch := range allStringSubmatch {
|
|
submatch[2], _ = strconv.Unquote(submatch[2])
|
|
submatch[3], _ = strconv.Unquote(submatch[3])
|
|
|
|
this.columnRenameMap[submatch[2]] = submatch[3]
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (this *Parser) GetNonTrivialRenames() map[string]string {
|
|
result := make(map[string]string)
|
|
for column, renamed := range this.columnRenameMap {
|
|
if column != renamed {
|
|
result[column] = renamed
|
|
}
|
|
}
|
|
return result
|
|
}
|
|
|
|
func (this *Parser) HasNonTrivialRenames() bool {
|
|
return len(this.GetNonTrivialRenames()) > 0
|
|
}
|