IoTDB在springboot2中的(二) 查询

上一章我们处理的基本的构建接入，以及插入的处理，那么接下来我们进行查询的操作处理。

我们继续在IoTDBSessionConfig工具类中加入查询的方法处理

/**
     * description: 根据SQL查询最新一条数据
     * author:zgy
     * @param sql sql查询语句，count查询语句，一般为count(id) as ct 必须加as
     * @return
     */
    public Map<String,Object> queryMapData(String sql)  {
        Map<String,Object> rtMap = new HashMap<>();
        try {
            SessionDataSet sessionDataSetWrapper = session.executeQueryStatement(sql,timeoutInMs);
            List<String> columnNames = sessionDataSetWrapper.getColumnNames();
            List<Map<String,Object>> rtList =  packagingData(sessionDataSetWrapper,columnNames);
            if(!rtList.isEmpty()){
                rtMap = rtList.get(0);
            }
        }catch (Exception e) {
            logger.error("查询Map返回数据集数据异常,异常SQL>{};：{}",sql,e.toString());
        }
        return rtMap;
    }

    /**
     * description: 根据SQL查询最新一条数据
     * author:zgy
     * @param sql sql查询语句，查询条件最前面加 last
     * @param clazz 类
     * @return
     */
    public IoTDBRecordable queryLastData(String sql, Class<? extends IoTDBRecordable> clazz)  {
        Map<String,Object> rtMap = new HashMap<>();
        try {
            IoTTableName iotTableName = clazz.getAnnotation(IoTTableName.class);
            String tableName = iotTableName.value();
            SessionDataSet sessionDataSetWrapper = session.executeQueryStatement(sql,timeoutInMs);
            List<String> columnNames = sessionDataSetWrapper.getColumnNames().subList(1,sessionDataSetWrapper.getColumnNames().size());
            List<Map<String,Object>> rtList =  packagingData(sessionDataSetWrapper,columnNames);
            for(Map<String,Object> map : rtList){
                if(map.get(TIMESERIES)!=null){
                    rtMap.put(map.get(TIMESERIES).toString().replace(tableName+".",""),map.get("Value"));
                }
            }
        }catch (Exception e) {
            logger.error(MSG_SQL,clazz,sql,e.toString());
        }
        return JSON.parseObject(JSON.toJSONString(rtMap),clazz);
    }

    /**
     * description: 根据SQL查询最新一条数据
     * author:zgy
     * @param sql sql查询语句，查询条件最前面加 last
     * @param tableName IoTDB表名
     * @return
     */
    public Map<String,Object> queryLastData(String sql,String tableName)  {
        Map<String,Object> rtMap = new HashMap<>();
        try {
            SessionDataSet sessionDataSetWrapper = session.executeQueryStatement(sql,timeoutInMs);
            List<String> columnNames = sessionDataSetWrapper.getColumnNames().subList(1,sessionDataSetWrapper.getColumnNames().size());
            List<Map<String,Object>> rtList =  packagingData(sessionDataSetWrapper,columnNames);
            for(Map<String,Object> map : rtList){
                if(map.get(TIMESERIES)!=null){
                    rtMap.put(map.get(TIMESERIES).toString().replace(tableName+".",""),map.get("Value"));
                }
            }
        }catch (Exception e) {
            logger.error(MSG_SQL,tableName,sql,e.toString());
        }
        return rtMap;
    }

    /**
     * description: 根据SQL查询
     * author: zgy
     * @param sql sql查询语句
     * @param clazz 类.
     * @return
     */
    public List<?> queryListData(String sql, Class<? extends IoTDBRecordable> clazz)  {
        List<IoTDBRecordable> rtList = new ArrayList<>();
        try {
            //获取实体类注释表名
            IoTTableName iotTableName = clazz.getAnnotation(IoTTableName.class);
            String tableName = iotTableName.value();
            //获取查询条件是否是*全查
            String selectParam = sql.substring(6, sql.indexOf("from")).trim();
            SessionDataSet sessionDataSetWrapper = session.executeQueryStatement(sql,timeoutInMs);
            List<String> columnNames = sessionDataSetWrapper.getColumnNames();
            //判断如果第一个列值为Time时间戳去掉
            if(Objects.equals(columnNames.get(0),"Time")){
                columnNames.remove(0);
            }
            List<Map<String,Object>> list =  packagingData(sessionDataSetWrapper,columnNames);
            for(Map<String,Object> map : list){
                //如果是*全查需要进行key值得转换
                if(Objects.equals("*",selectParam)){
                    Map<String,Object> rtMap = new HashMap<>();
                    //循环进行key值得转换
                    for(Map.Entry<String,Object> it : map.entrySet()){
                        rtMap.put(it.getKey().replace(tableName+".",""),it.getValue());
                    }
                    //map转换实体
                    rtList.add(JSON.parseObject(JSON.toJSONString(rtMap),clazz));
                }else{//如果不为*则依照别名来查
                    //map转换实体
                    rtList.add(JSON.parseObject(JSON.toJSONString(map),clazz));
                }
            }
        }catch (Exception e) {
            logger.error("[{}]查询列表数据异常,异常SQL>{};：{}",clazz,sql,e.toString());
        }
        return rtList;
    }

    /**
     * 根据SQL查询 List集合
     * @param sql 查询条件
     * @param tableName 表名
     * @return
     */
    public List<Map<String,Object>> query(String sql,String tableName) {
        List<Map<String,Object>> rtList = new ArrayList<>();
        try {
            SessionDataSet sessionDataSetWrapper = session.executeQueryStatement(sql,timeoutInMs);
            //获取查询条件是否是*全查
            String selectParam = sql.substring(6, sql.indexOf("from")).trim();
            List<String> columnNames = sessionDataSetWrapper.getColumnNames();
            //判断如果第一个列值为Time时间戳去掉
            if(Objects.equals(columnNames.get(0),"Time")){
                columnNames.remove(0);
            }
            List<Map<String,Object>> itsList =  packagingData(sessionDataSetWrapper,columnNames);
            for(Map<String,Object> map : itsList){
                //如果是*全查需要进行key值得转换
                if(Objects.equals("*",selectParam)){
                    Map<String,Object> rtMap = new HashMap<>();
                    //循环进行key值得转换
                    for(Map.Entry<String,Object> it : map.entrySet()){
                        rtMap.put(it.getKey().replace(tableName+".",""),it.getValue());
                    }
                    //map转换实体
                    rtList.add(rtMap);
                }else{//如果不为*则依照别名来查
                    //map转换实体
                    rtList.add(map);
                }
            }
        } catch (Exception e) {
            logger.error(MSG_SQL,tableName,sql,e.toString());
            return rtList;
        }
        return rtList;
    }

    /**
     * 封装处理数据
     * @param sessionDataSet
     * @param titleList
     * @throws StatementExecutionException
     * @throws IoTDBConnectionException
     */
    private List<Map<String,Object>> packagingData(SessionDataSet sessionDataSet, List<String> titleList)
            throws StatementExecutionException, IoTDBConnectionException {
        int fetchSize = sessionDataSet.getFetchSize();
        List<Map<String,Object>> rtList = new ArrayList<>();
        if (fetchSize > 0) {
            while (sessionDataSet.hasNext()) {
                Map<String,Object> rtMap = new HashMap<>();
                RowRecord next = sessionDataSet.next();
                List<Field> fields = next.getFields();
                String timeString = new java.text.SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(next.getTimestamp());
                rtMap.put("time",timeString);
                for (int i = 0; i < fields.size(); i++) {
                    Field field = fields.get(i);
                    // 这里的需要按照类型获取
                    if(field.getDataType()==null||field.getObjectValue(field.getDataType())==null){
                        rtMap.put(titleList.get(i), null);
                    }else{
                        rtMap.put(titleList.get(i), field.getObjectValue(field.getDataType()).toString());
                    }
                }
                rtList.add(rtMap);
            }
        }
        return rtList;
    }

