-
Notifications
You must be signed in to change notification settings - Fork 4.7k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Fix][Common] rewrite code generate,fix bit shift (#6914)
* rewrite code generate,fix bit shift * fix ut * add algorithm from licenses file * add algorithm from licenses file * add algorithm from licenses file * add algorithm from licenses file * add algorithm from licenses file * fix ut
- Loading branch information
Showing
17 changed files
with
142 additions
and
139 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
74 changes: 74 additions & 0 deletions
74
...uler-common/src/main/java/org/apache/dolphinscheduler/common/utils/CodeGenerateUtils.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,74 @@ | ||
| /** Copyright 2010-2012 Twitter, Inc.*/ | ||
|
|
||
| package org.apache.dolphinscheduler.common.utils; | ||
|
|
||
| import java.net.InetAddress; | ||
| import java.net.UnknownHostException; | ||
| import java.util.Objects; | ||
|
|
||
| /** | ||
| * Rewriting based on Twitter snowflake algorithm | ||
| */ | ||
| public class CodeGenerateUtils { | ||
| // start timestamp | ||
| private static final long START_TIMESTAMP = 1609430400000L; //2021-01-01 00:00:00 | ||
| // Each machine generates 32 in the same millisecond | ||
| private static final long LOW_DIGIT_BIT = 5L; | ||
| private static final long MIDDLE_BIT = 2L; | ||
| private static final long MAX_LOW_DIGIT = ~(-1L << LOW_DIGIT_BIT); | ||
| // The displacement to the left | ||
| private static final long MIDDLE_LEFT = LOW_DIGIT_BIT; | ||
| private static final long HIGH_DIGIT_LEFT = LOW_DIGIT_BIT + MIDDLE_BIT; | ||
| private final long machineHash; | ||
| private long lowDigit = 0L; | ||
| private long recordMillisecond = -1L; | ||
|
|
||
| private static final long SYSTEM_TIMESTAMP = System.currentTimeMillis(); | ||
| private static final long SYSTEM_NANOTIME = System.nanoTime(); | ||
|
|
||
| private CodeGenerateUtils() throws CodeGenerateException { | ||
| try { | ||
| this.machineHash = Math.abs(Objects.hash(InetAddress.getLocalHost().getHostName())) % (2 << (MIDDLE_BIT - 1)); | ||
| } catch (UnknownHostException e) { | ||
| throw new CodeGenerateException(e.getMessage()); | ||
| } | ||
| } | ||
|
|
||
| private static CodeGenerateUtils instance = null; | ||
|
|
||
| public static synchronized CodeGenerateUtils getInstance() throws CodeGenerateException { | ||
| if (instance == null) { | ||
| instance = new CodeGenerateUtils(); | ||
| } | ||
| return instance; | ||
| } | ||
|
|
||
| public synchronized long genCode() throws CodeGenerateException { | ||
| long nowtMillisecond = systemMillisecond(); | ||
| if (nowtMillisecond < recordMillisecond) { | ||
| throw new CodeGenerateException("New code exception because time is set back."); | ||
| } | ||
| if (nowtMillisecond == recordMillisecond) { | ||
| lowDigit = (lowDigit + 1) & MAX_LOW_DIGIT; | ||
| if (lowDigit == 0L) { | ||
| while (nowtMillisecond <= recordMillisecond) { | ||
| nowtMillisecond = systemMillisecond(); | ||
| } | ||
| } | ||
| } else { | ||
| lowDigit = 0L; | ||
| } | ||
| recordMillisecond = nowtMillisecond; | ||
| return (nowtMillisecond - START_TIMESTAMP) << HIGH_DIGIT_LEFT | machineHash << MIDDLE_LEFT | lowDigit; | ||
| } | ||
|
|
||
| private long systemMillisecond() { | ||
| return SYSTEM_TIMESTAMP + (System.nanoTime() - SYSTEM_NANOTIME) / 1000000; | ||
| } | ||
|
|
||
| public static class CodeGenerateException extends Exception { | ||
| public CodeGenerateException(String message) { | ||
| super(message); | ||
| } | ||
| } | ||
| } |
94 changes: 0 additions & 94 deletions
94
...heduler-common/src/main/java/org/apache/dolphinscheduler/common/utils/SnowFlakeUtils.java
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.