Datum
install_rcmd(PG_FUNCTION_ARGS)
{
char *cmd = PG_TEXT_GET_STR(PG_GETARG_TEXT_P(0));
load_r_cmd(cmd);
PG_RETURN_TEXT_P(PG_STR_GET_TEXT("OK"));
}
Datum
plr_set_display(PG_FUNCTION_ARGS)
{
char *display = PG_TEXT_GET_STR(PG_GETARG_TEXT_P(0));
size_t d_len = strlen(display);
if (d_len)
{
char *denv;
MemoryContext oldcontext;
/* Needs to live until/unless we explicitly delete it */
oldcontext = MemoryContextSwitchTo(TopMemoryContext);
denv = palloc(9 + d_len);
MemoryContextSwitchTo(oldcontext);
sprintf(denv, "DISPLAY=%s", display);
putenv(denv);
}
PG_RETURN_TEXT_P(PG_STR_GET_TEXT("OK"));
}
Datum
plr_set_rhome(PG_FUNCTION_ARGS)
{
char *rhome = PG_TEXT_GET_STR(PG_GETARG_TEXT_P(0));
size_t rh_len = strlen(rhome);
if (rh_len)
{
char *rhenv;
MemoryContext oldcontext;
/* Needs to live until/unless we explicitly delete it */
oldcontext = MemoryContextSwitchTo(TopMemoryContext);
rhenv = palloc(8 + rh_len);
MemoryContextSwitchTo(oldcontext);
sprintf(rhenv, "R_HOME=%s", rhome);
putenv(rhenv);
}
PG_RETURN_TEXT_P(PG_STR_GET_TEXT("OK"));
}
/*
* plr_quote_literal() - quote literal strings that are to
* be used in SPI_exec query strings
*/
SEXP
plr_quote_literal(SEXP rval)
{
const char *value;
text *value_text;
text *result_text;
SEXP result;
/* extract the C string */
PROTECT(rval = AS_CHARACTER(rval));
value = CHAR(STRING_ELT(rval, 0));
/* convert using the pgsql quote_literal function */
value_text = PG_STR_GET_TEXT(value);
result_text = DatumGetTextP(DirectFunctionCall1(quote_literal, PointerGetDatum(value_text)));
/* copy result back into an R object */
PROTECT(result = NEW_CHARACTER(1));
SET_STRING_ELT(result, 0, COPY_TO_USER_STRING(PG_TEXT_GET_STR(result_text)));
UNPROTECT(2);
return result;
}
