mariadb-columnstore-engine/tools/qfe/cli.cpp

#ifdef _MSC_VER
#define WIN32_LEAN_AND_MEAN
#define NOMINMAX
#include <windows.h>
#include <winsock2.h>
#include <ws2tcpip.h>
#include <iphlpapi.h>
#include <icmpapi.h>
#include <stdio.h>
#else
#include <poll.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#endif
#include <iostream>
#include <string>
#include <sstream>
#include <iomanip>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <cerrno>
#include <cstring>
#include <cstdlib>
using namespace std;

#include "socktype.h"
#include "socketio.h"
using namespace qfe;

namespace
{
void usage()
{
	cout << "usage: cli [-h] [-s serverip] [-p port] [-c schema] <query>" << endl;
	cout << '\t' << "-s serverip connect to serverip instead of 127.0.0.1" << endl;
	cout << '\t' << "-p port     connect on port instead of 9198" << endl;
	cout << '\t' << "-c schema   use schema as default instead of tpch1" << endl;
	cout << '\t' << "-h          display this help" << endl;
}
}

int main(int argc, char** argv)
{
	opterr = 0;
	int c;
	string serverip("127.0.0.1");
	string schema("tpch1");
	short PortNo = 9198;
	while ((c = getopt(argc, argv, "s:c:p:h")) != -1)
		switch (c)
		{
		case 's':
			serverip = optarg;
			break;
		case 'c':
			schema = optarg;
			break;
		case 'p':
			PortNo = atoi(optarg);
			if (PortNo == 0)
			{
				usage();
				return 1;
			}
			break;
		case 'h':
		case '?':
		default:
			usage();
			return 0;
			break;
		}

	if (argc - optind < 1)
	{
		usage();
		return 1;
	}

	string query(argv[optind+0]);

	SockType fd = -1;

#ifdef _MSC_VER
	WSAData wsadata;
	const WORD minVersion = MAKEWORD(2, 2);
	if (WSAStartup(minVersion, &wsadata) != 0)
		cerr << "networking startup error: " << IDBSysErrorStr(WSAGetLastError()) << endl;
#endif
	fd = ::socket(PF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (fd < 0)
	{
#ifdef _MSC_VER
		cerr << "socket create error: " << IDBSysErrorStr(WSAGetLastError()) << endl;
#else
		cerr << "socket create error: " << strerror(errno) << endl;
#endif
		return 1;
	}

	int rc = 0;
	struct sockaddr_in serv_addr;
	struct addrinfo hints;
	struct addrinfo* res=0;
	memset(&hints, 0, sizeof(struct addrinfo));
	hints.ai_family = PF_INET;
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_protocol = IPPROTO_TCP;
	rc = getaddrinfo(serverip.c_str(), 0, &hints, &res);
	if (rc != 0)
	{
		cerr << "Error resolving '" << serverip << "': " << gai_strerror(rc) << endl;
		return 1;
	}
	sockaddr_in* sain=0;
	sain = reinterpret_cast<sockaddr_in*>(res->ai_addr);
	memset(&serv_addr, 0, sizeof(serv_addr));
	serv_addr.sin_family = PF_INET;
	serv_addr.sin_addr = sain->sin_addr;
	serv_addr.sin_port = htons(PortNo);
	freeaddrinfo(res);

	rc = ::connect(fd, (sockaddr*)&serv_addr, sizeof(serv_addr));
	if (rc < 0)
	{
#ifdef _MSC_VER
		cerr << "socket connect error: " << IDBSysErrorStr(WSAGetLastError()) << endl;
		WSACleanup();
#else
		cerr << "socket connect error: " << strerror(errno) << endl;
#endif
		return 1;
	}

	socketio::writeString(fd, schema);
	socketio::writeString(fd, query);

	uint32_t flag=0;
	string row;
	row = socketio::readString(fd);
	if (row != "OK")
	{
		cerr << "query failed: " << row << endl;
		goto bailout;
	}

	row = socketio::readString(fd);
	while (!row.empty())
	{
		cout << row << endl;
		row = socketio::readString(fd);
	}

	flag=0;
	SockWriteFcn(fd, &flag, 4);
bailout:
	::shutdown(fd, SHUT_RDWR);
#ifdef _MSC_VER
	::closesocket(fd);
	WSACleanup();
#else
	::close(fd);
#endif
	fd = -1;

	return 0;
}