1.背景
项目中使用的数据库是PostgreSQL,在做测试时,想使用H2代替。
2.问题
2.1 保留字:在PostgreSQL中使用了几个H2的保留字,例如 "end", "offset", "foreign",这些保留字是不能作为表的字段名。
2.2 字段类型:在PostgreSQL中double precision类型需要被替换成H2的float。
3.方法
liquibase提供一个ant任务:generateChangeLog,可以将PostgreSQL数据库中表,约束和数据都导出到一个xml文件中。然后通过liquibase根据导出的xml文件,在H2中生成数据库。
4.结果
生成xml文件,内容是liquibase的ChangeLog。
在使用generateChangeLog时,需要做一些扩展。
1.generateChangeLog任务
<generateChangeLog
outputFile="xxxxx\\structure.xml"
driver="org.postgresql.Driver"
url="jdbc:postgresql://localhost:5432/db"
username="name"
password="pass"
classpathref="classpath"
loglevel="debug"
databaseClass="xxx.ExtendPostgresDatabase"
changeLogFile="xxxx\\structure.ext"
diffTypes="tables, columns, views, indexes, sequences"
/>
outputFile指定生成的文件,包含指定生成文件的liquibase.serializer.core.xml.XMLChangeLogSerializer。
databaseClass指定任务执行时数据库的类型。
changeLogFile指定生成的自定义文件,当changeLogFile指定时outputFile会被忽略。需要注册自定义的liquibase.serializer.ChangeLogSerializer。
diffTypes指定generateChangeLog任务执行时哪些数据库元素被导出。
2.扩展PostgresDatabase类
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import liquibase.database.core.PostgresDatabase;
import liquibase.database.typeconversion.TypeConverterFactory;
import liquibase.serializer.ChangeLogSerializerFactory;
import liquibase.snapshot.DatabaseSnapshotGeneratorFactory;
public class ExtendPostgresDatabase
extends PostgresDatabase
{
public static final List<String> reservedWords = Arrays.asList( new String[] { "end", "offset", "foreign" } );
public static final List<String> ignoreTables = new ArrayList<String>();
public static final List<String> ignoreIndexes = new ArrayList<String>();
public static final List<String> ignorePrimaryKeys = new ArrayList<String>();
public static final List<String> ignoreForeignKeys = new ArrayList<String>();
public ExtendPostgresDatabase()
{
TypeConverterFactory.getInstance().register( new Postgres2H2TypeConverter() );//注册类型转换器,将Postgres的数据类型转换成H2
ChangeLogSerializerFactory.getInstance().register( new ExtendXMLChangeLogSerializer() );//注册ChangeLog文件生成器
DatabaseSnapshotGeneratorFactory.getInstance().register( new ExtendPostgresDatabaseSnapshotGenerator() );//注册数据库快照
}
@Override
public boolean isReservedWord( String word )
{
boolean reservedWord = super.isReservedWord( word );
if ( reservedWord )
{
return reservedWord;
}
return reservedWords.contains( word.toLowerCase() );
}
}
3.扩展PostgresDatabaseSnapshotGenerator类
import java.util.Set;
import liquibase.database.Database;
import liquibase.database.structure.ForeignKey;
import liquibase.database.structure.Index;
import liquibase.database.structure.PrimaryKey;
import liquibase.database.structure.Table;
import liquibase.diff.DiffStatusListener;
import liquibase.exception.DatabaseException;
import liquibase.snapshot.DatabaseSnapshot;
import liquibase.snapshot.jvm.PostgresDatabaseSnapshotGenerator;
public class ExtendPostgresDatabaseSnapshotGenerator
extends PostgresDatabaseSnapshotGenerator
{
@Override
public DatabaseSnapshot createSnapshot( Database database, String requestedSchema, Set<DiffStatusListener> listeners )
throws DatabaseException
{
DatabaseSnapshot snapshot = super.createSnapshot( database, requestedSchema, listeners );
//过滤掉不希望被导出的数据库元素
for ( String key : ExtendPostgresDatabase.ignoreIndexes )
{
Index index = snapshot.getIndex( key );
snapshot.getIndexes().remove( index );
}
for ( String key : ExtendPostgresDatabase.ignoreTables )
{
Table table = snapshot.getTable( key );
snapshot.getTables().remove( table );
}
for ( String key : ExtendPostgresDatabase.ignorePrimaryKeys )
{
PrimaryKey primaryKey = snapshot.getPrimaryKey( key );
snapshot.getPrimaryKeys().remove( primaryKey );
}
for ( String key : ExtendPostgresDatabase.ignoreForeignKeys )
{
ForeignKey foreignKey = snapshot.getForeignKey( key );
snapshot.getForeignKeys().remove( foreignKey );
}
return snapshot;
}
@Override
public boolean supports( Database database )
{
return database instanceof ExtendPostgresDatabase;
}
@Override
public int getPriority( Database database )
{
return super.getPriority( database ) + 1;
}
}
4.扩展XMLChangeLogSerializer类
import liquibase.change.ColumnConfig;
import liquibase.serializer.core.xml.XMLChangeLogSerializer;
import org.w3c.dom.Element;
public class ExtendXMLChangeLogSerializer
extends XMLChangeLogSerializer
{
@Override
public String[] getValidFileExtensions()
{
return new String[] { "ext" };//配合generateChangeLog任务中指定的changeLogFile的后缀名
}
public Element createNode( ColumnConfig columnConfig )
{
//如果列的名字是保留字,需要加上转义符,否则在导入H2时会失败。
Element createNode = super.createNode( columnConfig );
String tagName = createNode.getTagName();
if ( tagName.equalsIgnoreCase( "column" ) )
{
String name = createNode.getAttribute( "name" );
if(ExtendPostgresDatabase.reservedWords.contains(name.toLowerCase()))
{
createNode.setAttribute( "name", "\"" + name + "\"" );
}
}
return createNode;
}
}
5.扩展H2TypeConverter类
import liquibase.database.Database;
import liquibase.database.structure.Column;
import liquibase.database.typeconversion.core.H2TypeConverter;
public class Postgres2H2TypeConverter
extends H2TypeConverter
{
@Override
public int getPriority()
{
return super.getPriority()+1;
}
@Override
public boolean supports( Database database )
{
return database instanceof ExtendPostgresDatabase;
}
@Override
public String convertToDatabaseTypeString( Column referenceColumn, Database database )
{
//将PostgreSQL的类型转换成H2类型
if ( "LONGVARCHAR".equals( referenceColumn.getTypeName().toUpperCase() ) )
{
return "LONGVARCHAR";
}
if ( "FLOAT4".equals( referenceColumn.getTypeName().toUpperCase() ) )
{
return "FLOAT";
}
return super.convertToDatabaseTypeString( referenceColumn, database );
}
}
分享到:
相关推荐
C# 操作PostgreSQL 数据库 .
PostgreSQL数据库内核分析PostgreSQL数据库内核分析PostgreSQL数据库内核分析PostgreSQL数据库内核分析PostgreSQL数据库内核分析PostgreSQL数据库内核分析PostgreSQL数据库内核分析PostgreSQL数据库内核分析...
《PostgreSQL数据库内核分析》彭智勇 完整版44M
postGREsql数据库资料,经典资料,带完整书签。可以复制文字
Python编写PostgreSQL数据库结构比对程序源代码,该源代码导入PyCharm即可运行使用。
postgresql数据库jdbc驱动,jar包。。。。。。。。。。。。
asp连接postgresql数据库 的源码
1.nacos服务,适配postgresql数据库。 2.提供nacos,postgresql的创建nacos数据库脚本。 3.nacos/conf/nacos-pg.sql数据库脚本文件。 4.nacos版本1.4.2。
postgresql数据库定时备份脚本(linux),亲测可用总结
如何恢复PostgreSQL数据库. 如何恢复PostgreSQL数据库
PostgreSQL数据库内核分析完整版。 作者简介: 彭智勇:武汉大学计算机学院教授,博士生导师,日本京都大学博士,武汉大学珞珈特聘教授,中国计算机学会数据库专业委员会委员,长期从事数据库理论与技术的研究和教学...
postgresQL数据库备份脚本,添加到任务计划中就可以备份数据库文件了。
Window 下的PostgreSQL 数据库备份和恢复工具[GUI],供大家一起共同分享使用。
将excal批量数据插入到postgresql数据库(目前实现字符串拼接的方式)。。
postgresql数据库备份和恢复文档资料,详细介绍了PostgreSQL数据库的备份与恢复 方法,值得学习下
本套程序是在VS2005下C#开发,利用PostgreSQL作为数据库。 本程序的重点是如何利用C#对PostgreSQL数据库...将App_Data下的数据库文件导入到PostgreSQL 8.3版里面,库名为:HYGL 然后修改web.config中的配置即可使用