时间滚动的代码实现
精度
精度的抽像类
import org.apache.shardingsphere.api.sharding.standard.PreciseShardingAlgorithm;
import org.apache.shardingsphere.api.sharding.standard.PreciseShardingValue;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.Collection;
import java.util.Date;
public abstract class RollDayPreciseAlgorithm implements PreciseShardingAlgorithm {
public abstract String getSuffix();
@Override
public String doSharding(Collection availableTargetNames, PreciseShardingValue shardingValue) {
Object obj = shardingValue.getValue();
String result;
if (obj instanceof Date){
result = shardingValue.getLogicTableName() + eval(getSuffix(),(Date)obj);
} else if (obj instanceof String) {
if ( !RollDayUtils.checkDateStr((String)obj)) {
throw new RuntimeException("无法识别的日期类型,源于自符串");
}
SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyyMMdd");
Date date;
try {
date = simpleDateFormat.parse(((String) obj).replaceAll("-",""));
} catch (ParseException e) {
throw new RuntimeException("无法识别的日期类型");
}
result = shardingValue.getLogicTableName() + eval(getSuffix(),date);
} else {
throw new RuntimeException("无法识别的日期类型");
}
return result;
}
private String eval(String suffixFormat , Date date) {
SimpleDateFormat simpleDateFormat = new SimpleDateFormat(suffixFormat);
return simpleDateFormat.format(date);
}
}
实现类精度DayOfMonthPreciseAlgorithm
t_log_dd 即31张表
package com.xxx.xxx.rule.scene.rollday.dayofmonth;
import com.xxx.xxx.rule.scene.rollday.RollDayPreciseAlgorithm;
public class DayOfMonthPreciseAlgorithm extends RollDayPreciseAlgorithm {
@Override
public String getSuffix() {
return "_dd";
}
}
实现类精度DayOfYearPreciseAlgorithm
t_log_MMdd 即366张表
package com.xxx.xxx.rule.scene.rollday.dayofyear;
import com.xxx.xxx.rule.scene.rollday.RollDayPreciseAlgorithm;
public class DayOfYearPreciseAlgorithm extends RollDayPreciseAlgorithm {
@Override
public String getSuffix() {
return "_MMdd";
}
}
实现类精度MonthOfYearPreciseAlgorithm
即t_log_MM 即12张表
package com.xxx.xxx.rule.scene.rollday.monthofyear;
import com.xxx.xxx.rule.scene.rollday.RollDayPreciseAlgorithm;
public class MonthOfYearPreciseAlgorithm extends RollDayPreciseAlgorithm {
@Override
public String getSuffix() {
return "_MM";
}
}
范围
范围抽像类
入口为 doSharding
package com.xxx.xxx.rule.scene.rollday;
import com.google.common.collect.BoundType;
import com.google.common.collect.Range;
import org.apache.shardingsphere.api.sharding.standard.RangeShardingAlgorithm;
import org.apache.shardingsphere.api.sharding.standard.RangeShardingValue;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.*;
public abstract class RollDayRangeAlgorithm implements RangeShardingAlgorithm {
protected int addUnit = Calendar.DATE;
public abstract Collection<String> getAll(String logicTableName);
public abstract String assemblyTable(Date date , String logicTableName);
private Collection<String> get4String(String logicTableName , Range<String> range) {
String dateStart = range.lowerEndpoint();
String dateEnd = range.upperEndpoint();
if ( !RollDayUtils.checkDateStr(dateStart) || !RollDayUtils.checkDateStr(dateEnd) ) {
throw new RuntimeException("无法识别的日期类型");
}
dateStart = dateStart.replaceAll("-","");
dateEnd = dateEnd.replaceAll("-","");
SimpleDateFormat sf = new SimpleDateFormat("yyyyMMdd");
try {
Date start = sf.parse(dateStart);
Date end = sf.parse(dateEnd);
return getByDate(logicTableName,start,end);
} catch (ParseException e) {
throw new RuntimeException("无法识别的日期类型");
}
}
private Collection<String> getByDate(String logicTableName , Date start ,Date end){
Calendar tempStart = Calendar.getInstance();
tempStart.setTime(start);
Calendar tempEnd = Calendar.getInstance();
tempEnd.setTime(end);
Collection<String> result = new HashSet<String>();
while (tempStart.before(tempEnd)) {//两个日期的天数
result.add(assemblyTable(tempStart.getTime(),logicTableName));
tempStart.add(addUnit, 1);
}
// add 考虑到时间类型有可能含有时分秒等元素,这里要将upper的值再次加入到集合中
{
result.add(assemblyTable(end,logicTableName));
}
return result;
}
private Collection<String> get4Date(String logicTableName , Range<Date> range) {
Date dateStart = range.lowerEndpoint();
Date dateEnd = range.upperEndpoint();
return getByDate(logicTableName,dateStart,dateEnd);
}
public Collection<String> getSuffix(String logicTableName, Range range) {
if (range.upperEndpoint() instanceof Date) {
return get4Date(logicTableName,range);
} else if (range.upperEndpoint() instanceof String) {
return get4String(logicTableName,range);
} else {
throw new RuntimeException("不支持的格式化类型,请检查SQL参数");
}
}
public Collection<String> getSuffix4OnlyUpper(String logicTableName, Object value, BoundType type) {
return getAll(logicTableName);
}
public Collection<String> getSuffix4OnlyLower(String logicTableName, Object value, BoundType type) {
return getAll(logicTableName);
}
public Collection<String> doSharding(Collection availableTargetNames, RangeShardingValue shardingValue) {
return doSharding0(shardingValue.getLogicTableName(),shardingValue.getValueRange());
}
public String doSharding1(String logicTableName, Object shardingValue) {
if (shardingValue instanceof Date) {
return assemblyTable((Date)shardingValue,logicTableName);
} else if (shardingValue instanceof String) {
if ( !RollDayUtils.checkDateStr((String)shardingValue) ) {
throw new RuntimeException("无法识别的日期类型");
}
String nShardingValue = ((String)shardingValue).replaceAll("-","");
SimpleDateFormat sf = new SimpleDateFormat("yyyyMMdd");
Date shardingDateValue = null;
try {
shardingDateValue = sf.parse(nShardingValue);
} catch (ParseException e) {
throw new RuntimeException("无法识别的日期类型");
}
return assemblyTable(shardingDateValue,logicTableName);
} else {
throw new RuntimeException("不支持的格式化类型,请检查SQL参数");
}
}
public Collection<String> doSharding0(String logicTableName, Range shardingValue) {
Range range = shardingValue;
Object lower = range.hasLowerBound() ? range.lowerEndpoint() : null;
Object upper = range.hasUpperBound() ? range.upperEndpoint() : null;
if (lower==null && upper==null ) {
return new HashSet();
} else if (lower==null) {
return getSuffix4OnlyUpper(logicTableName,upper,range.upperBoundType());
} else if (upper==null) {
return getSuffix4OnlyLower(logicTableName,lower,range.lowerBoundType());
}
return getSuffix(logicTableName,range);
}
public static void main(String[] args) {
Range r = Range.lessThan(10);
System.out.println(r.hasUpperBound());
System.out.println(r.hasLowerBound());
System.out.println(r.upperBoundType());
}
}
范围实现类DayOfMonthRangeAlgorithm
t_log_dd 即31张表
package com.xxx.xxx.rule.scene.rollday.dayofmonth;
import com.psbc.dmf.rule.scene.rollday.RollDayRangeAlgorithm;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Collection;
import java.util.Date;
import java.util.HashSet;
public class DayOfMonthRangeAlgorithm extends RollDayRangeAlgorithm {
public DayOfMonthRangeAlgorithm() {
this.addUnit = Calendar.DATE;
}
private String suffixFormat = "_dd";
public Collection<String> getAll(String logicTableName) {
String fs = suffixFormat.replaceAll("dd","%02d");
Collection<String> result = new HashSet<String>();
for (int i=1;i<=31;i++) {
result.add(logicTableName+String.format(fs,i));
}
return result;
}
public String assemblyTable(Date date , String logicTableName) {
String fromatStr = suffixFormat.replaceAll("dd","%s");
SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyyMMdd");
String temp = simpleDateFormat.format(date);
String MM = temp.substring(6);
return logicTableName + String.format(fromatStr,MM);
}
}
范围实现类DayOfYearRangeAlgorithm
t_log_MMdd即366张表
package com.psbc.dmf.rule.scene.rollday.dayofyear;
import com.psbc.dmf.rule.scene.rollday.RollDayRangeAlgorithm;
import java.text.SimpleDateFormat;
import java.util.*;
public class DayOfYearRangeAlgorithm extends RollDayRangeAlgorithm {
public DayOfYearRangeAlgorithm() {
this.addUnit = Calendar.DATE;
}
private String suffixFormat = "_MMdd";
public Collection<String> getAll(String logicTableName) {
String fs = suffixFormat.replaceAll("dd","%02d").replaceAll("MM","%02d");
Collection<String> result = new HashSet<String>();
for (int i=1;i<=12;i++) {
int days = 0;
switch(i) {
case 2:
days = 29;
break;
case 4:
days = 30;
break;
case 6:
days = 30;
break;
case 9:
days = 30;
break;
case 11:
days = 30;
break;
default:
days = 31;
}
for (int j=1;j<=days;j++) {
result.add(logicTableName+String.format(fs,i,j));
}
}
return result;
}
public String assemblyTable(Date date , String logicTableName) {
String fromatStr = suffixFormat.replaceAll("dd","%s").replaceAll("MM","%s");
SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyyMMdd");
String temp = simpleDateFormat.format(date);
String MM = temp.substring(4,6);
String dd = temp.substring(6);
return logicTableName + String.format(fromatStr,MM,dd);
}
}
范围MonthOfYearRangeAlgorithm
t_log_MM 即12张表
package com.xxx.xxx.rule.scene.rollday.monthofyear;
import com.xxx.xxx.rule.scene.rollday.RollDayRangeAlgorithm;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Collection;
import java.util.Date;
import java.util.HashSet;
public class MonthOfYearRangeAlgorithm extends RollDayRangeAlgorithm {
public MonthOfYearRangeAlgorithm() {
this.addUnit = Calendar.MONTH;
}
private String suffixFormat = "_MM";
public Collection<String> getAll(String logicTableName) {
String fs = suffixFormat.replaceAll("MM","%02d");
Collection<String> result = new HashSet<String>();
for (int i=1;i<=12;i++) {
result.add(logicTableName+String.format(fs,i));
}
return result;
}
public String assemblyTable(Date date , String logicTableName) {
String fromatStr = suffixFormat.replaceAll("MM","%s");
SimpleDateFormat simpleDateFormat = new SimpleDateFormat("yyyyMMdd");
String temp = simpleDateFormat.format(date);
String MM = temp.substring(4,6);
return logicTableName + String.format(fromatStr,MM);
}
}