MyBatis之Base64加密数据源

一般来讲，MyBatis的数据源都是通过这种形式直接配置的：
mybatis-config.xml:

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE configuration
PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
	<environments default="development">
		<environment id="development">
			<transactionManager type="JDBC" />
			<dataSource type="POOLED">
				<property name="driver" value="com.mysql.jdbc.Driver" />
				<property name="url" value="jdbc:mysql://127.0.0.1:3306/test" />
				<property name="username" value="root" />
				<property name="password" value="11111111" />
			</dataSource>
		</environment>
	</environments>
</configuration>

显然，这样把数据库连接的所有信息，明文的这样展示出来是一件很危险的事情。

所以，我就去研究了一下，如果对数据源信息进行加密。

我采用的就是简单Base64对文件进行加密，当然，有人说java是可以反编译的，对，没错，我们只防君子，不能防小人。

所以，我要做的事情是Base64对MyBatis的数据源进行加密.

那么，整理下思路，我要做的工作，主要有以下3点：

1.将url，username，password，进行加密，获得密文

2.将密文配置到MyBatis中,并MyBatis的创建SqlSessionFactory之前，对密文进行解密，然后获得数据库连接

3.关联数据源

1.将url，username，password，进行加密，获得密文

/**
 * 工具类，将Sting字符串进行加密
 * 
 * @author robin
 *
 */
public class String2Coder {
	public static void main(String[] args) {
		if (args.length < 1)
			return;
		byte[] buffer = Base64.getEncoder().encode(args[0].getBytes());
		System.out.println(new String(buffer));
		try {
			System.out.println("press any key to exit.");
			System.in.read();
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
}

这个类很简单，就是使用Base64对字段进行加密。当然，出于方便，可以直接将args[0]换成我们需要加密的字符串。

但是，我这样做，是为了顺便做个Base64加密的工具，然后将这个类导出为可执行的jar包。

然后，打开cmd，执行jar包,注意最后的 >out.txt 是将输出结果重定向到out.txt

打开out.txt文件:

cm9iaW4就是robin加密后的密文啦。

2.将密文配置到MyBatis中

2.1  首先需要说明的是，MyBatis，是支持将数据源分离成jdbc.propertis，所以第一步是准备jdbc.properties

jdbc.properties:

driver=com.mysql.jdbc.Driver url=****** username=****** password=******

2.2  通过代码获取jdbc.properties并将字段解密，创建SqlSessionFactory

package com.test.sql.session;

import java.io.IOException;
import java.io.InputStream;
import java.util.Base64;
import java.util.Properties;

import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;
import org.apache.log4j.Logger;

public class SqlSessionFactoryUtil {
	public static Logger logger = Logger.getLogger(SqlSessionFactoryUtil.class);

	private static SqlSessionFactory factory;

	public static Class<SqlSessionFactoryUtil> LOCK = SqlSessionFactoryUtil.class;

	public static SqlSession getSqlSession() {
		SqlSessionFactory sf = getFactory();
		return getFactory().openSession();
	}

	private static SqlSessionFactory getFactory() {
		if (factory != null)
			return factory;

		synchronized (LOCK) {
			if (factory == null) {
				InputStream inputStream = null;
				try {
					inputStream = Resources
							.getResourceAsStream("mybatis-config.xml");
					// read the encode jdbc.properties
					Properties prop = getJdbcPropertiesByDecode("jdbc.properties");
					factory = new SqlSessionFactoryBuilder().build(inputStream,
							prop);
				} catch (IOException e) {
					logger.error(e.getMessage(), e);
				} finally {
					if (inputStream != null)
						try {
							inputStream.close();
						} catch (IOException e) {
							e.printStackTrace();
						}
				}
			}
		}
		return factory;
	}

	private static Properties getJdbcPropertiesByDecode(String string)
			throws IOException {
		Properties prop = Resources.getResourceAsProperties("jdbc.properties");
		// String driver = prop.getProperty("driver");
		String url = prop.getProperty("url");
		String username = prop.getProperty("username");
		String password = prop.getProperty("password");

		byte[] buffer = Base64.getDecoder().decode(url.getBytes());
		prop.setProperty("url", new String(buffer));
		buffer = Base64.getDecoder().decode(username.getBytes());
		prop.setProperty("username", new String(buffer));
		buffer = Base64.getDecoder().decode(password.getBytes());
		prop.setProperty("password", new String(buffer));
		return prop;
	}

	public static void main(String[] args) throws IOException {
		Properties prop = Resources.getResourceAsProperties("jdbc.properties");
		// String driver = prop.getProperty("driver");
		String url = prop.getProperty("url");
		String username = prop.getProperty("username");
		String password = prop.getProperty("password");

		byte[] buffer = Base64.getDecoder().decode(url.getBytes());
		prop.setProperty("url", new String(buffer));
		buffer = Base64.getDecoder().decode(username.getBytes());
		prop.setProperty("username", new String(buffer));
		buffer = Base64.getDecoder().decode(password.getBytes());
		prop.setProperty("password", new String(buffer));
		
		System.out.println(prop);
	}
}

注意看getJdbcPropertiesByDecode方法，解密过程；

注意看factory = new SqlSessionFactoryBuilder().build(inputStream,prop);很清晰吧。

main，是用来测试我加密的密文，解密之后是否正确。

but!!!!到这里并没有结束。

笔者第一次配置的时候，天真的把dataSource全部删除掉了，我猜大概这样可以了吧，但，你会发现，并不行。

3.关联数据源

为什么不行，可以思考下，虽然我们通过factory = new SqlSessionFactoryBuilder().build(inputStream,prop);中的prop把url,username,password注入了进去，但还是没有获取到数据源。从设计的角度来看，jdbc.properties里面，我完全可以将url写成urllllllll,username写成name,password写成pwd。这样MyBatis是无法识别，所以最后一步就是关联。

再来看下关联后的mybatis-config.xml

mybatis-config.xml:

<?xml version="1.0" encoding="UTF-8" ?>
<!DOCTYPE configuration
PUBLIC "-//mybatis.org//DTD Config 3.0//EN"
"http://mybatis.org/dtd/mybatis-3-config.dtd">
<configuration>
	<environments default="development">
		<environment id="development">
			<transactionManager type="JDBC" />
			<dataSource type="POOLED">
				<property name="driver" value="${driver}" />
				<property name="url" value="${url}" />
				<property name="username" value="${username}" />
				<property name="password" value="${password}" />
			</dataSource>
		</environment>
	</environments>
</configuration>

此处的$url对应的就是第二部 prop.set("url",url);中的"url"了。

到此，结束。

笔者突然想到一个问题，就是MyBatis如何配置多个数据源。

其实很简单嘛，有几个数据源就配置几个mybatis-config.xml，也对应几个SqlSessionFactory。

完全没有任何问题。