-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add pinecone_helpers.c and pinecone_indexes() function
- Loading branch information
1 parent
078a42e
commit b861c24
Showing
5 changed files
with
162 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,106 @@ | ||
| #include "postgres.h" | ||
| #include "fmgr.h" | ||
| #include <nodes/execnodes.h> | ||
| #include "funcapi.h" | ||
| #include "pinecone_api.h" | ||
| #include "pinecone.h" | ||
| #include "cJSON.h" | ||
| #include "utils/builtins.h" | ||
|
|
||
| PGDLLEXPORT PG_FUNCTION_INFO_V1(pinecone_indexes); | ||
| Datum | ||
| pinecone_indexes(PG_FUNCTION_ARGS) { | ||
| ReturnSetInfo *rsinfo = (ReturnSetInfo *) fcinfo->resultinfo; | ||
| Tuplestorestate *tupstore; | ||
| TupleDesc tupdesc; | ||
| MemoryContext per_query_ctx, oldcontext; | ||
| cJSON *indexes; | ||
| cJSON *index; | ||
|
|
||
| /* check to see if caller supports us returning a tuplestore */ | ||
| if (rsinfo == NULL || !IsA(rsinfo, ReturnSetInfo)) | ||
| ereport(ERROR, | ||
| (errcode(ERRCODE_FEATURE_NOT_SUPPORTED), | ||
| errmsg("set-valued function called in context that cannot accept a set"))); | ||
| if (!(rsinfo->allowedModes & SFRM_Materialize)) | ||
| ereport(ERROR, | ||
| (errcode(ERRCODE_FEATURE_NOT_SUPPORTED), | ||
| errmsg("materialize mode required, but it is not allowed in this context"))); | ||
|
|
||
| /* get a tuple descriptor for our result type */ | ||
| switch (get_call_result_type(fcinfo, NULL, &tupdesc)) | ||
| { | ||
| case TYPEFUNC_COMPOSITE: | ||
| /* success */ | ||
| break; | ||
| case TYPEFUNC_RECORD: | ||
| /* failed to determine actual type of RECORD */ | ||
| ereport(ERROR, | ||
| (errcode(ERRCODE_FEATURE_NOT_SUPPORTED), | ||
| errmsg("function returning record called in context that cannot accept type record"))); | ||
| break; | ||
| default: | ||
| /* result type isn't a tuple */ | ||
| ereport(ERROR, | ||
| (errcode(ERRCODE_FEATURE_NOT_SUPPORTED), | ||
| errmsg("function result type must be a row type"))); | ||
| break; | ||
| } | ||
|
|
||
| // create a tuple store and tuple descriptor in the per-query context | ||
| per_query_ctx = rsinfo->econtext->ecxt_per_query_memory; | ||
| oldcontext = MemoryContextSwitchTo(per_query_ctx); | ||
| /* create a tuple store */ | ||
| tupdesc = CreateTupleDescCopy(tupdesc); | ||
| tupstore = tuplestore_begin_heap(true, false, 100); | ||
| MemoryContextSwitchTo(oldcontext); | ||
|
|
||
| // validate the api key | ||
| if (pinecone_api_key == NULL || strlen(pinecone_api_key) == 0) { | ||
| ereport(ERROR, (errmsg("Pinecone API key is not set"))); | ||
| } | ||
| indexes = list_indexes(pinecone_api_key); | ||
| elog(DEBUG1, "Indexes: %s", cJSON_Print(indexes)); | ||
|
|
||
| cJSON_ArrayForEach(index, indexes) { | ||
| Datum values[30]; | ||
| bool nulls[30]; | ||
| HeapTuple tuple; | ||
| for (int i = 0; i < tupdesc->natts; i++) { | ||
| Form_pg_attribute attr = TupleDescAttr(tupdesc, i); | ||
| char* name = NameStr(attr->attname); | ||
| Oid type = attr->atttypid; | ||
| cJSON *value = cJSON_GetObjectItem(index, name); | ||
| switch (type) { | ||
| case INT4OID: | ||
| nulls[i] = value == NULL || !cJSON_IsNumber(value); | ||
| if (!nulls[i]) values[i] = Int32GetDatum((int)cJSON_GetNumberValue(value)); | ||
| break; | ||
| case TEXTOID: | ||
| nulls[i] = value == NULL || !cJSON_IsString(value); | ||
| if (!nulls[i]) values[i] = PointerGetDatum(cstring_to_text(cJSON_GetStringValue(value))); | ||
| break; | ||
| case BOOLOID: | ||
| nulls[i] = value == NULL || !cJSON_IsBool(value); | ||
| if (!nulls[i]) values[i] = BoolGetDatum(cJSON_IsTrue(value)); | ||
| break; | ||
| case JSONOID: | ||
| nulls[i] = value == NULL; | ||
| if (!nulls[i]) values[i] = PointerGetDatum(cstring_to_text(cJSON_Print(value))); | ||
| break; | ||
| default: | ||
| ereport(ERROR, (errmsg("Unsupported type"))); | ||
| break; | ||
| } | ||
| } | ||
| tuple = heap_form_tuple(tupdesc, values, nulls); | ||
| tuplestore_puttuple(tupstore, tuple); | ||
| heap_freetuple(tuple); | ||
| } | ||
|
|
||
| rsinfo->returnMode = SFRM_Materialize; | ||
| rsinfo->setResult = tupstore; | ||
| rsinfo->setDesc = tupdesc; | ||
| // when returning a set, we must return a null Datum | ||
| return (Datum) 0; | ||
| } |