InfluxDB 的JAVA调用

创建Spring Boot的Maven工程

导入Spring Boot相关包

<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter</artifactId>
</dependency>
<dependency>
	<groupId>org.springframework.boot</groupId>
	<artifactId>spring-boot-starter-test</artifactId>
	<scope>test</scope>
</dependency>
加入InfluxDB的相关包

下图是InfluxDB的jar包,及相关依赖包。
在这里插入图片描述

pom.xml文件

<dependency>
	<groupId>org.influxdb</groupId>
	<artifactId>influxdb-java</artifactId>
	<version>2.5</version>
</dependency>
<dependency>
	<groupId>com.squareup.retrofit2</groupId>
	<artifactId>converter-gson</artifactId>
	<version>2.2.0</version>
</dependency>
<dependency>
	<groupId>com.squareup.retrofit2</groupId>
	<artifactId>converter-moshi</artifactId>
	<version>2.1.0</version>
</dependency>
<dependency>
	<groupId>com.squareup.retrofit2</groupId>
	<artifactId>retrofit</artifactId>
	<version>2.2.0</version>
</dependency>
<dependency>
	<groupId>com.squareup.okhttp3</groupId>
	<artifactId>logging-interceptor</artifactId>
	<version>3.8.0</version>
</dependency>
<dependency>
	<groupId>com.squareup.okhttp3</groupId>
	<artifactId>okhttp</artifactId>
	<version>3.8.0</version>
</dependency>
<dependency>
	<groupId>com.squareup.okio</groupId>
	<artifactId>okio</artifactId>
	<version>1.11.0</version>
</dependency>
<dependency>
	<groupId>com.google.code.gson</groupId>
	<artifactId>gson</artifactId>
	<version>2.8.0</version>
</dependency>
<dependency>
	<groupId>com.squareup.moshi</groupId>
	<artifactId>moshi</artifactId>
	<version>1.1.0</version>
</dependency>
<dependency>
	<groupId>com.google.guava</groupId>
	<artifactId>guava</artifactId>
	<version>18.0</version>
</dependency>
<dependency>
	<groupId>com.google.collections</groupId>
	<artifactId>google-collections</artifactId>
	<version>1.0-rc1</version>
