dab95af366
Reviewers: teon.banek, mferencevic Reviewed By: teon.banek, mferencevic Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D1546
147 lines
4.9 KiB
Plaintext
147 lines
4.9 KiB
Plaintext
#>cpp
|
|
#pragma once
|
|
|
|
#include <unordered_map>
|
|
|
|
#include "communication/rpc/messages.hpp"
|
|
#include "database/state_delta.hpp"
|
|
#include "distributed/updates_rpc_messages.capnp.h"
|
|
#include "storage/address_types.hpp"
|
|
#include "storage/gid.hpp"
|
|
#include "transactions/type.hpp"
|
|
#include "utils/serialization.hpp"
|
|
cpp<#
|
|
|
|
(lcp:namespace distributed)
|
|
|
|
(lcp:capnp-namespace "distributed")
|
|
|
|
(lcp:capnp-import 'db "/database/state_delta.capnp")
|
|
(lcp:capnp-import 'dis "/distributed/serialization.capnp")
|
|
(lcp:capnp-import 'storage "/storage/serialization.capnp")
|
|
(lcp:capnp-import 'utils "/utils/serialization.capnp")
|
|
|
|
(lcp:capnp-type-conversion "tx::TransactionId" "UInt64")
|
|
(lcp:capnp-type-conversion "gid::Gid" "UInt64")
|
|
(lcp:capnp-type-conversion "storage::Label" "Storage.Common")
|
|
(lcp:capnp-type-conversion "storage::EdgeType" "Storage.Common")
|
|
(lcp:capnp-type-conversion "storage::Property" "Storage.Common")
|
|
(lcp:capnp-type-conversion "storage::EdgeAddress" "Storage.Address")
|
|
(lcp:capnp-type-conversion "storage::VertexAddress" "Storage.Address")
|
|
|
|
(lcp:define-enum update-result
|
|
(done
|
|
serialization-error
|
|
lock-timeout-error
|
|
update-deleted-error
|
|
unable-to-delete-vertex-error)
|
|
(:documentation "The result of sending or applying a deferred update to a worker.")
|
|
(:serialize))
|
|
|
|
(lcp:define-rpc update
|
|
(:request ((member "database::StateDelta" :capnp-type "Db.StateDelta")))
|
|
(:response ((member "UpdateResult"))))
|
|
|
|
(lcp:define-rpc update-apply
|
|
(:request ((member "tx::TransactionId")))
|
|
(:response ((member "UpdateResult"))))
|
|
|
|
(lcp:define-struct create-result ()
|
|
((result "UpdateResult")
|
|
(gid "gid::Gid" :documentation "Only valid if creation was successful."))
|
|
(:serialize :capnp))
|
|
|
|
(lcp:define-struct create-vertex-req-data ()
|
|
((tx-id "tx::TransactionId")
|
|
(labels "std::vector<storage::Label>"
|
|
:capnp-save (lcp:capnp-save-vector "storage::capnp::Common" "storage::Label")
|
|
:capnp-load (lcp:capnp-load-vector "storage::capnp::Common" "storage::Label"))
|
|
(properties "std::unordered_map<storage::Property, query::TypedValue>"
|
|
:capnp-type "Utils.Map(Storage.Common, Dis.TypedValue)"
|
|
:capnp-save
|
|
(lambda (builder member)
|
|
#>cpp
|
|
utils::SaveMap<storage::capnp::Common, capnp::TypedValue>(
|
|
${member}, &${builder},
|
|
[](auto *builder, const auto &entry) {
|
|
auto key_builder = builder->initKey();
|
|
entry.first.Save(&key_builder);
|
|
auto value_builder = builder->initValue();
|
|
distributed::SaveCapnpTypedValue(entry.second, &value_builder);
|
|
});
|
|
cpp<#)
|
|
:capnp-load
|
|
(lambda (reader member)
|
|
#>cpp
|
|
utils::LoadMap<storage::capnp::Common, capnp::TypedValue>(
|
|
&${member}, ${reader},
|
|
[](const auto &reader) {
|
|
storage::Property prop;
|
|
prop.Load(reader.getKey());
|
|
query::TypedValue value;
|
|
distributed::LoadCapnpTypedValue(reader.getValue(), &value);
|
|
return std::make_pair(prop, value);
|
|
});
|
|
cpp<#)))
|
|
(:serialize :capnp))
|
|
|
|
(lcp:define-rpc create-vertex
|
|
(:request ((member "CreateVertexReqData")))
|
|
(:response ((member "CreateResult"))))
|
|
|
|
(lcp:define-struct create-edge-req-data ()
|
|
((from "gid::Gid")
|
|
(to "storage::VertexAddress")
|
|
(edge-type "storage::EdgeType")
|
|
(tx-id "tx::TransactionId"))
|
|
(:serialize :capnp))
|
|
|
|
(lcp:define-rpc create-edge
|
|
(:request ((member "CreateEdgeReqData")))
|
|
(:response ((member "CreateResult"))))
|
|
|
|
(lcp:define-struct add-in-edge-req-data ()
|
|
((from "storage::VertexAddress")
|
|
(edge-address "storage::EdgeAddress")
|
|
(to "gid::Gid")
|
|
(edge-type "storage::EdgeType")
|
|
(tx-id "tx::TransactionId"))
|
|
(:serialize :capnp))
|
|
|
|
(lcp:define-rpc add-in-edge
|
|
(:request ((member "AddInEdgeReqData")))
|
|
(:response ((member "UpdateResult"))))
|
|
|
|
(lcp:define-struct remove-vertex-req-data ()
|
|
((gid "gid::Gid")
|
|
(tx-id "tx::TransactionId")
|
|
(check-empty :bool))
|
|
(:serialize :capnp))
|
|
|
|
(lcp:define-rpc remove-vertex
|
|
(:request ((member "RemoveVertexReqData")))
|
|
(:response ((member "UpdateResult"))))
|
|
|
|
(lcp:define-struct remove-edge-data ()
|
|
((tx-id "tx::TransactionId")
|
|
(edge-id "gid::Gid")
|
|
(vertex-from-id "gid::Gid")
|
|
(vertex-to-address "storage::VertexAddress"))
|
|
(:serialize :capnp))
|
|
|
|
(lcp:define-rpc remove-edge
|
|
(:request ((member "RemoveEdgeData")))
|
|
(:response ((member "UpdateResult"))))
|
|
|
|
(lcp:define-struct remove-in-edge-data ()
|
|
((tx-id "tx::TransactionId")
|
|
(vertex "gid::Gid")
|
|
(edge-address "storage::EdgeAddress"))
|
|
(:serialize :capnp))
|
|
|
|
(lcp:define-rpc remove-in-edge
|
|
(:request ((member "RemoveInEdgeData")))
|
|
(:response ((member "UpdateResult"))))
|
|
|
|
(lcp:pop-namespace) ;; distributed
|