com.sharksharding.util.xml.CreateCoreXml Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of shark Show documentation
shark mysql sharding
The newest version!
/*
 * Copyright 2015-2101 gaoxianglong
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.sharksharding.util.xml;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.Marshaller;

/**
 * 生成kratos核心配置文件
 * 
 * @author gaoxianglong
 * 
 * @version 1.3.5
 */
public class CreateCoreXml {
	private String isShard;
	private String wr_index;
	private String dbRuleArray;
	private String tbRuleArray;
	private String dbSize;
	private String consistent;
	private String shardMode;
	private String sqlPath;
	private boolean isShow;
	private String tbSuffix;
	private Marshaller marshaller;

	public CreateCoreXml() {
		try {
			JAXBContext jAXBContext = JAXBContext.newInstance(Beans.class);
			marshaller = jAXBContext.createMarshaller();
			marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 生成kratos的核心信息配置文件
	 * 
	 * @author gaoxianglong
	 * 
	 * @exception Exception
	 * 
	 * @return boolean 生成结果
	 */
	public boolean createCoreXml(File savePath) {
		boolean result = false;
		try {
			/* 创建配置文件的根目录标签 */
			Beans beans = new Beans();
			/* 创建子标签 */
			Bean bean0 = new Bean();
			bean0.setId("jdbcTemplate");
			bean0.setClass_("org.springframework.jdbc.core.JdbcTemplate");
			/* 创建子标签 */
			Property dataSource = new Property();
			dataSource.setName("dataSource");
			dataSource.setRef("dataSourceGroup");
			List propertys = new ArrayList();
			propertys.add(dataSource);
			bean0.setProperty(propertys);

			Bean bean1 = new Bean();
			bean1.setId("shardRule");
			bean1.setClass_("com.sharksharding.core.shard.ShardRule");
			bean1.setInit_method("init");
			/* 创建子标签 */
			Property isShard = new Property();
			isShard.setName("isShard");
			isShard.setValue("true");
			Property wr_index = new Property();
			wr_index.setName("wr_index");
			wr_index.setValue(this.getWr_index());
			Property shardMode = new Property();
			shardMode.setName("shardMode");
			shardMode.setValue(this.isShardMode());
			Property consistent = new Property();
			consistent.setName("consistent");
			consistent.setValue(this.getConsistent());
			Property dbRuleArray = new Property();
			dbRuleArray.setName("dbRuleArray");
			dbRuleArray.setValue(this.getDbRuleArray());
			Property tbRuleArray = new Property();
			tbRuleArray.setName("tbRuleArray");
			tbRuleArray.setValue(this.getTbRuleArray());
			Property tbSuffix = new Property();
			tbSuffix.setName("tbSuffix");
			tbSuffix.setValue(this.getTbSuffix());
			propertys = new ArrayList();
			propertys.add(isShard);
			propertys.add(wr_index);
			propertys.add(shardMode);
			propertys.add(consistent);
			propertys.add(dbRuleArray);
			propertys.add(tbRuleArray);
			propertys.add(tbSuffix);
			bean1.setProperty(propertys);
			/* 创建子标签 */
			// ConstructorArg constructor_arg = new ConstructorArg();
			// constructor_arg.setName("isShard");
			// constructor_arg.setValue(this.isShard());
			// bean1.setConstructor_arg(constructor_arg);
			/* 创建子标签 */
			Bean bean2 = new Bean();
			bean2.setId("dataSourceGroup");
			bean2.setClass_("com.sharksharding.core.config.SharkDatasourceGroup");
			/* 创建子标签 */
			Property targetDataSources = new Property();
			targetDataSources.setName("targetDataSources");
			/* 创建子标签 */
			Map map = new Map();
			map.setKey_type("java.lang.Integer");
			List entrys = new ArrayList();
			for (int i = 0; i < Integer.parseInt(dbSize); i++) {
				/* 创建子标签 */
				Entry entry = new Entry();
				entry.setKey(String.valueOf(i));
				entry.setValue_ref("dataSource" + (i + 1));
				entrys.add(entry);
			}
			map.setEntry(entrys);
			targetDataSources.setMap(map);
			propertys = new ArrayList();
			propertys.add(targetDataSources);
			bean2.setProperty(propertys);
			/* 创建子标签 */
			Bean bean3 = new Bean();
			bean3.setId("sqlTemplate");
			bean3.setClass_("com.sharksharding.sql.SQLTemplate");
			/* 创建子标签 */
			ConstructorArg constructor_arg1 = new ConstructorArg();
			constructor_arg1.setName("path");
			constructor_arg1.setValue(this.getSqlPath());
			bean3.setConstructor_arg(constructor_arg1);
			List beanList = new ArrayList();
			beanList.add(bean0);
			beanList.add(bean1);
			beanList.add(bean2);
			beanList.add(bean3);
			beans.setBean(beanList);
			if (this.getIsShow())
				marshaller.marshal(beans, System.out);
			marshaller.marshal(beans, savePath);
			result = true;
		} catch (Exception e) {
			e.printStackTrace();
		}
		return result;
	}

	public boolean getIsShow() {
		return isShow;
	}

	public void setIsShow(boolean isShow) {
		this.isShow = isShow;
	}

	public String isShard() {
		return isShard;
	}

	public void setShard(String isShard) {
		this.isShard = isShard;
	}

	public String isShardMode() {
		return shardMode;
	}

	public void setShardMode(String shardMode) {
		this.shardMode = shardMode;
	}

	public String getDbRuleArray() {
		return dbRuleArray;
	}

	public void setDbRuleArray(String dbRuleArray) {
		this.dbRuleArray = dbRuleArray;
	}

	public String getTbRuleArray() {
		return tbRuleArray;
	}

	public void setTbRuleArray(String tbRuleArray) {
		this.tbRuleArray = tbRuleArray;
	}

	public String getDbSize() {
		return dbSize;
	}

	public void setDbSize(String dbSize) {
		this.dbSize = dbSize;
	}

	public Marshaller getMarshaller() {
		return marshaller;
	}

	public void setMarshaller(Marshaller marshaller) {
		this.marshaller = marshaller;
	}

	public String getConsistent() {
		return consistent;
	}

	public void setConsistent(String consistent) {
		this.consistent = consistent;
	}

	public String getSqlPath() {
		return sqlPath;
	}

	public void setSqlPath(String sqlPath) {
		this.sqlPath = sqlPath;
	}

	public String getWr_index() {
		return wr_index;
	}

	public void setWr_index(String wr_index) {
		this.wr_index = wr_index;
	}

	public String getTbSuffix() {
		return tbSuffix;
	}

	public void setTbSuffix(String tbSuffix) {
		this.tbSuffix = tbSuffix;
	}
}