You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-08-05 16:15:50 +03:00
clang format apply
This commit is contained in:
@@ -30,158 +30,156 @@
|
||||
using namespace std;
|
||||
using namespace storagemanager;
|
||||
|
||||
void usage(const char *progname)
|
||||
void usage(const char* progname)
|
||||
{
|
||||
cerr << progname << " reads from stdin and puts it in a file managed by StorageManager" << endl;
|
||||
cerr << "Usage: " << progname << " output_file" << endl;
|
||||
cerr << progname << " reads from stdin and puts it in a file managed by StorageManager" << endl;
|
||||
cerr << "Usage: " << progname << " output_file" << endl;
|
||||
}
|
||||
|
||||
bool SMOnline()
|
||||
{
|
||||
struct sockaddr_un addr;
|
||||
memset(&addr, 0, sizeof(addr));
|
||||
addr.sun_family = AF_UNIX;
|
||||
strcpy(&addr.sun_path[1], &socket_name[1]); // first char is null...
|
||||
int clientSocket = ::socket(AF_UNIX, SOCK_STREAM, 0);
|
||||
int err = ::connect(clientSocket, (const struct sockaddr *) &addr, sizeof(addr));
|
||||
if (err >= 0)
|
||||
{
|
||||
::close(clientSocket);
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
struct sockaddr_un addr;
|
||||
memset(&addr, 0, sizeof(addr));
|
||||
addr.sun_family = AF_UNIX;
|
||||
strcpy(&addr.sun_path[1], &socket_name[1]); // first char is null...
|
||||
int clientSocket = ::socket(AF_UNIX, SOCK_STREAM, 0);
|
||||
int err = ::connect(clientSocket, (const struct sockaddr*)&addr, sizeof(addr));
|
||||
if (err >= 0)
|
||||
{
|
||||
::close(clientSocket);
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
void putOffline(const char *fname, int prefixlen)
|
||||
void putOffline(const char* fname, int prefixlen)
|
||||
{
|
||||
uint8_t data[8192];
|
||||
int read_err, write_err;
|
||||
ssize_t count, offset = 0;
|
||||
try
|
||||
uint8_t data[8192];
|
||||
int read_err, write_err;
|
||||
ssize_t count, offset = 0;
|
||||
try
|
||||
{
|
||||
boost::scoped_ptr<IOCoordinator> ioc(IOCoordinator::get());
|
||||
struct stat _stat;
|
||||
read_err = ioc->open(fname, O_CREAT | O_TRUNC | O_WRONLY, &_stat);
|
||||
if (read_err < 0)
|
||||
{
|
||||
boost::scoped_ptr<IOCoordinator> ioc(IOCoordinator::get());
|
||||
struct stat _stat;
|
||||
read_err = ioc->open(fname, O_CREAT | O_TRUNC | O_WRONLY, &_stat);
|
||||
if (read_err < 0)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Failed to open/create " << &fname[prefixlen] << ": " <<
|
||||
strerror_r(l_errno, (char *) data, 8192) << endl;
|
||||
exit(1);
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
read_err = ::read(STDIN_FILENO, data, 8192);
|
||||
if (read_err < 0)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Error reading stdin: " << strerror_r(l_errno, (char *) data, 8192) << endl;
|
||||
exit(1);
|
||||
}
|
||||
count = 0;
|
||||
while (count < read_err)
|
||||
{
|
||||
write_err = ioc->write(fname, &data[count], offset + count, read_err - count);
|
||||
if (write_err < 0)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Error writing to " << &fname[prefixlen] << ": " <<
|
||||
strerror_r(l_errno, (char *) data, 8192) << endl;
|
||||
exit(1);
|
||||
}
|
||||
count += write_err;
|
||||
}
|
||||
offset += read_err;
|
||||
} while (read_err > 0);
|
||||
int l_errno = errno;
|
||||
cerr << "Failed to open/create " << &fname[prefixlen] << ": " << strerror_r(l_errno, (char*)data, 8192)
|
||||
<< endl;
|
||||
exit(1);
|
||||
}
|
||||
catch (exception &e)
|
||||
{
|
||||
cerr << "smput putOffline FAIL: " << e.what() << endl;
|
||||
}
|
||||
}
|
||||
|
||||
void putOnline(const char *fname, int prefixlen)
|
||||
{
|
||||
uint8_t data[8192];
|
||||
int read_err, write_err;
|
||||
ssize_t count;
|
||||
|
||||
idbdatafile::SMFileFactory ffactory;
|
||||
boost::scoped_ptr<idbdatafile::SMDataFile> df(
|
||||
dynamic_cast<idbdatafile::SMDataFile *>(ffactory.open(fname, "w", 0, 0)));
|
||||
|
||||
if (!df)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Failed to open/create " << &fname[prefixlen] << ": " <<
|
||||
strerror_r(l_errno, (char *) data, 8192) << endl;
|
||||
exit(1);
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
read_err = ::read(STDIN_FILENO, data, 8192);
|
||||
if (read_err < 0)
|
||||
read_err = ::read(STDIN_FILENO, data, 8192);
|
||||
if (read_err < 0)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Error reading stdin: " << strerror_r(l_errno, (char*)data, 8192) << endl;
|
||||
exit(1);
|
||||
}
|
||||
count = 0;
|
||||
while (count < read_err)
|
||||
{
|
||||
write_err = ioc->write(fname, &data[count], offset + count, read_err - count);
|
||||
if (write_err < 0)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Error reading stdin: " << strerror_r(l_errno, (char *) data, 8192) << endl;
|
||||
exit(1);
|
||||
}
|
||||
count = 0;
|
||||
while (count < read_err)
|
||||
{
|
||||
write_err = df->write(&data[count], read_err - count);
|
||||
if (write_err < 0)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Error writing to " << &fname[prefixlen] << ": " <<
|
||||
strerror_r(l_errno, (char *) data, 8192) << endl;
|
||||
exit(1);
|
||||
}
|
||||
count += write_err;
|
||||
int l_errno = errno;
|
||||
cerr << "Error writing to " << &fname[prefixlen] << ": " << strerror_r(l_errno, (char*)data, 8192)
|
||||
<< endl;
|
||||
exit(1);
|
||||
}
|
||||
count += write_err;
|
||||
}
|
||||
offset += read_err;
|
||||
} while (read_err > 0);
|
||||
}
|
||||
catch (exception& e)
|
||||
{
|
||||
cerr << "smput putOffline FAIL: " << e.what() << endl;
|
||||
}
|
||||
}
|
||||
|
||||
int makePathPrefix(char *target, int targetlen)
|
||||
void putOnline(const char* fname, int prefixlen)
|
||||
{
|
||||
// MCOL-3438 -> add bogus directories to the front of each param
|
||||
Config *config = Config::get();
|
||||
int prefixDepth = stoi(config->getValue("ObjectStorage", "common_prefix_depth"));
|
||||
target[0] = '/';
|
||||
target[1] = 0;
|
||||
int bufpos = 1;
|
||||
|
||||
for (int i = 0; i < prefixDepth; i++)
|
||||
uint8_t data[8192];
|
||||
int read_err, write_err;
|
||||
ssize_t count;
|
||||
|
||||
idbdatafile::SMFileFactory ffactory;
|
||||
boost::scoped_ptr<idbdatafile::SMDataFile> df(
|
||||
dynamic_cast<idbdatafile::SMDataFile*>(ffactory.open(fname, "w", 0, 0)));
|
||||
|
||||
if (!df)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Failed to open/create " << &fname[prefixlen] << ": " << strerror_r(l_errno, (char*)data, 8192)
|
||||
<< endl;
|
||||
exit(1);
|
||||
}
|
||||
|
||||
do
|
||||
{
|
||||
read_err = ::read(STDIN_FILENO, data, 8192);
|
||||
if (read_err < 0)
|
||||
{
|
||||
if (bufpos + 3 >= targetlen)
|
||||
{
|
||||
cerr << "invalid prefix depth in ObjectStorage/common_prefix_depth";
|
||||
exit(1);
|
||||
}
|
||||
memcpy(&target[bufpos], "x/\0", 3);
|
||||
bufpos += 2;
|
||||
int l_errno = errno;
|
||||
cerr << "Error reading stdin: " << strerror_r(l_errno, (char*)data, 8192) << endl;
|
||||
exit(1);
|
||||
}
|
||||
return bufpos;
|
||||
count = 0;
|
||||
while (count < read_err)
|
||||
{
|
||||
write_err = df->write(&data[count], read_err - count);
|
||||
if (write_err < 0)
|
||||
{
|
||||
int l_errno = errno;
|
||||
cerr << "Error writing to " << &fname[prefixlen] << ": " << strerror_r(l_errno, (char*)data, 8192)
|
||||
<< endl;
|
||||
exit(1);
|
||||
}
|
||||
count += write_err;
|
||||
}
|
||||
} while (read_err > 0);
|
||||
}
|
||||
|
||||
int main(int argc, char **argv)
|
||||
int makePathPrefix(char* target, int targetlen)
|
||||
{
|
||||
if (argc != 2)
|
||||
// MCOL-3438 -> add bogus directories to the front of each param
|
||||
Config* config = Config::get();
|
||||
int prefixDepth = stoi(config->getValue("ObjectStorage", "common_prefix_depth"));
|
||||
target[0] = '/';
|
||||
target[1] = 0;
|
||||
int bufpos = 1;
|
||||
|
||||
for (int i = 0; i < prefixDepth; i++)
|
||||
{
|
||||
if (bufpos + 3 >= targetlen)
|
||||
{
|
||||
usage(argv[0]);
|
||||
return 1;
|
||||
cerr << "invalid prefix depth in ObjectStorage/common_prefix_depth";
|
||||
exit(1);
|
||||
}
|
||||
|
||||
char prefix[8192];
|
||||
int prefixlen = makePathPrefix(prefix, 8192);
|
||||
|
||||
if (SMOnline())
|
||||
putOnline(strncat(prefix, argv[1], 8192 - prefixlen), prefixlen);
|
||||
else
|
||||
putOffline(strncat(prefix, argv[1], 8192 - prefixlen), prefixlen);
|
||||
return 0;
|
||||
memcpy(&target[bufpos], "x/\0", 3);
|
||||
bufpos += 2;
|
||||
}
|
||||
return bufpos;
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
if (argc != 2)
|
||||
{
|
||||
usage(argv[0]);
|
||||
return 1;
|
||||
}
|
||||
|
||||
char prefix[8192];
|
||||
int prefixlen = makePathPrefix(prefix, 8192);
|
||||
|
||||
if (SMOnline())
|
||||
putOnline(strncat(prefix, argv[1], 8192 - prefixlen), prefixlen);
|
||||
else
|
||||
putOffline(strncat(prefix, argv[1], 8192 - prefixlen), prefixlen);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user