blob: 8103ce0925b820a4831b69d3405452db0d77e3b7 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
#pragma once
#include <ydb/public/sdk/cpp/client/resources/ydb_resources.h>
#include <ydb/public/api/grpc/ydb_table_v1.grpc.pb.h>
#include <ydb/public/sdk/cpp/client/ydb_proto/accessor.h>
#include <util/random/random.h>
#include "client_session.h"
#include "data_query.h"
#include "request_migrator.h"
namespace NYdb {
namespace NTable {
using namespace NThreading;
class TTablePartIterator::TReaderImpl {
public:
using TSelf = TTablePartIterator::TReaderImpl;
using TResponse = Ydb::Table::ReadTableResponse;
using TStreamProcessorPtr = NGrpc::IStreamRequestReadProcessor<TResponse>::TPtr;
using TReadCallback = NGrpc::IStreamRequestReadProcessor<TResponse>::TReadCallback;
using TGRpcStatus = NGrpc::TGrpcStatus;
using TBatchReadResult = std::pair<TResponse, TGRpcStatus>;
TReaderImpl(TStreamProcessorPtr streamProcessor, const TString& endpoint);
~TReaderImpl();
bool IsFinished();
TAsyncSimpleStreamPart<TResultSet> ReadNext(std::shared_ptr<TSelf> self);
private:
TStreamProcessorPtr StreamProcessor_;
TResponse Response_;
bool Finished_;
TString Endpoint_;
};
class TScanQueryPartIterator::TReaderImpl {
public:
using TSelf = TScanQueryPartIterator::TReaderImpl;
using TResponse = Ydb::Table::ExecuteScanQueryPartialResponse;
using TStreamProcessorPtr = NGrpc::IStreamRequestReadProcessor<TResponse>::TPtr;
using TReadCallback = NGrpc::IStreamRequestReadProcessor<TResponse>::TReadCallback;
using TGRpcStatus = NGrpc::TGrpcStatus;
using TBatchReadResult = std::pair<TResponse, TGRpcStatus>;
TReaderImpl(TStreamProcessorPtr streamProcessor, const TString& endpoint);
~TReaderImpl();
bool IsFinished() const;
TAsyncScanQueryPart ReadNext(std::shared_ptr<TSelf> self);
private:
TStreamProcessorPtr StreamProcessor_;
TResponse Response_;
bool Finished_;
TString Endpoint_;
};
}
}
|
