gfwleak/galaxy-platform-algorithm-snowflake
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新部分位置代码

Browse Source
This commit is contained in:
qidaijie
2019-08-09 16:50:12 +08:00
parent 739c8fdb6e
commit 0da2d582e0
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
SnowFlake/SnowflakeId.java
View File

@@ -105,11 +105,12 @@ public class SnowflakeId {
if (tmpWorkerId > maxWorkerId || tmpWorkerId < 0) { if (tmpWorkerId > maxWorkerId || tmpWorkerId < 0) {
throw new IllegalArgumentException(String.format("worker Id can't be greater than %d or less than 0", maxWorkerId)); throw new IllegalArgumentException(String.format("worker Id can't be greater than %d or less than 0", maxWorkerId));
} }
if (FlowWriteConfig.DATA_CENTER_ID_NUM > maxDataCenterId || FlowWriteConfig.DATA_CENTER_ID_NUM < 0) { int dataCenterId = FlowWriteConfig.DATA_CENTER_ID_NUM;
if (dataCenterId > maxDataCenterId || dataCenterId < 0) {
throw new IllegalArgumentException(String.format("datacenter Id can't be greater than %d or less than 0", maxDataCenterId)); throw new IllegalArgumentException(String.format("datacenter Id can't be greater than %d or less than 0", maxDataCenterId));
} }
this.workerId = tmpWorkerId; this.workerId = tmpWorkerId;
this.dataCenterId = FlowWriteConfig.DATA_CENTER_ID_NUM; this.dataCenterId = dataCenterId;
} }
// ==============================Methods========================================== // ==============================Methods==========================================

9
Zookeeper/ZookeeperUtils.java
View File

@@ -41,7 +41,7 @@ public class ZookeeperUtils implements Watcher {
createNode("/Snowflake/" + FlowWriteConfig.KAFKA_TOPIC, "0".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE); createNode("/Snowflake/" + FlowWriteConfig.KAFKA_TOPIC, "0".getBytes(), ZooDefs.Ids.OPEN_ACL_UNSAFE);
int workerId; int workerId;
try { try {
connectZookeeper(FlowWriteConfig.ZOOKEEPER_SERVERS); connectZookeeper();
Stat stat = zookeeper.exists(path, true); Stat stat = zookeeper.exists(path, true);
workerId = Integer.parseInt(getNodeDate(path)); workerId = Integer.parseInt(getNodeDate(path));
if (workerId > 55) { if (workerId > 55) {
@@ -68,11 +68,10 @@ public class ZookeeperUtils implements Watcher {
/** /**
* 连接zookeeper * 连接zookeeper
* *
* @param host 地址
*/ */
private void connectZookeeper(String host) { private void connectZookeeper() {
try { try {
zookeeper = new ZooKeeper(host, SESSION_TIME_OUT, this); zookeeper = new ZooKeeper(FlowWriteConfig.ZOOKEEPER_SERVERS, SESSION_TIME_OUT, this);
countDownLatch.await(); countDownLatch.await();
} catch (IOException | InterruptedException e) { } catch (IOException | InterruptedException e) {
e.printStackTrace(); e.printStackTrace();
@@ -118,7 +117,7 @@ public class ZookeeperUtils implements Watcher {
*/ */
private void createNode(String path, byte[] date, List<ACL> acls) { private void createNode(String path, byte[] date, List<ACL> acls) {
try { try {
connectZookeeper(FlowWriteConfig.ZOOKEEPER_SERVERS); connectZookeeper();
Stat exists = zookeeper.exists(path, true); Stat exists = zookeeper.exists(path, true);
if (exists == null) { if (exists == null) {
zookeeper.create(path, date, acls, CreateMode.PERSISTENT); zookeeper.create(path, date, acls, CreateMode.PERSISTENT);