Klagarge/MinecraftElectricalAge
Archived
1
0
Fork 0
Code Issues Pull Requests Activity

implement full DatabaseConnector and do some tests

Browse Source
This commit is contained in:
Rémi Heredero
2023-05-17 20:52:03 +02:00
parent ee64e95be3
commit e306d5eacb
3 changed files with 165 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/main/java/ch/hevs/isi/MinecraftController.java
View File

@@ -1,5 +1,8 @@
package ch.hevs.isi;
import ch.hevs.isi.core.BooleanDataPoint;
import ch.hevs.isi.core.FloatDataPoint;
import ch.hevs.isi.db.DatabaseConnector;
import ch.hevs.isi.utils.Utility;
public class MinecraftController {
@@ -74,5 +77,20 @@ public class MinecraftController {
// Start coding here ...
// Initialize the database connector
if((dbProtocol != null) && (dbHostName != null)){
DatabaseConnector.url = dbProtocol+ "://" + dbHostName;
DatabaseConnector.getMySelf().initialize(dbProtocol+ "://" + dbHostName);
}
if(dbName != null){
DatabaseConnector.org = dbName;
}
if(dbUserName != null){
DatabaseConnector.bucket = dbUserName;
}
// Initialize the Modbus TCP connector
}
}

160
src/main/java/ch/hevs/isi/db/DatabaseConnector.java
View File

