Files
doris/be/src/util/quantile_state.h
Adonis Ling e412dd12e8 [chore](build) Use include-what-you-use to optimize includes (PART II) (#18761)
Currently, there are some useless includes in the codebase. We can use a tool named include-what-you-use to optimize these includes. By using a strict include-what-you-use policy, we can get lots of benefits from it.
2023-04-19 23:11:48 +08:00

72 lines
2.1 KiB
C++

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the License for the
// specific language governing permissions and limitations
// under the License.
#pragma once
#include <stddef.h>
#include <stdint.h>
#include <algorithm>
#include <memory>
#include <vector>
#include "slice.h"
namespace doris {
class TDigest;
const static int QUANTILE_STATE_EXPLICIT_NUM = 2048;
const static int QUANTILE_STATE_COMPRESSION_MIN = 2048;
const static int QUANTILE_STATE_COMPRESSION_MAX = 10000;
enum QuantileStateType {
EMPTY = 0,
SINGLE = 1, // single element
EXPLICIT = 2, // more than one elements,stored in vector
TDIGEST = 3 // TDIGEST object
};
template <typename T>
class QuantileState {
public:
QuantileState();
explicit QuantileState(float compression);
explicit QuantileState(const Slice& slice);
void set_compression(float compression);
bool deserialize(const Slice& slice);
size_t serialize(uint8_t* dst) const;
void merge(const QuantileState<T>& other);
void add_value(const T& value);
void clear();
bool is_valid(const Slice& slice);
size_t get_serialized_size();
T get_value_by_percentile(float percentile) const;
T get_explicit_value_by_percentile(float percentile) const;
~QuantileState() = default;
private:
QuantileStateType _type = EMPTY;
std::shared_ptr<TDigest> _tdigest_ptr;
T _single_data;
std::vector<T> _explicit_data;
float _compression;
};
using QuantileStateDouble = QuantileState<double>;
} // namespace doris