379 lines
12 KiB
Java
379 lines
12 KiB
Java
package com.njcn.influx.deprecated;
|
||
|
||
|
||
import lombok.Data;
|
||
import lombok.extern.slf4j.Slf4j;
|
||
import okhttp3.OkHttpClient;
|
||
import org.influxdb.InfluxDB;
|
||
import org.influxdb.InfluxDB.ConsistencyLevel;
|
||
import org.influxdb.InfluxDBFactory;
|
||
import org.influxdb.dto.*;
|
||
import org.influxdb.dto.Point.Builder;
|
||
|
||
import java.time.Instant;
|
||
import java.time.LocalDateTime;
|
||
import java.time.ZoneId;
|
||
import java.time.format.DateTimeFormatter;
|
||
import java.time.format.DateTimeFormatterBuilder;
|
||
import java.time.temporal.ChronoField;
|
||
import java.util.ArrayList;
|
||
import java.util.HashMap;
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
import java.util.concurrent.TimeUnit;
|
||
|
||
/**
|
||
* 类的介绍：
|
||
*
|
||
* @author xuyang
|
||
* @version 1.0.0
|
||
* @createTime 2021/11/16 10:20
|
||
*/
|
||
@Slf4j
|
||
@Data
|
||
@Deprecated
|
||
public class InfluxDbUtils {
|
||
static OkHttpClient.Builder client = new OkHttpClient.Builder()
|
||
.connectTimeout(1000,TimeUnit.SECONDS)
|
||
.readTimeout(1000, TimeUnit.SECONDS)
|
||
.writeTimeout(1000, TimeUnit.SECONDS);
|
||
private static final int FRACTION_MIN_WIDTH = 0;
|
||
private static final int FRACTION_MAX_WIDTH = 9;
|
||
private static final boolean ADD_DECIMAL_POINT = true;
|
||
private static final DateTimeFormatter RFC3339_FORMATTER = new DateTimeFormatterBuilder()
|
||
.appendPattern("yyyy-MM-dd'T'HH:mm:ss")
|
||
.appendFraction(ChronoField.NANO_OF_SECOND, FRACTION_MIN_WIDTH, FRACTION_MAX_WIDTH, ADD_DECIMAL_POINT)
|
||
.appendZoneOrOffsetId()
|
||
.toFormatter();
|
||
|
||
/**用户名*/
|
||
private String username;
|
||
/**密码*/
|
||
private String password;
|
||
/**链接地址*/
|
||
private String openurl;
|
||
/**数据库*/
|
||
private String dbName;
|
||
/**保留策略*/
|
||
private String retentionPolicy;
|
||
|
||
private InfluxDB influxDB;
|
||
|
||
public InfluxDbUtils(String username, String password, String url, String dbName, String retentionPolicy) {
|
||
this.username = username;
|
||
this.password = password;
|
||
this.openurl = url;
|
||
this.dbName = dbName;
|
||
this.retentionPolicy = retentionPolicy == null || retentionPolicy.equals("") ? "autogen" : retentionPolicy;
|
||
influxDbBuild();
|
||
}
|
||
|
||
|
||
/**
|
||
* 连接时序数据库 ，若不存在则创建
|
||
*
|
||
* @return
|
||
*/
|
||
public InfluxDB influxDbBuild() {
|
||
if (influxDB == null) {
|
||
influxDB = InfluxDBFactory.connect(openurl, username, password,client);
|
||
}
|
||
try {
|
||
// if (!influxDB.databaseExists(database)) {
|
||
// influxDB.createDatabase(database);
|
||
// }
|
||
} catch (Exception e) {
|
||
// 该数据库可能设置动态代理，不支持创建数据库
|
||
// e.printStackTrace();
|
||
} finally {
|
||
influxDB.setRetentionPolicy(retentionPolicy);
|
||
}
|
||
influxDB.setLogLevel(InfluxDB.LogLevel.NONE);
|
||
return influxDB;
|
||
}
|
||
|
||
|
||
|
||
|
||
/**
|
||
* 创建数据库
|
||
*
|
||
* @param dbName
|
||
*/
|
||
@SuppressWarnings("deprecation")
|
||
public void createDB(String dbName) {
|
||
influxDB.createDatabase(dbName);
|
||
}
|
||
|
||
/**
|
||
* 删除数据库
|
||
*
|
||
* @param dbName
|
||
*/
|
||
@SuppressWarnings("deprecation")
|
||
public void deleteDB(String dbName) {
|
||
influxDB.deleteDatabase(dbName);
|
||
}
|
||
|
||
/**
|
||
* 测试连接是否正常
|
||
*
|
||
* @return true 正常
|
||
*/
|
||
public boolean ping() {
|
||
boolean isConnected = false;
|
||
Pong pong;
|
||
try {
|
||
pong = influxDB.ping();
|
||
if (pong != null) {
|
||
isConnected = true;
|
||
}
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
}
|
||
return isConnected;
|
||
}
|
||
|
||
/**
|
||
* 创建自定义保留策略
|
||
*
|
||
* @param policyName 策略名
|
||
* @param days 保存天数
|
||
* @param replication 保存副本数量
|
||
* @param isDefault 是否设为默认保留策略
|
||
*/
|
||
public void createRetentionPolicy(String dataBaseName, String policyName, int days, int replication,
|
||
Boolean isDefault) {
|
||
String sql = String.format("CREATE RETENTION POLICY \"%s\" ON \"%s\" DURATION %sd REPLICATION %s ", policyName,
|
||
dataBaseName, days, replication);
|
||
if (isDefault) {
|
||
sql = sql + " DEFAULT";
|
||
}
|
||
query(sql);
|
||
}
|
||
|
||
/**
|
||
* 创建默认的保留策略
|
||
*
|
||
* 策略名：hour，保存天数：30天，保存副本数量：1,设为默认保留策略
|
||
*/
|
||
public void createDefaultRetentionPolicy() {
|
||
String command = String
|
||
.format("CREATE RETENTION POLICY \"%s\" ON \"%s\" DURATION %s REPLICATION %s DEFAULT", "hour", dbName,
|
||
"30d", 1);
|
||
this.query(command);
|
||
}
|
||
|
||
/**
|
||
* 查询
|
||
*
|
||
* @param command 查询语句
|
||
* @return
|
||
*/
|
||
public QueryResult query(String command) {
|
||
return influxDB.query(new Query(command, dbName));
|
||
}
|
||
|
||
/**
|
||
* 插入
|
||
*
|
||
* @param measurement 表
|
||
* @param tags 标签
|
||
* @param fields 字段
|
||
*/
|
||
public void insert(String measurement, Map<String, String> tags, Map<String, Object> fields, long time,
|
||
TimeUnit timeUnit) {
|
||
Builder builder = Point.measurement(measurement);
|
||
builder.tag(tags);
|
||
builder.fields(fields);
|
||
if (0 != time) {
|
||
builder.time(time, timeUnit);
|
||
}
|
||
influxDB.write(dbName, retentionPolicy, builder.build());
|
||
}
|
||
|
||
|
||
|
||
/**
|
||
* 批量写入测点
|
||
*
|
||
* @param batchPoints
|
||
*/
|
||
public void batchInsert(BatchPoints batchPoints, TimeUnit timeUnit) {
|
||
influxDB.write(batchPoints);
|
||
// influxDB.enableGzip();
|
||
// influxDB.enableBatch(2000,100,timeUnit);
|
||
// influxDB.disableGzip();
|
||
// influxDB.disableBatch();
|
||
}
|
||
|
||
/**
|
||
* 批量写入数据
|
||
*
|
||
* @param database 数据库
|
||
* @param retentionPolicy 保存策略
|
||
* @param consistency 一致性
|
||
* @param records 要保存的数据（调用BatchPoints.lineProtocol()可得到一条record）
|
||
*/
|
||
public void batchInsert(final String database, final String retentionPolicy, final ConsistencyLevel consistency,TimeUnit timeUnit, final List<String> records) {
|
||
influxDB.write(database, retentionPolicy, consistency, records);
|
||
}
|
||
|
||
/**
|
||
* 批量写入数据
|
||
*
|
||
* @param database 数据库
|
||
* @param retentionPolicy 保存策略
|
||
* @param consistency 一致性
|
||
* @param records 要保存的数据（调用BatchPoints.lineProtocol()可得到一条record）
|
||
*/
|
||
public void batchInsert(final String database, final String retentionPolicy, final ConsistencyLevel consistency, final List<String> records) {
|
||
influxDB.write(database, retentionPolicy, consistency, records);
|
||
}
|
||
|
||
|
||
/**
|
||
* 删除
|
||
*
|
||
* @param command 删除语句
|
||
* @return 返回错误信息
|
||
*/
|
||
public String deleteMeasurementData(String command) {
|
||
QueryResult result = influxDB.query(new Query(command, dbName));
|
||
return result.getError();
|
||
}
|
||
|
||
/**
|
||
* 关闭数据库
|
||
*/
|
||
public void close() {
|
||
influxDB.close();
|
||
}
|
||
|
||
/**
|
||
* 构建Point
|
||
*
|
||
* @param measurement
|
||
* @param time
|
||
* @param fields
|
||
* @return
|
||
*/
|
||
public Point pointBuilder(String measurement, long time, TimeUnit timeUnit, Map<String, String> tags,
|
||
Map<String, Object> fields) {
|
||
Point point = Point.measurement(measurement).time(time, timeUnit).tag(tags).fields(fields).build();
|
||
return point;
|
||
}
|
||
|
||
|
||
|
||
|
||
|
||
//单条查询语句封装
|
||
/**
|
||
* 查询结果封装到map
|
||
* @param commond 单条sql语句
|
||
* @return 查询结果
|
||
*/
|
||
@Deprecated
|
||
public List<Map<String, Object>> getResult(String commond){
|
||
List<Map<String, Object>> retList = new ArrayList<>();
|
||
QueryResult queryResult = influxDB.query(new Query(commond,dbName));
|
||
List<QueryResult.Result> results = queryResult.getResults();
|
||
if (results==null||results.isEmpty()){
|
||
return retList;
|
||
}
|
||
QueryResult.Result result = results.get(0);
|
||
List<QueryResult.Series> seriess = result.getSeries();
|
||
if (seriess==null||seriess.isEmpty()){
|
||
return retList;
|
||
}
|
||
QueryResult.Series series = seriess.get(0);
|
||
List<String> columns = series.getColumns();
|
||
List<List<Object>> values = series.getValues();
|
||
for (List<Object> columnValue :values){
|
||
Map<String, Object> map = new HashMap<>(1);
|
||
for (int i=0;i<columnValue.size();i++){
|
||
if(columns.get(i).equals("time")){
|
||
long aa = Instant.parse(columnValue.get(i).toString()).minusMillis(TimeUnit.HOURS.toMillis(8)).getEpochSecond();
|
||
map.put(columns.get(i), aa);
|
||
}else {
|
||
map.put(columns.get(i),columnValue.get(i));
|
||
}
|
||
}
|
||
retList.add(map);
|
||
}
|
||
return retList;
|
||
}
|
||
|
||
@Deprecated
|
||
public List<Map<String, Object>> getResult(String commond, String type){
|
||
|
||
List<Map<String, Object>> retList = new ArrayList<>();
|
||
QueryResult queryResult = influxDB.query(new Query(commond,dbName));
|
||
List<QueryResult.Result> results = queryResult.getResults();
|
||
if (results==null||results.isEmpty()){
|
||
return retList;
|
||
}
|
||
QueryResult.Result result = results.get(0);
|
||
List<QueryResult.Series> seriess = result.getSeries();
|
||
if (seriess==null||seriess.isEmpty()){
|
||
return retList;
|
||
}
|
||
QueryResult.Series series = seriess.get(0);
|
||
List<String> columns = series.getColumns();
|
||
List<List<Object>> values = series.getValues();
|
||
for (List<Object> columnValue :values){
|
||
Map<String, Object> map = new HashMap<>(1);
|
||
for (int i=0;i<columnValue.size();i++){
|
||
if(columns.get(i).equals("time")){
|
||
Instant aa = Instant.parse(columnValue.get(i).toString()).minusMillis(TimeUnit.HOURS.toMillis(8));
|
||
LocalDateTime localDateTime =LocalDateTime.ofInstant(aa,ZoneId.systemDefault());
|
||
String time = localDateTime.format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
|
||
map.put(columns.get(i), time);
|
||
}else {
|
||
map.put(columns.get(i),columnValue.get(i));
|
||
}
|
||
}
|
||
retList.add(map);
|
||
}
|
||
return retList;
|
||
}
|
||
|
||
|
||
public List<Map<String, Object>> getMapResult(String commond){
|
||
|
||
List<Map<String, Object>> retList = new ArrayList<>();
|
||
QueryResult queryResult = influxDB.query(new Query(commond,dbName));
|
||
List<QueryResult.Result> results = queryResult.getResults();
|
||
if (results==null||results.isEmpty()){
|
||
return retList;
|
||
}
|
||
QueryResult.Result result = results.get(0);
|
||
List<QueryResult.Series> seriess = result.getSeries();
|
||
if (seriess==null||seriess.isEmpty()){
|
||
return retList;
|
||
}
|
||
QueryResult.Series series = seriess.get(0);
|
||
List<String> columns = series.getColumns();
|
||
List<List<Object>> values = series.getValues();
|
||
for (List<Object> columnValue :values){
|
||
Map<String, Object> map = new HashMap<>(1);
|
||
for (int i=0;i<columnValue.size();i++){
|
||
if(columns.get(i).equals("time")){
|
||
Instant instant = Instant.from(RFC3339_FORMATTER.parse(String.valueOf(columnValue.get(i))));
|
||
LocalDateTime localDateTime =LocalDateTime.ofInstant(instant,ZoneId.systemDefault());
|
||
String time = localDateTime.format(DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss"));
|
||
map.put(columns.get(i), time);
|
||
}else {
|
||
map.put(columns.get(i),columnValue.get(i));
|
||
}
|
||
}
|
||
retList.add(map);
|
||
}
|
||
return retList;
|
||
}
|
||
|
||
|
||
|
||
}
|