B00tLoad/SnowflakeService
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

pom.xml: OkHttp version change

Browse Source 
README.md: Updated API documentation
ConfigurationValues.java: Changed default orchestrator ip
SnowflakeService.java: Added instance name
This commit is contained in:
Alix
2024-05-17 00:13:55 +02:00
parent c52d80ff6a
commit 6c0882eb1a
4 changed files with 42 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
src/main/java/space/b00tload/services/snowflake/SnowflakeService.java
View File

@@ -4,13 +4,16 @@ import ch.qos.logback.classic.LoggerContext;
import ch.qos.logback.classic.joran.JoranConfigurator;
import ch.qos.logback.core.joran.spi.JoranException;
import ch.qos.logback.core.util.StatusPrinter2;
import okhttp3.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import space.b00tload.utils.configuration.Configuration;
import java.io.IOException;
import java.nio.file.Paths;
import java.util.List;
import java.util.Objects;
import java.util.Random;
public class SnowflakeService {
@@ -18,12 +21,14 @@ public class SnowflakeService {
private static String APPLICATION_BASE;
private static String USER_AGENT;
private static Logger LOGGER;
private static String INSTANCE_NAME;
public static void main(String[] args) {
//Set up constants
SOFTWARE_VERSION = Objects.requireNonNullElse(SnowflakeService.class.getPackage().getImplementationVersion(), "0.0.1-indev");
USER_AGENT = "SnowflakeService " + SOFTWARE_VERSION + "(" + System.getProperty("os.name") + "; " + System.getProperty("os.arch") + ") Java/" + System.getProperty("java.version");
APPLICATION_BASE = List.of(args).contains("--docker") ? Paths.get("data", "b00tload-tools", "snowflake").toString() : Paths.get(System.getProperty("user.home"), ".b00tload-tools", "snowflake").toString();
INSTANCE_NAME = getHostname() + "_" + getPid() + "_" + new Random().nextInt(9999);
//Set up logger
LoggerContext loggerContext = (LoggerContext) LoggerFactory.getILoggerFactory();
@@ -50,8 +55,24 @@ public class SnowflakeService {
long machineBits;
long epoch;
if(!Configuration.getInstance().get(ConfigurationValues.ORCHESTRATOR_IP).equals("localhost")){
if(!Configuration.getInstance().get(ConfigurationValues.ORCHESTRATOR_IP).equals(ConfigurationValues.ORCHESTRATOR_IP.getDefaultValue())){
OkHttpClient httpClient = new OkHttpClient();
try {
try (Response r = httpClient.newCall(
new Request.Builder()
.url(Configuration.getInstance().get(ConfigurationValues.ORCHESTRATOR_IP))
.post(
new FormBody.Builder().addEncoded("name", INSTANCE_NAME).build()
)
.build()
).execute()){
if(r.code() == 200){
}
}
} catch (IOException e) {
throw new RuntimeException(e);
}
} else if((Long.parseLong(Configuration.getInstance().get(ConfigurationValues.MACHINE_ID))) != -1L) {
} else {
@@ -60,4 +81,16 @@ public class SnowflakeService {
}
private static String getHostname(){
String hostname = System.getenv("COMPUTERNAME"); // On Windows
if (hostname == null || hostname.isEmpty()) {
hostname = System.getenv("HOSTNAME"); // On Unix/Linux
}
return hostname;
}
private static long getPid(){
return ProcessHandle.current().pid();
}
}