1、queryMapData 方法的话就是查询结果为Map集合的处理，比如聚合函数的查询等等。

2、queryLastData 方法为根据SQL查询表中最新的一条记录数，返回实体类。

3、queryLastData 方法为根据sql和表名查询一个Map集合

4、queryListData 根据sql查询一个list实体集合

5、query 根据sql查询一个List<Map>集合

下面为测试代码

System.out.println("1、queryMapData查询......");
        Map<String,Object> mapOne=  iotDBSessionConfig.queryMapData("select count(id) as ct from root.syslog");
        System.out.println("结果："+ArrayUtils.toString(mapOne));
        System.out.println("2、queryLastData 根据sql和实体类查询......");
        IoTDBSysLog ioTDBSysLog = (IoTDBSysLog) iotDBSessionConfig.queryLastData("select last * from root.syslog",IoTDBSysLog.class);
        System.out.println("结果："+ioTDBSysLog.getLogIp()+"-->"+ioTDBSysLog.getCreateTime());
        System.out.println("3、queryLastData 根据sql和表名查询......");
        Map<String,Object> mapThree =  iotDBSessionConfig.queryLastData("select last * from root.syslog", IoTDBTableParam.SYSLOG_IOT_TABLE);
        System.out.println("结果："+ArrayUtils.toString(mapThree));
        System.out.println("4、queryListData 根据sql查询一个list实体集合......");
        List<IoTDBSysLog> list = (List<IoTDBSysLog>) iotDBSessionConfig.queryListData("select  * from root.syslog",IoTDBSysLog.class);
        System.out.println("结果条数："+list.size());
        for (IoTDBSysLog sysLog : list){
            System.out.println(sysLog.getLogIp()+"-->"+sysLog.getCreateTime());
        }
        System.out.println("5、query 根据sql查询一个List<Map>集合......");
        List<Map<String,Object>> listTwo = iotDBSessionConfig.query("select  * from root.syslog", IoTDBTableParam.SYSLOG_IOT_TABLE);
        System.out.println("结果条数："+list.size());
        for (Map<String,Object> map : listTwo){
            System.out.println(map.get("logIp")+"-->"+map.get("createTime"));
        }

结果如下

 其中3,5两个方法是因为项目中需要进行添加的，是根据动态传递的参数来识别对应的表名，简单进行了包装处理，如果搭建有更好的方法可以留言交流。