</dependency>
JAVA 相关操作
  • 实体类

    package com.influxdb.pojo;
     
    import java.io.Serializable;
     
    public class CodeInfo implements Serializable{
     
    	private static final long serialVersionUID = 1L;
    	
    	private Long id;
    	private String name;
    	private String code;
    	private String descr;
    	private String descrE;
    	private String createdBy;
    	private Long createdAt;
    	
    	private String time;
    	private String tagCode;
    	private String tagName;
     
    	public Long getId() {
    		return id;
    	}
     
    	public void setId(Long id) {
    		this.id = id;
    	}
     
    	public String getName() {
    		return name;
    	}
     
    	public void setName(String name) {
    		this.name = name;
    	}
     
    	public String getCode() {
    		return code;
    	}
     
    	public void setCode(String code) {
    		this.code = code;
    	}
     
    	public String getDescr() {
    		return descr;
    	}
     
    	public void setDescr(String descr) {
    		this.descr = descr;
    	}
     
    	public String getDescrE() {
    		return descrE;
    	}
     
    	public void setDescrE(String descrE) {
    		this.descrE = descrE;
    	}
     
    	public String getCreatedBy() {
    		return createdBy;
    	}
     
    	public void setCreatedBy(String createdBy) {
    		this.createdBy = createdBy;
    	}
     
    	public Long getCreatedAt() {
    		return createdAt;
    	}
     
    	public void setCreatedAt(Long createdAt) {
    		this.createdAt = createdAt;
    	}
     
    	public String getTime() {
    		return time;
    	}
     
    	public void setTime(String time) {
    		this.time = time;
    	}
     
    	public String getTagCode() {
    		return tagCode;
    	}
     
    	public void setTagCode(String tagCode) {
    		this.tagCode = tagCode;
    	}
     
    	public String getTagName() {
    		return tagName;
    	}
     
    	public void setTagName(String tagName) {
    		this.tagName = tagName;
    	}
     
    }
    
  • 连接、方法类

    package com.influxdb.test;
     
    import java.util.Map;
     
    import org.influxdb.InfluxDB;
    import org.influxdb.InfluxDBFactory;
    import org.influxdb.dto.Point;
    import org.influxdb.dto.Point.Builder;
    import org.influxdb.dto.Query;
    import org.influxdb.dto.QueryResult;
    import org.springframework.util.StringUtils;
     
    /**
     * 时序数据库 InfluxDB 连接
     * @author Dai_LW
     *
     */
    public class InfluxDBConnect {
    	
    	private String username;//用户名
    	private String password;//密码
    	private String openurl;//连接地址
    	private String database;//数据库
    	
    	private InfluxDB influxDB;
    	
    	
    	public InfluxDBConnect(String username, String password, String openurl, String database){
    		this.username = username;
    		this.password = password;
    		this.openurl = openurl;
    		this.database = database;
    	}
    	
    	/**连接时序数据库;获得InfluxDB**/
    	public InfluxDB  influxDbBuild(){
    		if(influxDB == null){
    			influxDB = InfluxDBFactory.connect(openurl, username, password);
    			influxDB.createDatabase(database);
    			
    		}
    		return influxDB;
    	}
    	
    	/**
    	 * 设置数据保存策略
    	 * defalut 策略名 /database 数据库名/ 30d 数据保存时限30天/ 1  副本个数为1/ 结尾DEFAULT 表示 设为默认的策略
    	 */
    	public void createRetentionPolicy(){
    		String command = String.format("CREATE RETENTION POLICY \"%s\" ON \"%s\" DURATION %s REPLICATION %s DEFAULT",
                    "defalut", database, "30d", 1);
    		this.query(command);
    	}
    	
    	/**
    	 * 查询
    	 * @param command 查询语句
    	 * @return
    	 */
    	public QueryResult query(String command){
    		return influxDB.query(new Query(command, database));
    	}
    	
    	/**
    	 * 插入
    	 * @param measurement 表
    	 * @param tags 标签
    	 * @param fields 字段
    	 */
    	public void insert(String measurement, Map<String, String> tags, Map<String, Object> fields){
    		Builder builder = Point.measurement(measurement);
    		builder.tag(tags);
    		builder.fields(fields);
    		
    		influxDB.write(database, "", builder.build());
    	}
    	
    	/**
    	 * 删除
    	 * @param command 删除语句
    	 * @return 返回错误信息
    	 */
    	public String deleteMeasurementData(String command){
    		QueryResult result = influxDB.query(new Query(command, database));
    		return result.getError();
    	}
    	
    	/**
    	 * 创建数据库
    	 * @param dbName
    	 */
    	public void createDB(String dbName){
    		influxDB.createDatabase(dbName);
    	}
    	
    	/**
    	 * 删除数据库
    	 * @param dbName
    	 */
    	public void deleteDB(String dbName){
    		influxDB.deleteDatabase(dbName);
    	}
    	
    	public String getUsername() {
    		return username;
    	}
     
    	public void setUsername(String username) {
    		this.username = username;
    	}
     
    	public String getPassword() {
    		return password;
    	}
     
    	public void setPassword(String password) {
    		this.password = password;
    	}
     
    	public String getOpenurl() {
    		return openurl;
    	}
     
    	public void setOpenurl(String openurl) {
    		this.openurl = openurl;
    	}
     
    	public void setDatabase(String database) {
    		this.database = database;
    	}
    }
    
  • 测试类

    package com.influxdb.test;
     
    import java.util.ArrayList;
    import java.util.Date;
    import java.util.HashMap;
    import java.util.List;
    import java.util.Map;
     
    import junit.framework.Assert;
     
    import org.influxdb.dto.QueryResult;
    import org.influxdb.dto.QueryResult.Result;
    import org.influxdb.dto.QueryResult.Series;
    import org.junit.Before;
    import org.junit.Test;
    import org.springframework.beans.BeanWrapperImpl;
     
    import com.influxdb.pojo.CodeInfo;
     
    public class InfluxDBTest {
    	
    	private InfluxDBConnect influxDB;
    	private String username = "admin";//用户名
    	private String password = "admin";//密码
    	private String openurl = "http://127.0.0.1:8086";//连接地址
    	private String database = "test_db";//数据库
    	private String measurement = "sys_code";
    	
    	@Before
    	public void setUp(){
    		//创建 连接
    		influxDB = new InfluxDBConnect(username, password, openurl, database);
    		
    		influxDB.influxDbBuild();
    		
    		influxDB.createRetentionPolicy();
    		
    //		influxDB.deleteDB(database);
    //		influxDB.createDB(database);
    	}
     
    	@Test
    	public void testInsert(){//测试数据插入
    		Map<String, String> tags = new HashMap<String, String>();
    		Map<String, Object> fields = new HashMap<String, Object>();
    		List<CodeInfo> list = new ArrayList<CodeInfo>();
    		
    		CodeInfo info1 = new CodeInfo();
    		info1.setId(1L);
    		info1.setName("BANKS");
    		info1.setCode("ABC");
    		info1.setDescr("中国农业银行");
    		info1.setDescrE("ABC");
    		info1.setCreatedBy("system");
    		info1.setCreatedAt(new Date().getTime());
    		
    		CodeInfo info2 = new CodeInfo();
    		info2.setId(2L);
    		info2.setName("BANKS");
    		info2.setCode("CCB");
    		info2.setDescr("中国建设银行");
    		info2.setDescrE("CCB");
    		info2.setCreatedBy("system");
    		info2.setCreatedAt(new Date().getTime());
    		
    		list.add(info1);
    		list.add(info2);
    		
    		for(CodeInfo info : list){
    			
    			tags.put("TAG_CODE", info.getCode());
    			tags.put("TAG_NAME", info.getName());
    			
    			fields.put("ID", info.getId());
    			fields.put("NAME", info.getName());
    			fields.put("CODE", info.getCode());
    			fields.put("DESCR", info.getDescr());
    			fields.put("DESCR_E", info.getDescrE());
    			fields.put("CREATED_BY", info.getCreatedBy());
    			fields.put("CREATED_AT", info.getCreatedAt());
    			
    			influxDB.insert(measurement, tags, fields);
    		}
    	}
    	
    	@Test
    	public void testQuery(){//测试数据查询
    		String command = "select * from sys_code";
    		QueryResult results = influxDB.query(command);
    		
    		if(results.getResults() == null){
    			return;
    		}
    		List<CodeInfo> lists = new ArrayList<CodeInfo>();
    		for (Result result : results.getResults()) {
    			
    			List<Series> series= result.getSeries();
    			for (Series serie : series) {
    //				Map<String, String> tags = serie.getTags();
    				List<List<Object>>  values = serie.getValues();
    				List<String> columns = serie.getColumns();
    				
    				lists.addAll(getQueryData(columns, values));
    			}
    		}
    		
    		Assert.assertTrue((!lists.isEmpty()));
    		Assert.assertEquals(2, lists.size());
    	}
    	
    	@Test
    	public void testQueryWhere(){//tag 列名 区分大小写
    		String command = "select * from sys_code where TAG_CODE='ABC'";
    		QueryResult results = influxDB.query(command);
    		
    		if(results.getResults() == null){
    			return;
    		}
    		List<CodeInfo> lists = new ArrayList<CodeInfo>();
    		for (Result result : results.getResults()) {
    			
    			List<Series> series= result.getSeries();
    			for (Series serie : series) {
    				List<List<Object>>  values = serie.getValues();
    				List<String> columns = serie.getColumns();
    				
    				lists.addAll(getQueryData(columns, values));
    			}
    		}
    		
    		Assert.assertTrue((!lists.isEmpty()));
    		Assert.assertEquals(1, lists.size());
    		
    		CodeInfo info = lists.get(0);
    		
    		Assert.assertEquals(info.getCode(), "ABC");
    		
    	}
    	
    	@Test
    	public void deletMeasurementData(){
    		String command = "delete from sys_code where TAG_CODE='ABC'";
    		String err = influxDB.deleteMeasurementData(command);
    		Assert.assertNull(err);
    	}
    	
    	/***整理列名、行数据***/
    	private List<CodeInfo> getQueryData(List<String> columns, List<List<Object>>  values){
    		List<CodeInfo> lists = new ArrayList<CodeInfo>();
    		
    		for (List<Object> list : values) {
    			CodeInfo info = new CodeInfo();
    			BeanWrapperImpl bean = new BeanWrapperImpl(info);
    			for(int i=0; i< list.size(); i++){
    				
    				String propertyName = setColumns(columns.get(i));//字段名
    				Object value = list.get(i);//相应字段值
    				bean.setPropertyValue(propertyName, value);
    			}
    			
    			lists.add(info);
    		}
    		
    		return lists;
    	}
    	
    	/***转义字段***/
    	private String setColumns(String column){
    		String[] cols = column.split("_");
    		StringBuffer sb = new StringBuffer();
    		for(int i=0; i< cols.length; i++){
    			String col = cols[i].toLowerCase();
    			if(i != 0){
    				String start = col.substring(0, 1).toUpperCase();
    				String end = col.substring(1).toLowerCase();
    				col = start + end;
    			}
    			sb.append(col);
    		}
    		return sb.toString();
    	}
    }
    
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值