summaryrefslogtreecommitdiffstats
path: root/yql/essentials/parser/pg_wrapper/pg_ops.h
blob: 3425080684da2e59dce34b0b49fc8952fa8613e4 (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
#pragma once

#include "comp_factory_utils.h"

#include <memory>

namespace NYql {

class TPgOpBase {
public:
    TPgOpBase() = default;
    ~TPgOpBase() = default;

private:
    const NKikimr::NMiniKQL::TThrowingBindTerminator BindTerminator;
};

// Base class for sign computation
class TPgSign: public TPgOpBase {
public:
    class TCallState {
    public:
        virtual TMaybe<i32> GetSign(const NUdf::TUnboxedValue& value) = 0;
        virtual ~TCallState() = default;
    };

    virtual ~TPgSign() = default;

    virtual std::unique_ptr<TCallState> MakeCallState() const = 0;

    static std::unique_ptr<TPgSign> Create(ui32 inputTypeOid);
};

// Concrete comparison operator: operName is "<" or ">"
class TPgCompareOp: public TPgOpBase {
public:
    TPgCompareOp(ui32 lhsTypeOid, ui32 rhsTypeOid, std::string_view operName);

    class TCallState {
    public:
        explicit TCallState(const TPgResolvedCallWithCast& call);

        TMaybe<bool> Compare(const NUdf::TUnboxedValue& lhs, const NUdf::TUnboxedValue& rhs);

    private:
        TPgResolvedCallWithCastState State;
        const TPgResolvedCallWithCast& Call;
    };

    TCallState MakeCallState() const;

private:
    TPgResolvedCallWithCast Call;
};

} // namespace NYql