java ssh方式连接mysql数据库
ssh方式连接mysql数据库Jsch这个包里面其实已经帮我们封装好了，关键代码就只有几行.
/**
* 数据备份
* @param backup
* @return
*/
public Map doDataBackup(BackupDTO backup) {
String schedule = backup.getSchedule();
String config = backup.getConfig();
BackupConfigDTO cfg = (BackupConfigDTO) JSONObject.toBean(JSONObject.fromObject(config),BackupConfigDTO.class);
String target = cfg.getTarget();
String[] tables = null;
String database = null;
String backupSource = backup.getBackupSource();
if(StringUtils.contains(backupSource, ":")){
String[] tempArr = backupSource.split(":");
database = tempArr[0];
tables = tempArr[1].split(",");
}else{
database = backupSource;
try {
List tnames = new ArrayList();
List> tablelist = getTables(cfg);
for (Map map : tablelist) {
tnames.add(map.get("tableName").toString());
}
tables = tnames.toArray(new String[tnames.size()]);
} catch (JSchException e) {
logger.debug(e);
return UtilMisc.toMap("flag","failure","msg",e.getMessage());
} catch (Exception e) {
logger.debug(e);
return UtilMisc.toMap("flag","failure","msg",e.getMessage());
}
}
if(tables.length < 1){
return UtilMisc.toMap("flag","failure","msg","数据库"+ database +" 中没有可以备份的表 ");
}
String backupLocation = backup.getBackupLocation();
String backupName = backup.getBackupName();
String destFileDir = FileUtil.normalize(backupLocation);
String dirStr = FileUtil.normalize(destFileDir + "/" + backupName +".sql");
File dirFile = new File(dirStr);
if(!dirFile.isAbsolute()){
return UtilMisc.toMap("flag","failure","msg","备份文件存放路径必须是绝对路径");
}
if("immediate".equals(schedule)){
DataSource ds = getDataSource(cfg);
JdbcTemplate jdbcTemplate = new JdbcTemplate(ds);
boolean isZip = backup.getNeedZip().intValue() == 1 ? true : false;
boolean isCover = backup.getCover().intValue() == 1 ? true : false;
if(isCover && dirFile.exists()){
dirFile.delete();
}
if(cfg.isIncludeCreateDB()){
try {
writeFile(dirFile, "CREATE DATABASE `"+ database +"`;\n\n");
} catch (IOException e) {
logger.debug(e);
return UtilMisc.toMap("flag","failure","msg","备份发生错误");
}
}
if(cfg.isIncludeUseDB()){
try {
writeFile(dirFile, "USE `"+ database +"`;\n\n");
} catch (IOException e) {
logger.debug(e);
return UtilMisc.toMap("flag","failure","msg","备份发生错误");
}
}
//备份结构
if(target.equals(BackupConst.BACKUP_TARGET_STRUCTURE_ONLY)){
for (String table : tables) {
String structure = null;
try {
structure = getTableStructure(jdbcTemplate, table, cfg);
} catch (Exception e1) {
logger.debug(e1);
return UtilMisc.toMap("flag","failure","msg","获取表结构发生错误");
}
try {
writeFile(dirFile, structure);
} catch (IOException e) {
logger.debug(e);
return UtilMisc.toMap("flag","failure","msg","向文件写入表结构时发生错误");
}
}
}
//备份数据
else if(target.equals(BackupConst.BACKUP_TARGET_DATA_ONLY)){
for (String table : tables) {
try {
writeData(jdbcTemplate, table, cfg, dirFile);
} catch (IOException e) {
logger.debug(e);
return UtilMisc.toMap("flag","failure","msg","向文件写入表数据时发生错误");
}
}
}
//备份数据&结构
else if(target.equals(BackupConst.BACKUP_TARGET_STRUCTURE_AND_DATA)){
for (String table : tables) {
String structure = null;
try {
structure = getTableStructure(jdbcTemplate, table, cfg);
} catch (Exception e1) {
logger.debug(e1);
return UtilMisc.toMap("flag","failure","msg","获取表结构发生错误");
}
try {
writeFile(dirFile, structure);
} catch (IOException e) {
logger.debug(e);
return UtilMisc.toMap("flag","failure","msg","向文件写入表结构时发生错误");
}
try {
writeData(jdbcTemplate, table, cfg, dirFile);
} catch (IOException e) {
logger.debug(e);
return UtilMisc.toMap("flag","failure","msg","向文件写入表数据时发生错误");
}
}
}
}else{
return UtilMisc.toMap("flag","success");
}
return null;
}
/**
* 向文件中写入表数据
* @param jdbcTemplate
* @param table
* @param cfg
* @param dirFile
* @throws IOException
*/
private void writeData(JdbcTemplate jdbcTemplate, String table,
BackupConfigDTO cfg, File dirFile) throws IOException {
StringBuffer content = new StringBuffer();
boolean batchInsert = cfg.isBatchInsert();
boolean lockInsert = cfg.isLockInsert();
StringBuffer insertStrBuff = new StringBuffer("insert into `"+ table +"`(");
String sql = "DESCRIBE `"+ table +"`";
List fields = new ArrayList();
List> result = jdbcTemplate.queryForList(sql);
for (Map map : result) {
fields.add(map.get("Field").toString());
}
for (String field : fields) {
insertStrBuff.append("`"+ field +"`,");
}
insertStrBuff = insertStrBuff.delete(insertStrBuff.length()-1, insertStrBuff.length());
insertStrBuff.append(")").append(" values ");
String insertStr = insertStrBuff.toString();
writeFile(dirFile, "/*Data for the table `"+ table +"` */\n\n");
if(fields.size() > 0){
//计算分次写入文件的写入次数
int rcdCount = jdbcTemplate.queryForObject("SELECT COUNT(*) FROM `"+ table +"`", Integer.class);
int cycleNum = 1;
if(rcdCount > 1000){
if(rcdCount%1000 == 0){
cycleNum = (int) Math.floor(rcdCount/1000);
}else{
cycleNum = (int) Math.floor(rcdCount/1000) + 1;
}
}
if(lockInsert){
writeFile(dirFile, "LOCK TABLES `"+ table +"` WRITE;\n\n");
}
if(batchInsert){
writeFile(dirFile, insertStr);
}
//多次写入
for (int i = 0; i < cycleNum; i++) {
content = new StringBuffer();
insertStrBuff = new StringBuffer(insertStr);
int recordStart = i*1000;
String sqlTemplate = "select * from "+ table +" limit "+ recordStart +",1000 ;";
List> records = jdbcTemplate.queryForList(sqlTemplate);
StringBuffer valuesBuff = new StringBuffer();
StringBuffer rowValues = null;
//迭代查询结果
for (Map map : records) {
//迭代字段
rowValues = new StringBuffer("(");
for (String field : fields) {
Object value = map.get(field);
String fieldValue = null;
if(value instanceof byte[]){
fieldValue = getHexString(((byte[]) value));
}
if(!(value instanceof byte[]) && value != null){
fieldValue = "'"+StringUtils.replace(value.toString(), "'", "\'")+"'";
fieldValue = StringUtils.replace(fieldValue, "\"", "\\\"");
fieldValue = StringUtils.replace(fieldValue, "\n", "\\n");
fieldValue = StringUtils.replace(fieldValue, "\r", "\\r");
}
if(value == null){
fieldValue = "NULL";
}
rowValues.append(fieldValue).append(",");
}
rowValues = rowValues.delete(rowValues.length()-1,rowValues.length());
rowValues.append(")");
if(batchInsert){
valuesBuff.append(rowValues).append(",");
}else{
content.append(insertStr).append(rowValues).append(";\r\n");
}
}
if(batchInsert && valuesBuff != null){
if(cycleNum > 0 && i < cycleNum - 1){
content = valuesBuff;
}else{
content = valuesBuff.delete(valuesBuff.length()-1, valuesBuff.length());
}
}
writeFile(dirFile, content.toString());
}
if(batchInsert){
writeFile(dirFile, ";\n\n");
}else{
writeFile(dirFile, "\r\n");
}
if(lockInsert){
writeFile(dirFile, "UNLOCK TABLES;\n\n");
}
}
}
private String getHexString(byte[] bs) {
String result = "";
for (int i = 0; i < bs.length; i++) {
result += Integer.toString((bs[i] & 0xff) + 0x100, 16).substring(1);
}
return "0x" + result;
}
/**
* 获取表创建语句
* @param jdbcTemplate
* @param tableName
* @return
*/
public String getTableStructure(JdbcTemplate jdbcTemplate, String tableName, BackupConfigDTO config) throws Exception{
String structure = null;
String remark = "/*Table structure for table `"+ tableName +"` */";
String sql = "SHOW CREATE TABLE `"+ tableName +"`";
List> result = jdbcTemplate.queryForList(sql);
for (Map map : result) {
structure = map.get("Create Table").toString();
}
StringBuffer strbuff = new StringBuffer();
strbuff.append(remark);
strbuff.append("\n\n");
if(config.isIncludeDel()){
strbuff.append("DROP TABLE IF EXISTS `"+ tableName +"`;");
strbuff.append("\n\n");
}
strbuff.append(structure);
strbuff.append(";\n\n");
return strbuff.toString();
}
/**
* 写文件
* @param destFile
* @param isZip
* @param isCover
* @param structure
* @throws IOException
*/
private void writeFile(File destFile, String structure) throws IOException {
if(!destFile.exists()){
destFile.createNewFile();
}
FileOutputStream fos = new FileOutputStream(destFile,true);
BufferedWriter out = new BufferedWriter(new OutputStreamWriter(fos, "utf8"));
out.write(structure);
out.flush();
fos.close();
out.close();
}
insert into `act_ge_bytearray`(`ID_`,`REV_`,`NAME_`,`DEPLOYMENT_ID_`,`BYTES_`,`GENERATED_`) values ('37111','1','review.form---','37110',0x0d0a3c0a3c2f7461626c653e0d0a,'0');
package com.app.backup;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.List;
import java.util.Map;
import javax.sql.DataSource;
import org.junit.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.core.JdbcTemplate;
import com.app.BaseTestCase;
import com.app.backup.service.DatabaseBackupService;
import com.jcraft.jsch.JSchException;
public class DatabaseBackupTest extends BaseTestCase{
@Autowired
public DatabaseBackupService dbBackupService;
@Test
public void getDataSourceTest() throws SQLException{
DataSource ds = dbBackupService.getDataSource("192.168.12.219","3308", "root", "root", "test");
List> list = new JdbcTemplate(ds).queryForList("select 1 from sns_study_stage");
System.out.println(list);
}
@Test
public void getDataSourceBySSHTest() throws JSchException, Exception{
DataSource ds = dbBackupService.getDataSourceBySSH("192.168.12.218","3308", "root", "root", "test");
List> list = null;
try {
list = new JdbcTemplate(ds).queryForList("SELECT 1 FROM `aq_role` LIMIT 1");
} catch (Exception e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
System.out.println(list);
}
}
对于StrictHostKeyChecking="no"不明确的，在这里可以找到解释
对于SSH客户端的配置选项StrictHostKeyChecking有不明确的可以在这里找到答案。