188 lines
7.1 KiB
C++
188 lines
7.1 KiB
C++
#include "parser.h"
|
|
#include "usql.h"
|
|
|
|
#include "linenoise.h"
|
|
|
|
// https://dev.to/joaoh82/what-would-sqlite-look-like-if-written-in-rust-part-1-2np4
|
|
|
|
using namespace std::chrono;
|
|
|
|
const std::vector<std::string> commands {
|
|
"select", "create", "load", "table"
|
|
};
|
|
|
|
std::string get_history_file_dir() {
|
|
std::string file{"/.usql_history.txt"};
|
|
const char *t = std::getenv("HOME");
|
|
|
|
if (t == nullptr) return "/tmp/" + file;
|
|
else return std::string{t} + "/" + file;
|
|
}
|
|
|
|
|
|
size_t last_token_index( std::string str ) {
|
|
// remove trailing white space
|
|
while( !str.empty() && std::isspace( str.back() ) ) str.pop_back() ;
|
|
|
|
// locate the last white space
|
|
return str.find_last_of( "() \t\n" ) ;
|
|
}
|
|
|
|
|
|
void completion(const char *buf, linenoiseCompletions *lc) {
|
|
if (buf != nullptr) {
|
|
std::string str{buf};
|
|
|
|
const auto pos = last_token_index(str);
|
|
if (pos == std::string::npos)
|
|
return; // cannot find what to complete
|
|
|
|
std::string token = str.substr(pos + 1);
|
|
std::string begining = str.substr(0, pos + 1);
|
|
|
|
for (const auto & command : commands) {
|
|
if (command.find(token) == 0) {
|
|
std::string completion_string = begining + command;
|
|
linenoiseAddCompletion(lc, completion_string.c_str());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
char *hints(const char *buf, int *color, int *bold) {
|
|
// if (!strcasecmp(buf,"hello")) {
|
|
// *color = 35;
|
|
// *bold = 0;
|
|
// return " World";
|
|
// }
|
|
return nullptr;
|
|
}
|
|
|
|
|
|
void setup_linenoise() {
|
|
std::string history_file = get_history_file_dir();
|
|
|
|
linenoiseHistorySetMaxLen(500);
|
|
linenoiseSetCompletionCallback(completion);
|
|
linenoiseSetHintsCallback(hints);
|
|
linenoiseSetMultiLine(1);
|
|
linenoiseHistoryLoad(history_file.c_str());
|
|
}
|
|
|
|
void linenoise_line_read(char *line) {
|
|
linenoiseHistoryAdd(line);
|
|
}
|
|
|
|
void close_linenoise() {
|
|
std::string history_file = get_history_file_dir();
|
|
|
|
linenoiseHistorySave(history_file.c_str());
|
|
}
|
|
|
|
|
|
void repl() {
|
|
std::string code;
|
|
std::string input;
|
|
|
|
setup_linenoise();
|
|
|
|
usql::USql uSql{};
|
|
|
|
|
|
while (true) {
|
|
char *line = linenoise(">>> ");
|
|
if (line == nullptr) break;
|
|
|
|
linenoise_line_read(line);
|
|
|
|
input = std::string(line);
|
|
|
|
if (input == "!quit" || input == "!q")
|
|
break;
|
|
else if (input == "!export" || input == "!x") {
|
|
std::cout << "File to export to: ";
|
|
std::getline(std::cin, input);
|
|
|
|
//write_file_contents(input, code);
|
|
} else if (!input.empty()) {
|
|
try {
|
|
time_point<high_resolution_clock> start_time = high_resolution_clock::now();
|
|
auto result = uSql.execute(input);
|
|
time_point<high_resolution_clock> end_time = high_resolution_clock::now();
|
|
|
|
std::cout << input << std::endl;
|
|
std::cout << "run time: " << duration_cast<milliseconds>(end_time - start_time).count() << " ms " << std::endl <<std::endl;
|
|
|
|
result->print();
|
|
|
|
code += input + "\n";
|
|
} catch (std::exception &e) {
|
|
std::cerr << e.what() << std::endl;
|
|
}
|
|
}
|
|
}
|
|
|
|
close_linenoise();
|
|
}
|
|
|
|
void debug() {
|
|
std::vector<std::string> sql_commands {
|
|
// "create table history_earnings_dates (datetime date, symbol varchar(8), time varchar(18), title varchar(256))",
|
|
// "set 'DATE_FORMAT' = '%Y-%m-%d'",
|
|
// "load into history_earnings_dates '/Users/vaclavt/Development/mlisp_fin/data/history_earnings_dates.csv'"
|
|
// "insert into history_earnings_dates (symbol,time,datetime,title) values ('BABA', '07:00:00', '2021-11-04', 'Alibaba Group Holding')",
|
|
// "insert into history_earnings_dates (symbol,time,datetime,title) values ('BABA', '07:00:00', '2021-11-04', 'Alibaba Group Holding')",
|
|
// "delete from history_earnings_dates where symbol='BABA' and datetime=to_date('2021-11-04', '%Y-%m-%d')",
|
|
// "select * from history_earnings_dates"
|
|
|
|
// "create table sf1 (symbol varchar(8), dimension varchar(3), calendar_date date, date_key date, report_period date, last_updated date, accoci float, assets float, assetsavg float, assetsc float, assetsnc float, assetturnover float, bvps float, capex float, cashneq float, cashnequsd float, cor float, consolinc float, currentratio float, de float, debt float, debtc float, debtnc float, debtusd float, deferredrev float, depamor float, deposits float, divyield float, dps float, ebit float, ebitda float, ebitdamargin float, ebitdausd float, ebitusd float, ebt float, eps float, epsdil float, epsusd float, equity float, equityavg float, equityusd float, ev float, evebit float, evebitda float, fcf float, fcfps float, fxusd float, gp float, grossmargin float, intangibles float, intexp float, invcap float, invcapavg float, inventory float, investments float, investmentsc float, investmentsnc float, liabilities float, liabilitiesc float, liabilitiesnc float, marketcap float, ncf float, ncfbus float, ncfcommon float, ncfdebt float, ncfdiv float, ncff float, ncfi float, ncfinv float, ncfo float, ncfx float, netinc float, netinccmn float, netinccmnusd float, netincdis float, netincnci float, netmargin float, opex float, opinc float, payables float, payoutratio float, pb float, pe float, pe1 float, ppnenet float, prefdivis float, price float, ps float, ps1 float, receivables float, retearn float, revenue float, revenueusd float, rnd float, roa float, roe float, roic float, ros float, sbcomp float, sgna float, sharefactor float, sharesbas float, shareswa float, shareswadil float, sps float, tangibles float, taxassets float, taxexp float, taxliabilities float, tbvps float, workingcapital float)",
|
|
// "set 'DATE_FORMAT' = '%Y-%m-%d'",
|
|
// "load into sf1 '/srv/SHARADAR_SF1.csv'",
|
|
// "create index sf1_symbol on sf1(symbol)",
|
|
// "set 'USE_INDEXSCAN' = 'false'",
|
|
// "select dimension, to_string(calendar_date, '%d.%m.%Y'), pp(eps, \"%.2f\"), pp(shareswadil), pp(revenue), pp(netinc), pp(cashneq), pp(assets), pp(debt), pp(ncfdebt), pp(roe*100), pp(intangibles), calendar_date from sf1 where symbol = 'MU' and dimension = 'ARQ' order by dimension, calendar_date desc limit 5",
|
|
// "set 'USE_INDEXSCAN' = 'true'",
|
|
// "select dimension, to_string(calendar_date, '%d.%m.%Y'), pp(eps, \"%.2f\"), pp(shareswadil), pp(revenue), pp(netinc), pp(cashneq), pp(assets), pp(debt), pp(ncfdebt), pp(roe*100), pp(intangibles), calendar_date from sf1 where symbol = 'MU' and dimension = 'ARQ' order by dimension, calendar_date desc limit 5"
|
|
|
|
"create table a (i integer not null, s varchar(64))",
|
|
"insert into a (i, s) values(1, 'one')",
|
|
"insert into a (i, s) values(2, 'two')",
|
|
"insert into a (i, s) values(2, 'second two')",
|
|
"insert into a (i, s) values(3, 'three')",
|
|
"create index a_i on a(i)",
|
|
"create index a_s on a(s)",
|
|
"set 'USE_INDEXSCAN' = 'true'",
|
|
"select * from a where 1 = i",
|
|
"select * from a where i = 2",
|
|
"select max(i) from a where s = 'two'"
|
|
};
|
|
|
|
usql::USql uSql{};
|
|
|
|
for (const auto &command : sql_commands) {
|
|
time_point<high_resolution_clock> start_time = high_resolution_clock::now();
|
|
auto result = uSql.execute(command);
|
|
time_point<high_resolution_clock> end_time = high_resolution_clock::now();
|
|
|
|
std::cout << command << std::endl;
|
|
std::cout << "run time: " << duration_cast<milliseconds>(end_time - start_time).count() << " ms "
|
|
<< std::endl << std::endl;
|
|
|
|
result->print();
|
|
}
|
|
|
|
std::cout << std::endl << std::endl;
|
|
}
|
|
|
|
|
|
int main(int argc, char *argv[]) {
|
|
|
|
//#ifdef NDEBUG
|
|
// repl();
|
|
//#else
|
|
debug();
|
|
//#endif
|
|
|
|
return 0;
|
|
}
|