Apache IoTDB源码解析（0.11.2版本）：Session执行executeQueryStatement的源码解析

最新推荐文章于 2023-11-16 20:54:45 发布

编程写手

最新推荐文章于 2023-11-16 20:54:45 发布

阅读量470

点赞数

分类专栏：时序数据库文章标签： java 服务器

本文链接：https://blog.csdn.net/weixin_45492007/article/details/125235818

版权

时序数据库专栏收录该内容

12 篇文章 1 订阅

订阅专栏

当前版本：IoTDB 0.11.2

1. 声明

当前内容主要为记录下Session执行一次sql查询的过程的源码解析。
前面内容中有发现IoTDB本身就是使用Thrift来实现客户端和服务器，并解析得到Session就是Client的包装产物，所以这里的交互一般都是直接调用服务器的接口完成的！

2. 分析查询的Demo

下面给出一个简单的查询Demo，然后对其进行分析操作

		Session session = new Session("localhost",6667,"root","root");
		try {
			session.open();
			SessionDataSet sessionDataSet = session.executeQueryStatement("select value1 from root.test");
			sessionDataSet.setFetchSize(1); // 只一个一个的获取
			int fetchSize = sessionDataSet.getFetchSize();
			System.out.println(fetchSize);
			int nextNum = 0;
			while(sessionDataSet.hasNext()) {
				RowRecord next = sessionDataSet.next(); // 如果获取的数据量
				System.out.println(next.getTimestamp());
				nextNum++;
			}
			System.out.println("nextNum="+nextNum);
			sessionDataSet.closeOperationHandle(); 
		} catch (IoTDBConnectionException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}finally {
			try {
				session.close();
			} catch (IoTDBConnectionException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		}
		
	}

这里采用了一个非常简单的查询执行

打开连接：open
执行sql ：executeQueryStatement
得到结果：hasNext和next
关闭结果集：closeOperationHandle();
关闭连接：close

所以对于session的操作可以直接映射到服务端，服务端的具体实现是：TSServiceImpl这个实现类,实现了对Thrift的Server端的处理

1. 找到openSession方法(对应session.open)
在这里插入图片描述
该方法中使用了认证管理器对，给定的用户名和密码进行校验，对于登录成功的给与一个sessionId标记(默认从1开始)并缓存到服务器中，所以就会出现下面

也就是说，对于登录的用户给与一个sessionId，后面通过这个sessionId来判断该Client是否登录成功，所以open方式实际上就是登录并得到sessionId标记的方法

2. 找到executeQueryStatement方法(对应session.executeQueryStatement)

在这里插入图片描述
通过上面验证了，后续的发送请求中必须携带sessionId(并且还是登录的sessionId)才能发送请求到IoTDB的server端，发现这个其实就是对传递的sql语句进行解析为PhysicalPlan，后面通过执行器进行执行查询的操作(这里会返回查询结果)

并且从createQueryDataSet方法中可以知道会缓存QueryDataSet

3. sessionDataSet.hasNext()的源码分析

 public boolean hasNext() throws StatementExecutionException, IoTDBConnectionException {
   return ioTDBRpcDataSet.next();
 }
 // ioTDBRpcDataSet.next()对应的方法
 public boolean next() throws StatementExecutionException, IoTDBConnectionException {
    if (hasCachedResults()) {
      constructOneRow();
      return true;
    }
    if (emptyResultSet) {
      return false;
    }
    if (fetchResults()) { // 这个地方会请求server端的fetchResults方法
      constructOneRow();
      return true;
    }
    return false;
  }

所以hasNext方法就是为了确认数据是否存在，如果本地的数据迭代完毕后会尝试请求服务器判断是否还有数据！

4. 查看sessionDataSet.next();方法源码

public RowRecord next() throws StatementExecutionException, IoTDBConnectionException {
    if (!ioTDBRpcDataSet.hasCachedRecord && !hasNext()) {
      return null;
    }
    ioTDBRpcDataSet.hasCachedRecord = false;

    return constructRowRecordFromValueArray();
  }

其实这个地方还是会请求hasNext方法，并完成数据的构造为RowRecord实例

5. sessionDataSet.closeOperationHandle(); 方法的源码解析

public void closeOperationHandle() throws StatementExecutionException, IoTDBConnectionException {
    try {
      ioTDBRpcDataSet.close(); // 发现会调用close方法，但是实际上映射为TSCloseOperationReq closeReq = new TSCloseOperationReq(sessionId);
    } catch (TException e) {
      throw new IoTDBConnectionException(e.getMessage());
    }
  }