IdGenerator.java 1.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869
  1. package org.maxkey.util;
  2. public class IdGenerator {
  3. String strategy = "uuid";
  4. int datacenterId;
  5. int machineId;
  6. SnowFlakeId snowFlakeId = new SnowFlakeId(0,0);
  7. StringGenerator stringGenerator = new StringGenerator();
  8. public String generate(){
  9. if(strategy.equalsIgnoreCase("uuid")) {
  10. return stringGenerator.uuidGenerate();
  11. }else if(strategy.equalsIgnoreCase("SnowFlake")) {
  12. return snowFlakeId.nextId()+"";
  13. }else {
  14. return stringGenerator.randomGenerate();
  15. }
  16. }
  17. public IdGenerator() {
  18. super();
  19. }
  20. public IdGenerator(String strategy) {
  21. super();
  22. this.strategy = strategy;
  23. }
  24. public int getDatacenterId() {
  25. return datacenterId;
  26. }
  27. public void setDatacenterId(int datacenterId) {
  28. this.datacenterId = datacenterId;
  29. }
  30. public int getMachineId() {
  31. return machineId;
  32. }
  33. public void setMachineId(int machineId) {
  34. this.machineId = machineId;
  35. }
  36. public SnowFlakeId getSnowFlakeId() {
  37. return snowFlakeId;
  38. }
  39. public void setSnowFlakeId(SnowFlakeId snowFlakeId) {
  40. this.snowFlakeId = snowFlakeId;
  41. }
  42. public StringGenerator getStringGenerator() {
  43. return stringGenerator;
  44. }
  45. public void setStringGenerator(StringGenerator stringGenerator) {
  46. this.stringGenerator = stringGenerator;
  47. }
  48. }