String FieldVisitorDump::operator() (const String & x) const
{
String res;
WriteBufferFromString wb(res);
writeQuoted(x, wb);
return res;
}
Foam::Ostream& Foam::OSstream::write(const token& t)
{
if (t.type() == token::VERBATIMSTRING)
{
write(char(token::HASH));
write(char(token::BEGIN_BLOCK));
writeQuoted(t.stringToken(), false);
write(char(token::HASH));
write(char(token::END_BLOCK));
}
else if (t.type() == token::VARIABLE)
{
writeQuoted( t.stringToken(), false);
}
return *this;
}
static inline String formatQuoted(T x)
{
String res;
WriteBufferFromString wb(res);
writeQuoted(x, wb);
return res;
}
static inline String formatQuotedWithPrefix(T x, const char * prefix)
{
String res;
WriteBufferFromString wb(res);
wb.write(prefix, strlen(prefix));
writeQuoted(x, wb);
return res;
}
std::string MySQLDictionarySource::quoteForLike(const std::string s)
{
std::string tmp;
tmp.reserve(s.size());
for (auto c : s)
{
if (c == '%' || c == '_' || c == '\\')
tmp.push_back('\\');
tmp.push_back(c);
}
std::string res;
{
WriteBufferFromString out(res);
writeQuoted(tmp, out);
}
return res;
}
std::string ExternalQueryBuilder::composeLoadAllQuery() const
{
WriteBufferFromOwnString out;
writeString("SELECT ", out);
if (dict_struct.id)
{
if (!dict_struct.id->expression.empty())
{
writeParenthesisedString(dict_struct.id->expression, out);
writeString(" AS ", out);
}
writeQuoted(dict_struct.id->name, out);
if (dict_struct.range_min && dict_struct.range_max)
{
writeString(", ", out);
if (!dict_struct.range_min->expression.empty())
{
writeParenthesisedString(dict_struct.range_min->expression, out);
writeString(" AS ", out);
}
writeQuoted(dict_struct.range_min->name, out);
writeString(", ", out);
if (!dict_struct.range_max->expression.empty())
{
writeParenthesisedString(dict_struct.range_max->expression, out);
writeString(" AS ", out);
}
writeQuoted(dict_struct.range_max->name, out);
}
}
else if (dict_struct.key)
{
auto first = true;
for (const auto & key : *dict_struct.key)
{
if (!first)
writeString(", ", out);
first = false;
if (!key.expression.empty())
{
writeParenthesisedString(key.expression, out);
writeString(" AS ", out);
}
writeQuoted(key.name, out);
}
}
for (const auto & attr : dict_struct.attributes)
{
writeString(", ", out);
if (!attr.expression.empty())
{
writeParenthesisedString(attr.expression, out);
writeString(" AS ", out);
}
writeQuoted(attr.name, out);
}
writeString(" FROM ", out);
if (!db.empty())
{
writeQuoted(db, out);
writeChar('.', out);
}
if (!schema.empty())
{
writeQuoted(schema, out);
writeChar('.', out);
}
writeQuoted(table, out);
if (!where.empty())
{
writeString(" WHERE ", out);
writeString(where, out);
}
writeChar(';', out);
return out.str();
}
std::string
ExternalQueryBuilder::composeLoadKeysQuery(const Columns & key_columns, const std::vector<size_t> & requested_rows, LoadKeysMethod method)
{
if (!dict_struct.key)
throw Exception{"Composite key required for method", ErrorCodes::UNSUPPORTED_METHOD};
WriteBufferFromOwnString out;
writeString("SELECT ", out);
auto first = true;
for (const auto & key_or_attribute : boost::join(*dict_struct.key, dict_struct.attributes))
{
if (!first)
writeString(", ", out);
first = false;
if (!key_or_attribute.expression.empty())
{
writeParenthesisedString(key_or_attribute.expression, out);
writeString(" AS ", out);
}
writeQuoted(key_or_attribute.name, out);
}
writeString(" FROM ", out);
if (!db.empty())
{
writeQuoted(db, out);
writeChar('.', out);
}
if (!schema.empty())
{
writeQuoted(schema, out);
writeChar('.', out);
}
writeQuoted(table, out);
writeString(" WHERE ", out);
if (!where.empty())
{
writeString("(", out);
writeString(where, out);
writeString(") AND (", out);
}
if (method == AND_OR_CHAIN)
{
first = true;
for (const auto row : requested_rows)
{
if (!first)
writeString(" OR ", out);
first = false;
composeKeyCondition(key_columns, row, out);
}
}
else /* if (method == IN_WITH_TUPLES) */
{
writeString(composeKeyTupleDefinition(), out);
writeString(" IN (", out);
first = true;
for (const auto row : requested_rows)
{
if (!first)
writeString(", ", out);
first = false;
composeKeyTuple(key_columns, row, out);
}
writeString(")", out);
}
if (!where.empty())
{
writeString(")", out);
}
writeString(";", out);
return out.str();
}
std::string ExternalQueryBuilder::composeLoadIdsQuery(const std::vector<UInt64> & ids)
{
if (!dict_struct.id)
throw Exception{"Simple key required for method", ErrorCodes::UNSUPPORTED_METHOD};
WriteBufferFromOwnString out;
writeString("SELECT ", out);
if (!dict_struct.id->expression.empty())
{
writeParenthesisedString(dict_struct.id->expression, out);
writeString(" AS ", out);
}
writeQuoted(dict_struct.id->name, out);
for (const auto & attr : dict_struct.attributes)
{
writeString(", ", out);
if (!attr.expression.empty())
{
writeParenthesisedString(attr.expression, out);
writeString(" AS ", out);
}
writeQuoted(attr.name, out);
}
writeString(" FROM ", out);
if (!db.empty())
{
writeQuoted(db, out);
writeChar('.', out);
}
if (!schema.empty())
{
writeQuoted(schema, out);
writeChar('.', out);
}
writeQuoted(table, out);
writeString(" WHERE ", out);
if (!where.empty())
{
writeString(where, out);
writeString(" AND ", out);
}
writeQuoted(dict_struct.id->name, out);
writeString(" IN (", out);
auto first = true;
for (const auto id : ids)
{
if (!first)
writeString(", ", out);
first = false;
writeString(DB::toString(id), out);
}
writeString(");", out);
return out.str();
}
/** The actual implemenation code for @a createServiceFile. */
bool createServiceFileCore(char **ppachTemplate,
struct SERVICEPARAMETERS *pParameters)
{
/* The size of the template data we have read. */
size_t cchTemplate = 0;
/* The size of the buffer we have allocated. */
size_t cbBuffer = 0;
/* How much of the template data we have written out. */
size_t cchWritten = 0;
int rc = VINF_SUCCESS;
/* First of all read in the file. */
while (rc != VINF_EOF)
{
size_t cchRead;
if (cchTemplate == cbBuffer)
{
cbBuffer += READ_SIZE;
*ppachTemplate = (char *)RTMemRealloc((void *)*ppachTemplate,
cbBuffer);
}
if (!*ppachTemplate)
{
RTStrmPrintf(g_pStdErr, "Out of memory.\n");
return false;
}
rc = RTStrmReadEx(g_pStdIn, *ppachTemplate + cchTemplate,
cbBuffer - cchTemplate, &cchRead);
if (RT_FAILURE(rc))
{
RTStrmPrintf(g_pStdErr, "Error reading input: %Rrc\n", rc);
return false;
}
if (!cchRead)
rc = VINF_EOF;
cchTemplate += cchRead;
}
while (true)
{
/* Find the next '%' character if any and write out up to there (or the
* end if there is no '%'). */
char *pchNext = (char *) memchr((void *)(*ppachTemplate + cchWritten),
'%', cchTemplate - cchWritten);
size_t cchToWrite = pchNext
? pchNext - *ppachTemplate - cchWritten
: cchTemplate - cchWritten;
rc = RTStrmWrite(g_pStdOut, *ppachTemplate + cchWritten, cchToWrite);
if (RT_FAILURE(rc))
{
RTStrmPrintf(g_pStdErr, "Error writing output: %Rrc\n", rc);
return false;
}
cchWritten += cchToWrite;
if (!pchNext)
break;
/* And substitute any of our well-known strings. We favour code
* readability over efficiency here. */
if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
COMMAND, sizeof(COMMAND) - 1))
{
if (!pParameters->pcszCommand)
{
RTStrmPrintf(g_pStdErr, "--command not specified.\n");
return false;
}
if (!writeCommand(pParameters->enmFormat,
pParameters->pcszCommand))
return false;
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
ARGUMENTS, sizeof(ARGUMENTS) - 1))
{
if ( pParameters->pcszArguments
&& !writeQuoted(pParameters->enmFormat,
pParameters->pcszArguments))
return false;
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
DESCRIPTION, sizeof(DESCRIPTION) - 1))
{
if (!pParameters->pcszDescription)
{
RTStrmPrintf(g_pStdErr, "--description not specified.\n");
return false;
}
if (!writePrintableString(pParameters->enmFormat,
pParameters->pcszDescription))
return false;
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
SERVICE_NAME, sizeof(SERVICE_NAME) - 1))
{
if ( !pParameters->pcszCommand
&& !pParameters->pcszServiceName)
{
RTStrmPrintf(g_pStdErr, "Neither --command nor --service-name specified.\n");
return false;
}
if (pParameters->pcszServiceName)
{
if (!writePrintableString(pParameters->enmFormat,
pParameters->pcszServiceName))
return false;
}
else
{
const char *pcszFileName = RTPathFilename(pParameters->pcszCommand);
const char *pcszSuffix = RTPathSuffix(pParameters->pcszCommand);
char *pszName = RTStrDupN(pcszFileName,
pcszSuffix
? pcszSuffix - pcszFileName
: RTPATH_MAX);
bool fRc;
if (!pszName)
{
RTStrmPrintf(g_pStdErr, "Out of memory.\n");
return false;
}
fRc = writePrintableString(pParameters->enmFormat,
pszName);
RTStrFree(pszName);
if (!fRc)
return false;
}
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
HAVE_ONESHOT, sizeof(HAVE_ONESHOT) - 1))
{
if (!pParameters->fOneShot)
skipLine(*ppachTemplate, cchTemplate, &cchWritten);
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
HAVE_DAEMON, sizeof(HAVE_DAEMON) - 1))
{
if (pParameters->fOneShot)
skipLine(*ppachTemplate, cchTemplate, &cchWritten);
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
STOP_COMMAND, sizeof(STOP_COMMAND) - 1))
{
if ( pParameters->pcszStopCommand
&& !writeCommand(pParameters->enmFormat,
pParameters->pcszStopCommand))
return false;
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
STOP_ARGUMENTS, sizeof(STOP_ARGUMENTS) - 1))
{
if ( pParameters->pcszStopArguments
&& !writeQuoted(pParameters->enmFormat,
pParameters->pcszStopArguments))
return false;
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
HAVE_STOP_COMMAND, sizeof(HAVE_STOP_COMMAND) - 1))
{
if (!pParameters->pcszStopCommand)
skipLine(*ppachTemplate, cchTemplate, &cchWritten);
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
NO_STOP_COMMAND, sizeof(NO_STOP_COMMAND) - 1))
{
if (pParameters->pcszStopCommand)
skipLine(*ppachTemplate, cchTemplate, &cchWritten);
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
STATUS_COMMAND, sizeof(STATUS_COMMAND) - 1))
{
if ( pParameters->pcszStatusCommand
&& !writeCommand(pParameters->enmFormat,
pParameters->pcszStatusCommand))
return false;
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
STATUS_ARGUMENTS, sizeof(STATUS_ARGUMENTS) - 1))
{
if ( pParameters->pcszStatusArguments
&& !writeQuoted(pParameters->enmFormat,
pParameters->pcszStatusArguments))
return false;
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
HAVE_STATUS_COMMAND,
sizeof(HAVE_STATUS_COMMAND) - 1))
{
if (!pParameters->pcszStatusCommand)
skipLine(*ppachTemplate, cchTemplate, &cchWritten);
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
NO_STATUS_COMMAND, sizeof(NO_STATUS_COMMAND) - 1))
{
if (pParameters->pcszStatusCommand)
skipLine(*ppachTemplate, cchTemplate, &cchWritten);
}
else if (getSequence(*ppachTemplate, cchTemplate, &cchWritten,
"%%", 2))
{
rc = RTStrmPutCh(g_pStdOut, '%');
if (RT_FAILURE(rc))
{
RTStrmPrintf(g_pStdErr, "Error writing output: %Rrc\n", rc);
return false;
}
}
else
{
RTStrmPrintf(g_pStdErr, "Unknown substitution sequence in input at \"%.*s\"\n",
RT_MIN(16, cchTemplate - cchWritten),
*ppachTemplate + cchWritten);
return false;
}
}
return true;
}
