package com.qianwen.core.sequence.range.db;
|
|
import javax.sql.DataSource;
|
import com.qianwen.core.sequence.exception.SeqException;
|
import com.qianwen.core.sequence.range.SeqRange;
|
import com.qianwen.core.sequence.range.SeqRangeMgr;
|
|
|
public class DbSeqRangeMgr implements SeqRangeMgr {
|
private DataSource dataSource;
|
private int step = 1;
|
private long stepStart = 0;
|
private int retryTimes = 3;
|
private String tableName = "range";
|
|
public boolean equals(final Object o) {
|
if (o == this) {
|
return true;
|
}
|
if (o instanceof DbSeqRangeMgr) {
|
DbSeqRangeMgr other = (DbSeqRangeMgr) o;
|
if (other.canEqual(this) && getStep() == other.getStep() && getStepStart() == other.getStepStart() && getRetryTimes() == other.getRetryTimes()) {
|
Object this$dataSource = getDataSource();
|
Object other$dataSource = other.getDataSource();
|
if (this$dataSource == null) {
|
if (other$dataSource != null) {
|
return false;
|
}
|
} else if (!this$dataSource.equals(other$dataSource)) {
|
return false;
|
}
|
Object this$tableName = getTableName();
|
Object other$tableName = other.getTableName();
|
return this$tableName == null ? other$tableName == null : this$tableName.equals(other$tableName);
|
}
|
return false;
|
}
|
return false;
|
}
|
|
protected boolean canEqual(final Object other) {
|
return other instanceof DbSeqRangeMgr;
|
}
|
|
public int hashCode() {
|
int result = (1 * 59) + getStep();
|
long $stepStart = getStepStart();
|
int result2 = (((result * 59) + ((int) (($stepStart >>> 32) ^ $stepStart))) * 59) + getRetryTimes();
|
Object $dataSource = getDataSource();
|
int result3 = (result2 * 59) + ($dataSource == null ? 43 : $dataSource.hashCode());
|
Object $tableName = getTableName();
|
return (result3 * 59) + ($tableName == null ? 43 : $tableName.hashCode());
|
}
|
|
public String toString() {
|
return "DbSeqRangeMgr(step=" + getStep() + ", stepStart=" + getStepStart() + ", retryTimes=" + getRetryTimes() + ", dataSource=" + getDataSource() + ", tableName=" + getTableName() + ")";
|
}
|
|
@Override // com.qianwen.core.sequence.range.SeqRangeMgr
|
public SeqRange nextRange(String name) throws SeqException {
|
Long newValue;
|
if (isEmpty(name)) {
|
throw new SecurityException("[DbSeqRangeMgr-nextRange] name is empty.");
|
}
|
for (int i = 0; i < getRetryTimes(); i++) {
|
Long oldValue = BaseDbHelper.selectRange(getDataSource(), getRealTableName(), name, getStepStart());
|
if (null != oldValue) {
|
if (oldValue.longValue() == this.stepStart) {
|
newValue = Long.valueOf(oldValue.longValue() + 1);
|
} else {
|
newValue = Long.valueOf(oldValue.longValue() + getStep());
|
}
|
if (BaseDbHelper.updateRange(getDataSource(), getRealTableName(), newValue, oldValue, name)) {
|
return new SeqRange(newValue.longValue(), newValue.longValue());
|
}
|
}
|
}
|
throw new SeqException("Retried too many times, retryTimes = " + getRetryTimes());
|
}
|
|
@Override // com.qianwen.core.sequence.range.SeqRangeMgr
|
public void init() {
|
checkParam();
|
if (BaseDbHelper.existTable(getDataSource(), getRealTableName()).booleanValue()) {
|
BaseDbHelper.createTable(getDataSource(), getRealTableName());
|
}
|
}
|
|
private String getRealTableName() {
|
return getTableName();
|
}
|
|
private boolean isEmpty(String str) {
|
return null == str || str.trim().length() == 0;
|
}
|
|
private void checkParam() {
|
if (this.step <= 0) {
|
throw new SecurityException("[DbSeqRangeMgr-checkParam] step must greater than 0.");
|
}
|
if (this.stepStart < 0) {
|
throw new SecurityException("[DbSeqRangeMgr-setStepStart] stepStart < 0.");
|
}
|
if (this.retryTimes <= 0) {
|
throw new SecurityException("[DbSeqRangeMgr-setRetryTimes] retryTimes must greater than 0.");
|
}
|
if (null == this.dataSource) {
|
throw new SecurityException("[DbSeqRangeMgr-setDataSource] dataSource is null.");
|
}
|
if (isEmpty(this.tableName)) {
|
throw new SecurityException("[DbSeqRangeMgr-setTableName] tableName is empty.");
|
}
|
}
|
|
public int getStep() {
|
return this.step;
|
}
|
|
public void setStep(int step) {
|
this.step = step;
|
}
|
|
public long getStepStart() {
|
return this.stepStart;
|
}
|
|
public void setStepStart(long stepStart) {
|
this.stepStart = stepStart;
|
}
|
|
public int getRetryTimes() {
|
return this.retryTimes;
|
}
|
|
public void setRetryTimes(int retryTimes) {
|
this.retryTimes = retryTimes;
|
}
|
|
public DataSource getDataSource() {
|
return this.dataSource;
|
}
|
|
public void setDataSource(DataSource dataSource) {
|
this.dataSource = dataSource;
|
}
|
|
public String getTableName() {
|
return this.tableName;
|
}
|
|
public void setTableName(String tableName) {
|
this.tableName = tableName;
|
}
|
}
|
