81 lines
2.8 KiB
C++
81 lines
2.8 KiB
C++
// Copyright 2022 Memgraph Ltd.
|
|
//
|
|
// Use of this software is governed by the Business Source License
|
|
// included in the file licenses/BSL.txt; by using this file, you agree to be bound by the terms of the Business Source
|
|
// License, and you may not use this file except in compliance with the Business Source License.
|
|
//
|
|
// As of the Change Date specified in that file, in accordance with
|
|
// the Business Source License, use of this software will be governed
|
|
// by the Apache License, Version 2.0, included in the file
|
|
// licenses/APL.txt.
|
|
|
|
#include "mg_procedure.h"
|
|
|
|
static void TryToWrite(struct mgp_list *args, mgp_func_context *ctx, mgp_func_result *result,
|
|
struct mgp_memory *memory) {
|
|
mgp_value *value{nullptr};
|
|
mgp_vertex *vertex{nullptr};
|
|
mgp_list_at(args, 0, &value);
|
|
mgp_value_get_vertex(value, &vertex);
|
|
|
|
const char *name;
|
|
mgp_list_at(args, 1, &value);
|
|
mgp_value_get_string(value, &name);
|
|
|
|
mgp_list_at(args, 2, &value);
|
|
|
|
// Setting a property should set an error
|
|
auto err_code = mgp_vertex_set_property(vertex, name, value);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
mgp_func_result_set_error_msg(result, "Cannot set property in the function!", memory);
|
|
return;
|
|
}
|
|
|
|
err_code = mgp_func_result_set_value(result, value, memory);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
mgp_func_result_set_error_msg(result, "Failed to construct return value!", memory);
|
|
return;
|
|
}
|
|
}
|
|
|
|
// Each module needs to define mgp_init_module function.
|
|
// Here you can register multiple functions/procedures your module supports.
|
|
extern "C" int mgp_init_module(struct mgp_module *module, struct mgp_memory *memory) {
|
|
{
|
|
mgp_func *func{nullptr};
|
|
auto err_code = mgp_module_add_function(module, "try_to_write", TryToWrite, &func);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
return 1;
|
|
}
|
|
|
|
mgp_type *type_vertex{nullptr};
|
|
mgp_type_node(&type_vertex);
|
|
err_code = mgp_func_add_arg(func, "argument", type_vertex);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
return 1;
|
|
}
|
|
|
|
mgp_type *type_string{nullptr};
|
|
mgp_type_string(&type_string);
|
|
err_code = mgp_func_add_arg(func, "name", type_string);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
return 1;
|
|
}
|
|
|
|
mgp_type *any_type{nullptr};
|
|
mgp_type_any(&any_type);
|
|
mgp_type *nullable_type{nullptr};
|
|
mgp_type_nullable(any_type, &nullable_type);
|
|
err_code = mgp_func_add_arg(func, "value", nullable_type);
|
|
if (err_code != mgp_error::MGP_ERROR_NO_ERROR) {
|
|
return 1;
|
|
}
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
// This is an optional function if you need to release any resources before the
|
|
// module is unloaded. You will probably need this if you acquired some
|
|
// resources in mgp_init_module.
|
|
extern "C" int mgp_shutdown_module() { return 0; }
|