lbug 0.16.1

An in-process property graph database management system built for query speed and scalability
Documentation 
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

#pragma once

#include <string>
#include <vector>

#include "common/arrow/arrow.h"
#include "common/arrow/arrow_nullmask_tree.h"

struct ArrowSchema;

namespace lbug {
namespace common {

struct ArrowSchemaHolder {
    std::vector<ArrowSchema> children;
    std::vector<ArrowSchema*> childrenPtrs;
    std::vector<std::vector<ArrowSchema>> nestedChildren;
    std::vector<std::vector<ArrowSchema*>> nestedChildrenPtr;
    std::vector<std::unique_ptr<char[]>> ownedTypeNames;
    std::vector<std::unique_ptr<char[]>> ownedMetadatas;
};

class ArrowConverter {
public:
    static std::unique_ptr<ArrowSchema> toArrowSchema(const std::vector<LogicalType>& dataTypes,
        const std::vector<std::string>& columnNames, bool fallbackExtensionTypes);

    static LogicalType fromArrowSchema(const ArrowSchema* schema);
    static void fromArrowArray(const ArrowSchema* schema, const ArrowArray* array,
        ValueVector& outputVector, ArrowNullMaskTree* mask, uint64_t srcOffset, uint64_t dstOffset,
        uint64_t count);
    static void fromArrowArray(const ArrowSchema* schema, const ArrowArray* array,
        ValueVector& outputVector);

private:
    static void initializeChild(ArrowSchema& child, const std::string& name = "");
    static void setArrowFormatForStruct(ArrowSchemaHolder& rootHolder, ArrowSchema& child,
        const LogicalType& dataType, bool fallbackExtensionTypes);
    static void setArrowFormatForUnion(ArrowSchemaHolder& rootHolder, ArrowSchema& child,
        const LogicalType& dataType, bool fallbackExtensionTypes);
    static void setArrowFormatForInternalID(ArrowSchemaHolder& rootHolder, ArrowSchema& child,
        const LogicalType& dataType, bool fallbackExtensionTypes);
    static void setArrowFormat(ArrowSchemaHolder& rootHolder, ArrowSchema& child,
        const LogicalType& dataType, bool fallbackExtensionTypes);
};

} // namespace common
} // namespace lbug