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 package liquibase.sqlgenerator.ext;
27
28 import liquibase.database.Database;
29 import liquibase.exception.ValidationErrors;
30 import liquibase.sql.Sql;
31 import liquibase.sql.UnparsedSql;
32 import liquibase.sqlgenerator.SqlGenerator;
33 import liquibase.sqlgenerator.SqlGeneratorChain;
34 import liquibase.sqlgenerator.core.AbstractSqlGenerator;
35 import liquibase.statement.core.CreateSequenceStatement;
36
37 import java.util.List;
38 import java.util.ArrayList;
39 import java.util.Arrays;
40
41 import static liquibase.ext.Constants.EXTENSION_PRIORITY;
42
43
44
45
46
47
48 public class MysqlSequenceGenerator extends AbstractSqlGenerator<CreateSequenceStatement> {
49 private static final String CREATE_SEQUENCE_STATEMENT = "CREATE TABLE IF NOT EXISTS %s (id bigint(19) NOT NULL auto_increment, PRIMARY KEY(id) )";
50 private static final String SET_START_VALUE_STATEMENT = "INSERT INTO %s VALUES (%s)";
51
52 @Override
53 public int getPriority() {
54 return EXTENSION_PRIORITY;
55 }
56
57 @Override
58 public boolean supports(final CreateSequenceStatement statement, final Database database) {
59 return ("mysql".equals(database.getTypeName()));
60 }
61
62 @Override
63 public ValidationErrors validate(CreateSequenceStatement statement, Database database, SqlGeneratorChain sqlGeneratorChain) {
64 return new ValidationErrors();
65 }
66
67 @Override
68 public Sql[] generateSql(CreateSequenceStatement statement, Database database, SqlGeneratorChain sqlGeneratorChain) {
69 List<Sql> list = new ArrayList<Sql>();
70 list.add(new UnparsedSql(String.format(CREATE_SEQUENCE_STATEMENT, statement.getSequenceName())));
71
72 if (statement.getStartValue() != null) {
73
74 list.add(new UnparsedSql(String.format(SET_START_VALUE_STATEMENT, statement.getSequenceName(), statement.getStartValue())));
75 }
76
77 list.addAll(Arrays.asList(sqlGeneratorChain.generateSql(statement, database)));
78
79 return list.toArray(new Sql[list.size()]);
80
81 }
82 }