lbug 0.22.0

An in-process property graph database management system built for query speed and scalability
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
57
58
59
60
61
62
63

#pragma once

#include <memory>
#include <shared_mutex>

#include "binder/ddl/property_definition.h"
#include "common/case_insensitive_map.h"

namespace lbug {
namespace catalog {

class LBUG_API PropertyDefinitionCollection {
public:
    PropertyDefinitionCollection() : nextColumnID{0}, nextPropertyID{0} {}
    explicit PropertyDefinitionCollection(common::column_id_t nextColumnID)
        : nextColumnID{nextColumnID}, nextPropertyID{0} {}
    EXPLICIT_COPY_DEFAULT_MOVE(PropertyDefinitionCollection);

    common::idx_t size() const { return definitions.size(); }

    bool contains(const std::string& name) const { return nameToPropertyIDMap.contains(name); }

    std::vector<binder::PropertyDefinition> getDefinitions() const;
    const binder::PropertyDefinition& getDefinition(const std::string& name) const;
    const binder::PropertyDefinition& getDefinition(common::idx_t idx) const;
    common::column_id_t getMaxColumnID() const;
    common::column_id_t getColumnID(const std::string& name) const;
    common::column_id_t getColumnID(common::property_id_t propertyID) const;
    common::property_id_t getPropertyID(const std::string& name) const;
    // Returns false if the column IDs were already compact and stay unchanged.
    bool vacuumColumnIDs(common::column_id_t nextColumnID);

    void add(const binder::PropertyDefinition& definition);
    void drop(const std::string& name);
    void rename(const std::string& name, const std::string& newName);

    std::string toCypher() const;

    void serialize(common::Serializer& serializer) const;
    static PropertyDefinitionCollection deserialize(common::Deserializer& deserializer);

private:
    PropertyDefinitionCollection(const PropertyDefinitionCollection& other)
        : nextColumnID{other.nextColumnID}, nextPropertyID{other.nextPropertyID},
          definitions{copyMap(other.definitions)}, columnIDs{other.columnIDs},
          nameToPropertyIDMap{other.nameToPropertyIDMap} {}

private:
    // Guards columnIDs (and nextColumnID updates in vacuum/add) against concurrent
    // checkpoint vacuum vs planning reads (see #1160). Vacuum only rewrites columnIDs,
    // so definitions/name map reads remain lock-free. Shared via shared_ptr so the
    // explicit-copy helper keeps working (copies share the lock, which is safe if
    // slightly conservative).
    mutable std::shared_ptr<std::shared_mutex> columnIDsMtx = std::make_shared<std::shared_mutex>();
    common::column_id_t nextColumnID;
    common::property_id_t nextPropertyID;
    std::map<common::property_id_t, binder::PropertyDefinition> definitions;
    std::unordered_map<common::property_id_t, common::column_id_t> columnIDs;
    common::case_insensitive_map_t<common::property_id_t> nameToPropertyIDMap;
};

} // namespace catalog
} // namespace lbug