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

// Copyright 2023 TiKV Project Authors. Licensed under Apache-2.0.

use crate::proto::kvrpcpb;
use crate::request::KvRequest;

pub trait Codec: Clone + Sync + Send + 'static {
    fn encode_request<R: KvRequest>(&self, _req: &mut R) {}
    // TODO: fn decode_response()
}

#[derive(Clone, Default)]
pub struct ApiV1TxnCodec {}

impl Codec for ApiV1TxnCodec {}

#[derive(Clone, Default)]
pub struct ApiV1RawCodec {}

impl Codec for ApiV1RawCodec {}

#[derive(Clone)]
pub struct ApiV2TxnCodec {
    _keyspace_id: u32,
}

impl ApiV2TxnCodec {
    pub fn new(keyspace_id: u32) -> Self {
        Self {
            _keyspace_id: keyspace_id,
        }
    }
}

impl Codec for ApiV2TxnCodec {
    fn encode_request<R: KvRequest>(&self, req: &mut R) {
        req.set_api_version(kvrpcpb::ApiVersion::V2);
        // TODO: req.encode_request(self);
    }
}

// TODO: pub struct ApiV2RawCodec

// EncodeRequest is just a type wrapper to avoid passing not encoded request to `PlanBuilder` by mistake.
#[derive(Clone)]
pub struct EncodedRequest<Req: KvRequest> {
    pub inner: Req,
}

impl<Req: KvRequest> EncodedRequest<Req> {
    pub fn new<C: Codec>(mut req: Req, codec: &C) -> Self {
        codec.encode_request(&mut req);
        Self { inner: req }
    }
}