11// mgconsole - console client for Memgraph database
2- // Copyright (C) 2016-2020 Memgraph Ltd. [https://memgraph.com]
2+ // Copyright (C) 2016-2021 Memgraph Ltd. [https://memgraph.com]
33//
44// This program is free software: you can redistribute it and/or modify
55// it under the terms of the GNU General Public License as published by
3737#include < mgclient.h>
3838#include < replxx.h>
3939
40- #include " version.hpp"
41- #include " utils/utils.hpp"
4240#include " utils/constants.hpp"
41+ #include " utils/utils.hpp"
42+ #include " version.hpp"
4343
4444using namespace std ::string_literals;
4545
@@ -95,10 +95,12 @@ int main(int argc, char **argv) {
9595
9696 gflags::ParseCommandLineFlags (&argc, &argv, true );
9797
98- format::CsvOptions csv_opts{FLAGS_csv_delimiter, FLAGS_csv_escapechar, FLAGS_csv_doublequote};
98+ format::CsvOptions csv_opts{FLAGS_csv_delimiter, FLAGS_csv_escapechar,
99+ FLAGS_csv_doublequote};
99100 format::OutputOptions output_opts{FLAGS_output_format, FLAGS_fit_to_screen};
100101
101- if (output_opts.output_format == constants::kCsvFormat && !csv_opts.ValidateDoubleQuote ()) {
102+ if (output_opts.output_format == constants::kCsvFormat &&
103+ !csv_opts.ValidateDoubleQuote ()) {
102104 console::EchoFailure (
103105 " Unsupported combination of 'csv-doublequote' and 'csv-escapechar'\n "
104106 " flags" ,
@@ -131,16 +133,16 @@ int main(int argc, char **argv) {
131133 password = *password_optional;
132134 } else {
133135 console::EchoFailure (" Password not submitted" ,
134- " Requested password for username " + FLAGS_username);
136+ " Requested password for username " + FLAGS_username);
135137 cleanup_resources ();
136138 return 1 ;
137139 }
138140 console::SetStdinEcho (true );
139141 }
140142
141143 fs::path history_dir = FLAGS_history;
142- if (FLAGS_history ==
143- (constants:: kDefaultHistoryBaseDir + " / " + constants::kDefaultHistoryMemgraphDir )) {
144+ if (FLAGS_history == (constants:: kDefaultHistoryBaseDir + " / " +
145+ constants::kDefaultHistoryMemgraphDir )) {
144146 // Fetch home dir for user.
145147 history_dir =
146148 utils::GetUserHomeDir () / constants::kDefaultHistoryMemgraphDir ;
@@ -230,8 +232,9 @@ int main(int argc, char **argv) {
230232 mg_memory::MgSessionParamsPtr params =
231233 mg_memory::MakeCustomUnique<mg_session_params>(mg_session_params_make ());
232234 if (!params) {
233- console::EchoFailure (" Connection failure" ,
234- " out of memory, failed to allocate `mg_session_params` struct" );
235+ console::EchoFailure (
236+ " Connection failure" ,
237+ " out of memory, failed to allocate `mg_session_params` struct" );
235238 }
236239 mg_session_params_set_host (params.get (), FLAGS_host.c_str ());
237240 mg_session_params_set_port (params.get (), FLAGS_port);
@@ -243,21 +246,23 @@ int main(int argc, char **argv) {
243246 mg_session_params_set_sslmode (
244247 params.get (), FLAGS_use_ssl ? MG_SSLMODE_REQUIRE : MG_SSLMODE_DISABLE);
245248
246- mg_memory::MgSessionPtr session = mg_memory::MakeCustomUnique<mg_session>(nullptr );
249+ mg_memory::MgSessionPtr session =
250+ mg_memory::MakeCustomUnique<mg_session>(nullptr );
247251 {
248252 mg_session *session_tmp;
249253 int status = mg_connect (params.get (), &session_tmp);
250254 session = mg_memory::MakeCustomUnique<mg_session>(session_tmp);
251255 if (status != 0 ) {
252- console::EchoFailure (" Connection failure" , mg_session_error (session.get ()));
256+ console::EchoFailure (" Connection failure" ,
257+ mg_session_error (session.get ()));
253258 cleanup_resources ();
254259 return 1 ;
255260 }
256261 }
257262
258263 console::EchoInfo (" mgconsole " s + gflags::VersionString ());
259264 console::EchoInfo (" Connected to 'memgraph://" + FLAGS_host + " :" +
260- std::to_string (FLAGS_port) + " '" );
265+ std::to_string (FLAGS_port) + " '" );
261266 console::EchoInfo (" Type :help for shell usage" );
262267 console::EchoInfo (" Quit the shell by typing Ctrl-D(eof) or :quit" );
263268 int num_retries = 3 ;
@@ -267,10 +272,12 @@ int main(int argc, char **argv) {
267272 console::EchoInfo (" Bye" );
268273 break ;
269274 }
270- if (query->empty ()) continue ;
275+ if (query->empty ())
276+ continue ;
271277 try {
272278 auto ret = query::ExecuteQuery (session.get (), *query);
273- if (ret.records .size () > 0 ) Output (ret.header , ret.records , output_opts, csv_opts);
279+ if (ret.records .size () > 0 )
280+ Output (ret.header , ret.records , output_opts, csv_opts);
274281 if (console::is_a_tty (STDIN_FILENO)) {
275282 std::string summary;
276283 if (ret.records .size () == 0 ) {
@@ -307,7 +314,8 @@ int main(int argc, char **argv) {
307314 int status = mg_connect (params.get (), &session_tmp);
308315 session = mg_memory::MakeCustomUnique<mg_session>(session_tmp);
309316 if (status != 0 ) {
310- console::EchoFailure (" Connection failure" , mg_session_error (session.get ()));
317+ console::EchoFailure (" Connection failure" ,
318+ mg_session_error (session.get ()));
311319 session.reset (nullptr );
312320 } else {
313321 is_connected = true ;
@@ -318,11 +326,11 @@ int main(int argc, char **argv) {
318326 if (is_connected) {
319327 num_retries = 3 ;
320328 console::EchoInfo (" Connected to 'memgraph://" + FLAGS_host + " :" +
321- std::to_string (FLAGS_port) + " '" );
329+ std::to_string (FLAGS_port) + " '" );
322330 } else {
323- console::EchoFailure (" Couldn't connect to" , " 'memgraph:// " + FLAGS_host + " : " +
324- std::to_string (FLAGS_port) +
325- " '" );
331+ console::EchoFailure (" Couldn't connect to" ,
332+ " 'memgraph:// " + FLAGS_host + " : " +
333+ std::to_string (FLAGS_port) + " '" );
326334 cleanup_resources ();
327335 return 1 ;
328336 }
0 commit comments