/root/doris/be/src/util/coding.cpp
Line | Count | Source |
1 | | // Copyright (c) 2011-present, Facebook, Inc. All rights reserved. |
2 | | // This source code is licensed under both the GPLv2 (found in the |
3 | | // COPYING file in the root directory) and Apache 2.0 License |
4 | | // (found in the LICENSE.Apache file in the root directory). |
5 | | // Copyright (c) 2011 The LevelDB Authors. All rights reserved. |
6 | | // Use of this source code is governed by a BSD-style license that can be |
7 | | // found in the LICENSE file. See the AUTHORS file for names of contributors. |
8 | | |
9 | | #include "util/coding.h" |
10 | | |
11 | | namespace doris { |
12 | | |
13 | 72.7M | uint8_t* encode_varint32(uint8_t* dst, uint32_t v) { |
14 | | // Operate on characters as unsigneds |
15 | 72.7M | static const int B = 128; |
16 | 72.7M | if (v < (1 << 7)) { |
17 | 72.1M | *(dst++) = v; |
18 | 72.1M | } else if (v < (1 << 14)) { |
19 | 471k | *(dst++) = v | B; |
20 | 471k | *(dst++) = v >> 7; |
21 | 471k | } else if (v < (1 << 21)) { |
22 | 130k | *(dst++) = v | B; |
23 | 130k | *(dst++) = (v >> 7) | B; |
24 | 130k | *(dst++) = v >> 14; |
25 | 18.4E | } else if (v < (1 << 28)) { |
26 | 8 | *(dst++) = v | B; |
27 | 8 | *(dst++) = (v >> 7) | B; |
28 | 8 | *(dst++) = (v >> 14) | B; |
29 | 8 | *(dst++) = v >> 21; |
30 | 18.4E | } else { |
31 | 18.4E | *(dst++) = v | B; |
32 | 18.4E | *(dst++) = (v >> 7) | B; |
33 | 18.4E | *(dst++) = (v >> 14) | B; |
34 | 18.4E | *(dst++) = (v >> 21) | B; |
35 | 18.4E | *(dst++) = v >> 28; |
36 | 18.4E | } |
37 | 72.7M | return dst; |
38 | 72.7M | } |
39 | | |
40 | | const uint8_t* decode_varint32_ptr_fallback(const uint8_t* p, const uint8_t* limit, |
41 | 378k | uint32_t* value) { |
42 | 378k | uint32_t result = 0; |
43 | 863k | for (uint32_t shift = 0; shift <= 28 && p < limit; shift += 7) { |
44 | 863k | uint32_t byte = *p; |
45 | 863k | p++; |
46 | 863k | if (byte & 128) { |
47 | | // More bytes are present |
48 | 485k | result |= ((byte & 127) << shift); |
49 | 485k | } else { |
50 | 378k | result |= (byte << shift); |
51 | 378k | *value = result; |
52 | 378k | return p; |
53 | 378k | } |
54 | 863k | } |
55 | 18.4E | return nullptr; |
56 | 378k | } |
57 | | |
58 | 242k | const uint8_t* decode_varint64_ptr(const uint8_t* p, const uint8_t* limit, uint64_t* value) { |
59 | 242k | uint64_t result = 0; |
60 | 873k | for (uint32_t shift = 0; shift <= 63 && p < limit; shift += 7) { |
61 | 873k | uint64_t byte = *p; |
62 | 873k | p++; |
63 | 873k | if (byte & 128) { |
64 | | // More bytes are present |
65 | 630k | result |= ((byte & 127) << shift); |
66 | 630k | } else { |
67 | 242k | result |= (byte << shift); |
68 | 242k | *value = result; |
69 | 242k | return p; |
70 | 242k | } |
71 | 873k | } |
72 | 1 | return nullptr; |
73 | 242k | } |
74 | | |
75 | | } // namespace doris |