mirror of
https://github.com/neondatabase/neon.git
synced 2026-05-21 07:00:38 +00:00
73a8c97ac8
proceeding https://github.com/neondatabase/neon/pull/7010, close https://github.com/neondatabase/neon/issues/6188 ## Summary of changes This pull request (should) fix all warnings except `-Wdeclaration-after-statement` in the neon extension compilation. --------- Signed-off-by: Alex Chi Z <chi@neon.tech>
95 lines
2.5 KiB
C
95 lines
2.5 KiB
C
/*-------------------------------------------------------------------------
|
|
*
|
|
* extension_server.c
|
|
* Request compute_ctl to download extension files.
|
|
*
|
|
* IDENTIFICATION
|
|
* contrib/neon/extension_server.c
|
|
*
|
|
*-------------------------------------------------------------------------
|
|
*/
|
|
#include "postgres.h"
|
|
|
|
#include <curl/curl.h>
|
|
|
|
#include "utils/guc.h"
|
|
|
|
#include "extension_server.h"
|
|
#include "neon_utils.h"
|
|
|
|
static int extension_server_port = 0;
|
|
|
|
static download_extension_file_hook_type prev_download_extension_file_hook = NULL;
|
|
|
|
/*
|
|
* to download all SQL (and data) files for an extension:
|
|
* curl -X POST http://localhost:8080/extension_server/postgis
|
|
* it covers two possible extension files layouts:
|
|
* 1. extension_name--version--platform.sql
|
|
* 2. extension_name/extension_name--version.sql
|
|
* extension_name/extra_files.csv
|
|
* to download specific library file:
|
|
* curl -X POST http://localhost:8080/extension_server/postgis-3.so?is_library=true
|
|
*/
|
|
static bool
|
|
neon_download_extension_file_http(const char *filename, bool is_library)
|
|
{
|
|
static CURL *handle = NULL;
|
|
|
|
CURLcode res;
|
|
char *compute_ctl_url;
|
|
bool ret = false;
|
|
|
|
if (handle == NULL)
|
|
{
|
|
handle = alloc_curl_handle();
|
|
|
|
curl_easy_setopt(handle, CURLOPT_CUSTOMREQUEST, "POST");
|
|
curl_easy_setopt(handle, CURLOPT_TIMEOUT, 3L /* seconds */ );
|
|
}
|
|
|
|
compute_ctl_url = psprintf("http://localhost:%d/extension_server/%s%s",
|
|
extension_server_port, filename, is_library ? "?is_library=true" : "");
|
|
|
|
elog(LOG, "Sending request to compute_ctl: %s", compute_ctl_url);
|
|
|
|
curl_easy_setopt(handle, CURLOPT_URL, compute_ctl_url);
|
|
|
|
/* Perform the request, res will get the return code */
|
|
res = curl_easy_perform(handle);
|
|
/* Check for errors */
|
|
if (res == CURLE_OK)
|
|
{
|
|
ret = true;
|
|
}
|
|
else
|
|
{
|
|
/*
|
|
* Don't error here because postgres will try to find the file and will
|
|
* fail with some proper error message if it's not found.
|
|
*/
|
|
elog(WARNING, "neon_download_extension_file_http failed: %s\n", curl_easy_strerror(res));
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
void
|
|
pg_init_extension_server()
|
|
{
|
|
/* Port to connect to compute_ctl on localhost */
|
|
/* to request extension files. */
|
|
DefineCustomIntVariable("neon.extension_server_port",
|
|
"connection string to the compute_ctl",
|
|
NULL,
|
|
&extension_server_port,
|
|
0, 0, INT_MAX,
|
|
PGC_POSTMASTER,
|
|
0, /* no flags required */
|
|
NULL, NULL, NULL);
|
|
|
|
/* set download_extension_file_hook */
|
|
prev_download_extension_file_hook = download_extension_file_hook;
|
|
download_extension_file_hook = neon_download_extension_file_http;
|
|
}
|