@@ -1,86 +1,190 @@
package ch.hevs.isi.db;
import ch.hevs.isi.MinecraftController;
import ch.hevs.isi.core.DataPoint;
import ch.hevs.isi.core.DataPointListener;
import ch.hevs.isi.core.FloatDataPoint;
import ch.hevs.isi.utils.Utility;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStreamWriter;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.ProtocolException;
import java.net.URL;
import java.util.Properties;
public class DatabaseConnector implements DataPointListener {
Properties properties = new Properties();
String default_token = System.getenv("SECRET_TOKEN");
String default_url;
String default_org;
String default_bucket;
String fullURL;
URL myURL;
HttpURLConnection con;
private Properties properties = new Properties(); // Properties of the config.properties file
private String fullURL = null; // Full URL of the InfluxDB server
private URL urlForWrite = null; // URL of the InfluxDB server
private HttpURLConnection con = null; // Connection to the InfluxDB server
private boolean initialized = false; // Boolean to know if the database connector is initialized
private TimeManager _timeManager = new TimeManager(3); // Time manager to manage the time of the data points
private long _timestamp = 0; // Timestamp of the data points
String default_token = System.getenv("SECRET_TOKEN"); // Token to access the InfluxDB server
public static String url = null; // URL of the InfluxDB server
public static String org = null; // Organization of the InfluxDB server
public static String bucket = null; // Bucket of the InfluxDB server
private static DatabaseConnector mySelf = null;
/**
* Private constructor of the database connector
* Read the config.properties file and initialize the database connector
*/
private DatabaseConnector (){
// Read the config.properties file
try (InputStream input = new FileInputStream("src/config.properties")) {
properties.load(input);
} catch (Exception e) {
e.printStackTrace();
}
default_url = properties.getProperty("URL");
default_org = properties.getProperty("ORG");
default_bucket = properties.getProperty("BUCKET");
// Get the URL, the organization and the bucket from the config.properties file if their are null
if (url == null){
url = properties.getProperty("URL");
}
if (org == null){
org = properties.getProperty("ORG");
}
if (bucket == null){
bucket = properties.getProperty("BUCKET");
}
}
/**
* Get the instance of the database connector
* @return The instance of the database connector
*/
public static DatabaseConnector getMySelf(){
if (mySelf == null){
mySelf = new DatabaseConnector();
}
return mySelf;
}
public void initialize(String url){
String org = default_org;
String bucket = default_bucket;
fullURL = url + "/api/v2/write?org=" + org + "&bucket=" + bucket;
System.out.println(fullURL);
/**
* Initialize the database connector
* @param url URL of the database. If null take the URL from the config.properties file
*/
public void initialize(String url){
try{
myURL = new URL(fullURL);
if(urlForWrite == null){
if(url == null){
url = properties.getProperty("URL");
}
fullURL = url + "/api/v2/write?org=" + org + "&bucket=" + bucket;
Utility.pDebug("URL: " + fullURL);
urlForWrite = new URL(fullURL);
}
initialized = true;
} catch (MalformedURLException e) {
throw new RuntimeException(e);
} catch (IOException e) {
throw new RuntimeException(e);
}
// If the user want to erase the previous data in the database
// Delete the previous data
if(MinecraftController.ERASE_PREVIOUS_DATA_INB_DB){
try {
// Create the URL to delete the data
fullURL = url + "/api/v2/delete?org=" + org + "&bucket=" + bucket;
Utility.pDebug("URL: " + fullURL);
URL urlForDelete = new URL(fullURL);
// Create the connection to the database
con = (HttpURLConnection) urlForDelete.openConnection();
con.setRequestMethod("POST");
con.setRequestProperty("Authorization", "Token " + default_token);
con.setRequestProperty("Content-Type", "text/plain");
con.setRequestProperty("Accept", "application/json");
con.setDoOutput(true);
} catch (MalformedURLException e) {
throw new RuntimeException(e);
} catch (ProtocolException e) {
throw new RuntimeException(e);
} catch (IOException e) {
throw new RuntimeException(e);
}
// Create the data to send to delete the data in the database and send it
String data = "{\n";
data += "\"start\": \"2000-01-01T00:00:00Z\",\n";
data += "\"stop\": \"2024-04-25T10:45:00Z\",\n";
data += "\"predicate\": \"_measurement=\\\"Minecraft\\\"\"\n";
data += "}";
Utility.pDebug(data);
sendDataToDatabase(data);
fullURL=null; // Reset the full URL
}
}
private void pushToDatabase(DataPoint dp){
String data = "measurement " + dp.toString();
/**
* Push the data point to the database
* @param dp Data point to push
*/
private void pushToDatabase(DataPoint dp) {
// Initialize the database connector if not already done
if(initialized == false){
initialize(null);
}
// Create the data to send
String data = "Minecraft ";
data += dp.toString();
data += " " + _timestamp;
// Send the data to the database
sendDataToDatabase(data);
}
private void sendDataToDatabase(String data){
try {
HttpURLConnection con = (HttpURLConnection) myURL.openConnection();
con.setRequestMethod("POST");
con.setRequestProperty("Authorization", "Token " + default_token);
con.setRequestProperty("Content-Type", "application/json");
con.setDoOutput(true);
// Create connection and set headers
if(con == null){
con = (HttpURLConnection) urlForWrite.openConnection();
con.setRequestMethod("POST");
con.setRequestProperty("Authorization", "Token " + default_token);
con.setRequestProperty("Content-Type", "application/json");
con.setDoOutput(true);
}
// Send data
OutputStreamWriter writer = new OutputStreamWriter(con.getOutputStream());
writer.write(data);
writer.flush();
// Get response code
int respondCode = con.getResponseCode();
System.out.println(dp.toString() + " -> Database. Code: " + respondCode);
if(respondCode != 204){
System.out.println("Error: " + respondCode + ", Data: " + data);
} else {
System.out.println(data + " -> Database");
}
con.disconnect();
con = null;
} catch (IOException e) {
e.printStackTrace();
}
}
/**
* Push the data point to the database
* @param dp Data point to push
*/
@Override
public void onNewValue(DataPoint dp) {
pushToDatabase(dp);
if(dp.getLabel().equals("CLOAK_FLOAT")) {
FloatDataPoint fdp = (FloatDataPoint) dp;
_timeManager.setTimestamp(fdp.getValue());
_timestamp = _timeManager.getNanosForDB();
}
if(_timestamp != 0){
pushToDatabase(dp);
}
}
}

27
src/test/java/Database.java
View File

@@ -1,3 +1,5 @@
import ch.hevs.isi.MinecraftController;
import ch.hevs.isi.core.BooleanDataPoint;
import ch.hevs.isi.core.DataPoint;
import ch.hevs.isi.core.FloatDataPoint;
import ch.hevs.isi.db.DatabaseConnector;
@@ -8,19 +10,20 @@ import java.util.Properties;
public class Database {
public static void main(String[] args) {
Properties properties = new Properties();
try (InputStream input = new FileInputStream("src/config.properties")) {
properties.load(input);
} catch (Exception e) {
e.printStackTrace();
}
String url = properties.getProperty("URL");
DatabaseConnector db = DatabaseConnector.getMySelf();
db.initialize(url);
MinecraftController.ERASE_PREVIOUS_DATA_INB_DB = true;
FloatDataPoint clock = new FloatDataPoint("CLOAK_FLOAT", false);
FloatDataPoint gridVoltage = new FloatDataPoint("GRID_U_FLOAT", true);
gridVoltage.setValue(500);
BooleanDataPoint solarPanel = new BooleanDataPoint("REMOTE_SOLAR_SW", true);
DatabaseConnector.getMySelf().initialize(null);
clock.setValue(0f);
for (float i = 0; i < 3; i += 0.1f) {
System.out.println("");
clock.setValue(i);
gridVoltage.setValue(750 + (100*i)%100);
}
}
}