时间滚动的代码实现

精度

精度的抽像类

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);
    }
}