diff --git a/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/SequenceConstant.java b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/SequenceConstant.java new file mode 100644 index 0000000..a04c2e1 --- /dev/null +++ b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/SequenceConstant.java @@ -0,0 +1,10 @@ +package com.lyms.platform.biz; + +public class SequenceConstant { + public static final String ID_KEY = "SEQUENCE_ID_KEY"; + public static final String KEY_TYPE = "SEQUENCE_KEY"; + public static final String KEY_REMARK = "SEQUENCE_INCREMENT_VALUE"; + public static final String QUEUE = "SEQUENCE_QUEUE"; + public static final String QUEUE_TYPE = "SEQUENCE_QUEUE"; + public static final String QUEUE_REMARK = "SEQUENCE_QUEUE"; +} diff --git a/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/IGenSequenceIdDao.java b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/IGenSequenceIdDao.java new file mode 100644 index 0000000..d539e0a --- /dev/null +++ b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/IGenSequenceIdDao.java @@ -0,0 +1,38 @@ +package com.lyms.platform.biz.dal; + +import java.util.List; + +public interface IGenSequenceIdDao { + + /** + * 弹出八位随机数 + * @return + */ + String poll(); + + /** + * 添加数据 + * @param list + */ + void add(List list); + + /** + * 原子性的返回当前id 并做自增操作 + * @return + */ + String next(); + + /** + * 是否生成数据 + * 定时检查剩余数据 + * 不够就生成一批数据添加 + * @return + */ + boolean check(); + + /** + * 初始检查 + */ + void init(); + +} diff --git a/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/impl/GenMongoSequenceIdDaoImpl.java b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/impl/GenMongoSequenceIdDaoImpl.java new file mode 100644 index 0000000..e6d3958 --- /dev/null +++ b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/impl/GenMongoSequenceIdDaoImpl.java @@ -0,0 +1,102 @@ +package com.lyms.platform.biz.dal.impl; + +import java.util.ArrayList; +import java.util.Date; +import java.util.List; + +import javax.annotation.PostConstruct; + +import org.apache.commons.collections.CollectionUtils; +import org.apache.commons.lang.StringUtils; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.data.mongodb.core.MongoTemplate; +import org.springframework.data.mongodb.core.query.Criteria; +import org.springframework.data.mongodb.core.query.Query; +import org.springframework.data.mongodb.core.query.Update; +import org.springframework.stereotype.Repository; + +import com.lyms.platform.biz.SequenceConstant; +import com.lyms.platform.biz.dal.IGenSequenceIdDao; +import com.lyms.platform.pojo.PlantformConfigModel; + +@Repository("genMongoSequenceIdDao") +public class GenMongoSequenceIdDaoImpl implements IGenSequenceIdDao { + + private Logger logger = LoggerFactory.getLogger(GenMongoSequenceIdDaoImpl.class); + + @Value("${sequence.index}") + private long index; + + @Value("${sequence.create.size}") + private long createSize; + + @Value("${sequence.min.size}") + private int minSize; + + @Autowired + private MongoTemplate mongoTemplate; + + @Override + @PostConstruct + public void init() { + PlantformConfigModel config = mongoTemplate.findOne(Query.query(Criteria.where("key").is(SequenceConstant.ID_KEY)), PlantformConfigModel.class); + if(config == null) { + config = new PlantformConfigModel(); + config.setCreateDate(new Date()); + config.setKey(SequenceConstant.ID_KEY); + config.setIncrementValue(index); + config.setRemark(SequenceConstant.KEY_REMARK); + config.setType(SequenceConstant.KEY_TYPE); + mongoTemplate.save(config); + logger.info("init sequence key"); + } + } + + @Override + public String poll() { + PlantformConfigModel config = mongoTemplate.findAndRemove(Query.query( + Criteria.where("key").is(SequenceConstant.QUEUE) + .and("type").is(SequenceConstant.QUEUE)).limit(1), + PlantformConfigModel.class); + if(config != null && StringUtils.isNotBlank(config.getValue())) { + return config.getValue(); + } + return null; + } + + @Override + public void add(List list) { + List configModels = new ArrayList<>(); + for (String value : list) { + PlantformConfigModel model = new PlantformConfigModel(); + model.setCreateDate(new Date()); + model.setKey(SequenceConstant.QUEUE); + model.setType(SequenceConstant.QUEUE_TYPE); + model.setRemark(SequenceConstant.QUEUE_REMARK); + model.setValue(value); + configModels.add(model); + } + mongoTemplate.insertAll(configModels); + } + + @Override + public synchronized String next() { + PlantformConfigModel plantformConfig = mongoTemplate.findAndModify(Query.query(Criteria.where("key").is(SequenceConstant.ID_KEY)), + new Update().inc("incrementValue", createSize), PlantformConfigModel.class); + if(plantformConfig != null) { + return String.valueOf(plantformConfig.getIncrementValue()); + } + return null; + } + + @Override + public boolean check() { + List datas = mongoTemplate.find(Query.query(Criteria.where("key").is(SequenceConstant.QUEUE) + .and("type").is(SequenceConstant.QUEUE)), PlantformConfigModel.class); + return CollectionUtils.isEmpty(datas) ? true : datas.size() > minSize ? false : true; + } + +} diff --git a/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/impl/GenRedisSequenceIdDaoImpl.java b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/impl/GenRedisSequenceIdDaoImpl.java new file mode 100644 index 0000000..e54d6a3 --- /dev/null +++ b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/dal/impl/GenRedisSequenceIdDaoImpl.java @@ -0,0 +1,40 @@ +package com.lyms.platform.biz.dal.impl; + +import java.util.List; + +import javax.annotation.PostConstruct; + +import org.springframework.stereotype.Repository; + +import com.lyms.platform.biz.dal.IGenSequenceIdDao; + +@Repository("genRedisSequenceIdDao") +public abstract class GenRedisSequenceIdDaoImpl implements IGenSequenceIdDao { + + @Override + public String poll() { + return null; + } + + @Override + public void add(List list) { + + } + + @Override + public String next() { + return null; + } + + @Override + public boolean check() { + return false; + } + + @Override + @PostConstruct + public void init() { + + } + +} diff --git a/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/service/GenSequenceIdService.java b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/service/GenSequenceIdService.java new file mode 100644 index 0000000..07ea67e --- /dev/null +++ b/platform-biz-patient-service/src/main/java/com/lyms/platform/biz/service/GenSequenceIdService.java @@ -0,0 +1,84 @@ +package com.lyms.platform.biz.service; + +import java.util.ArrayList; +import java.util.Collections; +import java.util.Date; +import java.util.List; + +import org.apache.commons.collections.CollectionUtils; +import org.apache.commons.lang.StringUtils; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Qualifier; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Service; + +import com.lyms.platform.biz.dal.IGenSequenceIdDao; + +@Service +public class GenSequenceIdService { + + private Logger logger = LoggerFactory.getLogger(GenSequenceIdService.class); + + @Autowired + @Qualifier("genMongoSequenceIdDao") + private IGenSequenceIdDao genSequenceIdDao; + + @Value("${sequence.create.size}") + private int createSize; + + + public String poll(String areaCode) { + String s = genSequenceIdDao.poll(); + if(StringUtils.isBlank(s)) { + generateData(); + s = genSequenceIdDao.poll(); + } + return areaCode + s; + } + + /** + * 定时扫描数据 当数据不足时 + * 自动生成一定量的数据进去 + */ + public void autoInsertId() { + if(genSequenceIdDao.check()) { + generateData(); + logger.info("generate sequence id....."); + } + } + + private synchronized void generateData() { + if(genSequenceIdDao.check()) { + List list = new ArrayList<>(); + Integer startValue = Integer.valueOf(genSequenceIdDao.next()); + Integer endValue = startValue + createSize; + for(int i = startValue; i < endValue; i++) { + list.add(String.valueOf(i)); + } + Collections.shuffle(list); + batchInsert(list, 1000); + } + } + + /** + * 批量保存数据 + * @param list 数据列表 + * @param size commit size + */ + private void batchInsert(List list, int size) { + List commitList = new ArrayList<>(); + for (int i = 0; i < list.size(); i++) { + commitList.add(list.get(i)); + if(i != 0 && commitList.size() % size == 0) { + genSequenceIdDao.add(commitList); + commitList.clear(); + } + } + if(CollectionUtils.isNotEmpty(commitList)) { + genSequenceIdDao.add(commitList); + } + } + +}