2019獨角獸企業重金招聘Python工程師標準>>>

/***
?* 先根據截取字段的日期分組，再根據字段值數據按天入表
?* @author Fx_demon
?*
?*/

public class PartitionBySubstringRangeDateDay extends AbstractPartitionAlgorithm
??????? implements RuleAlgorithm {
????
??? private static final long serialVersionUID = -6211713068380831727L;

??? private static final Logger LOGGER = LoggerFactory.getLogger(PartitionBySubstringRangeDateDay.class);

??? private int startIndex; // 從第N位開始截取字符串
??? private String dateFormat;// 日期格式 ，yyMMdd或者yyyyMMdd
??? private int dateFormatLength;

??? /** 日期分組 */
??? private List<String[]> rangeDate;
??? private int rangeDateSize;

??? /** 每個分組對應的數據庫節點數量 ,逗號分隔， 配置如 3 , 6 */
??? private int[] partitionCount;
??? private PartitionByDate[] partitionByDay;


??? @Override
??? public void init() {
??????? dateFormatLength = dateFormat.length();
??????? rangeDateSize = rangeDate.size();

??????? if (rangeDateSize != partitionCount.length) {
??????????? throw new IllegalArgumentException("rangeDate not match partitionCount");
??????? }
??? }


??? @Override
??? public Integer calculate(String columnValue) {
??????? String date = columnValue.substring(startIndex, startIndex + dateFormatLength);
??????? date = date.replace("-", "");
??????? int count = 0;
??????? for (int i = 0; i < rangeDateSize; i++) {
??????? ???
??????? ??? String rangeDateStart = rangeDate.get(i)[0].replace("-", "");
??????? ??? String rangeDateEnd = rangeDate.get(i)[1].replace("-", "");
??????? ???
??????????? if (date.compareTo(rangeDateStart) >= 0 && date.compareTo(rangeDateEnd) <= 0) {
??????????? ???
??????????? ??? this.partitionByDay[i].setsBeginDate(rangeDate.get(i)[0]);
??????????????? this.partitionByDay[i].setsEndDate(rangeDate.get(i)[1]);
??????????? ???
??????????????? return partitionByDay[i].calculate(columnValue) + count;
??????????? } else {
??????????????? count += partitionCount[i];
??????????? }

??????? }
??????? return null;
??? }


??? // 在columnValue截取日期的起始記錄
??? public void setStartIndex(String startIndex) {
??????? this.startIndex = Integer.parseInt(startIndex);
??? }


??? // 數據庫節點的數量
??? public void setPartitionCount(String partitionCount) {
??????? String[] partitionCountStr = partitionCount.split(",");
??????? this.partitionCount = new int[partitionCountStr.length];
??????? this.partitionByDay = new PartitionByDate[partitionCountStr.length];
??????? for (int i = 0; i < partitionCountStr.length; i++) {
??????????? this.partitionCount[i] = Integer.parseInt(partitionCountStr[i]);
??????????? this.partitionByDay[i] = new PartitionByDate();
??????????? this.partitionByDay[i].setDateFormat(dateFormat);
??????????? this.partitionByDay[i].setsBeginDate("2017-01-01");
??????????? this.partitionByDay[i].setsPartionDay("1");
??????????? this.partitionByDay[i].init();
??????? }
??? }


??? /** 日期分組，逗號分隔如 ，【 20170101@20170331 , 20170401@20170630】 或? 【 20170101@20170331 , 2017-04-01@2017-06-30】*/
??? public void setRangeDate(String rangeDateStr) {
??????? String[] rangeDateArr = rangeDateStr.split(",");
??????? this.rangeDate = new LinkedList<>();
??????? for (String rangeDate : rangeDateArr) {
??????????? String[] date = rangeDate.trim().split("@");
??????????? date[0] = date[0].trim();
??????????? date[1] = date[1].trim();
??????????? this.rangeDate.add(date);
??????? }
??? }


??? public void setDateFormat(final String dateFormat) {
??????? if ((!"yyMMdd".equals(dateFormat) && !"yyyyMMdd".equals(dateFormat) && ( !"yyyy-MM-dd".equals(dateFormat))) ) {
??????????? throw new IllegalArgumentException("only support yyyyMMdd or yyMMdd or yyyy-MM-dd ");
??????? }
??????? this.dateFormat = dateFormat;
??? }


??? @Override
??? public int getPartitionNum() {
??????? int sum = 0;
??????? for (int i : this.partitionCount)
??????????? sum += i;
??????? return sum;
??? }

}

轉載于:https://my.oschina.net/fxdemon/blog/862634

總結

以上是生活随笔為你收集整理的【数据库中间件】MyCat分表分库规则实现的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。