mage/cpp/csv_utils_module/algorithm/csv_utils.cpp at main · memgraph/mage · GitHub

This repository was archived by the owner on Jan 23, 2026. It is now read-only.

44 lines (34 loc) · 1.32 KB

#include "csv_utils.hpp"

namespace CsvUtils {

void CreateCsvFile(mgp_list *args, mgp_graph *memgraph_graph, mgp_result *result, mgp_memory *memory) {

mgp::MemoryDispatcherGuard guard(memory);

const auto arguments = mgp::List(args);

const auto record_factory = mgp::RecordFactory(result);

try {

const auto filepath = arguments[0].ValueString();

const auto content = arguments[1].ValueString();

const auto is_append = arguments[2].ValueBool();

std::ofstream fout;

fout.open(std::string(filepath), is_append ? std::ofstream::app : std::ofstream::out);

fout << content << std::flush;

fout.close();

auto record = record_factory.NewRecord();

record.Insert(std::string(kArgumentCreateCsvFile1).c_str(), filepath);

} catch (const std::exception &e) {

record_factory.SetErrorMessage(e.what());

}

return;

}

void DeleteCsvFile(mgp_list *args, mgp_graph *memgraph_graph, mgp_result *result, mgp_memory *memory) {

mgp::MemoryDispatcherGuard guard(memory);

const auto arguments = mgp::List(args);

const auto record_factory = mgp::RecordFactory(result);

try {

const std::string_view filepath = arguments[0].ValueString();

std::remove(std::string(filepath).c_str());

} catch (const std::exception &e) {

record_factory.SetErrorMessage(e.what());

}

return;

}

} // namespace CsvUtils

Expand file tree