1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
// Copyright 2021-present The Atlas Authors. All rights reserved.
// This source code is licensed under the Apache 2.0 license found
// in the LICENSE file in the root directory of this source tree.
package sqlparse
import (
"sync"
"ariga.io/atlas/cmd/atlas/internal/sqlparse/myparse"
"ariga.io/atlas/cmd/atlas/internal/sqlparse/pgparse"
"ariga.io/atlas/cmd/atlas/internal/sqlparse/sqliteparse"
"ariga.io/atlas/sql/migrate"
"ariga.io/atlas/sql/mysql"
"ariga.io/atlas/sql/postgres"
"ariga.io/atlas/sql/schema"
"ariga.io/atlas/sql/sqlite"
)
// A Parser represents an SQL file parser used to fix, search and enrich schema.Changes.
type Parser interface {
// FixChange fixes the changes according to the given statement.
FixChange(d migrate.Driver, stmt string, changes schema.Changes) (schema.Changes, error)
// ColumnFilledBefore checks if the column was filled with values before the given position
// in the file. For example:
//
// UPDATE <table> SET <column> = <value>
// UPDATE <table> SET <column> = <value> WHERE <column> IS NULL
//
ColumnFilledBefore(migrate.File, *schema.Table, *schema.Column, int) (bool, error)
}
// drivers specific fixers.
var drivers sync.Map
// Register a fixer with the given name.
func Register(name string, f Parser) {
drivers.Store(name, f)
}
// ParserFor returns a ChangesFixer for the given driver.
func ParserFor(name string) Parser {
f, ok := drivers.Load(name)
if ok {
return f.(Parser)
}
return nil
}
func init() {
Register(mysql.DriverName, &myparse.Parser{})
Register(postgres.DriverName, &pgparse.Parser{})
Register(sqlite.DriverName, &sqliteparse.FileParser{})
}
|
