pgis_fromflatgeobuf()

Datum pgis_fromflatgeobuf

(

PG_FUNCTION_ARGS

)

Definition at line 159 of file lwgeom_in_flatgeobuf.c.

{
        FuncCallContext *funcctx;
 
        TupleDesc tupdesc;
        bytea *data;
        MemoryContext oldcontext;
 
        struct flatgeobuf_decode_ctx *ctx;
 
        if (SRF_IS_FIRSTCALL()) {
                funcctx = SRF_FIRSTCALL_INIT();
                oldcontext = MemoryContextSwitchTo(funcctx->multi_call_memory_ctx);
 
                funcctx->max_calls = 0;
 
                if (get_call_result_type(fcinfo, NULL, &tupdesc) != TYPEFUNC_COMPOSITE)
                        ereport(ERROR,
                                        (errcode(ERRCODE_FEATURE_NOT_SUPPORTED),
                                         errmsg("first argument of function must be composite type")));
 
                data = PG_GETARG_BYTEA_PP(1);
 
                ctx = palloc0(sizeof(*ctx));
                ctx->tupdesc = tupdesc;
                ctx->ctx = palloc0(sizeof(flatgeobuf_ctx));
                ctx->ctx->size = VARSIZE_ANY_EXHDR(data);
                POSTGIS_DEBUGF(3, "VARSIZE_ANY_EXHDR %lld", ctx->ctx->size);
                ctx->ctx->buf = palloc(ctx->ctx->size);
                memcpy(ctx->ctx->buf, VARDATA_ANY(data), ctx->ctx->size);
                ctx->ctx->offset = 0;
                ctx->done = false;
                ctx->fid = 0;
 
                funcctx->user_fctx = ctx;
 
                if (ctx->ctx->size == 0) {
                        POSTGIS_DEBUG(2, "no data");
                        MemoryContextSwitchTo(oldcontext);
                        SRF_RETURN_DONE(funcctx);
                }
 
                flatgeobuf_check_magicbytes(ctx);
                flatgeobuf_decode_header(ctx->ctx);
 
                POSTGIS_DEBUGF(2, "header decoded now at offset %lld", ctx->ctx->offset);
 
                if (ctx->ctx->size == ctx->ctx->offset) {
                        POSTGIS_DEBUGF(2, "no feature data offset %lld", ctx->ctx->offset);
                        MemoryContextSwitchTo(oldcontext);
                        SRF_RETURN_DONE(funcctx);
                }
 
                // TODO: get table and verify structure against header
                MemoryContextSwitchTo(oldcontext);
        }
 
        funcctx = SRF_PERCALL_SETUP();
        ctx = funcctx->user_fctx;
 
        if (!ctx->done) {
                flatgeobuf_decode_row(ctx);
                POSTGIS_DEBUG(2, "Calling SRF_RETURN_NEXT");
                SRF_RETURN_NEXT(funcctx, ctx->result);
        } else {
                POSTGIS_DEBUG(2, "Calling SRF_RETURN_DONE");
                SRF_RETURN_DONE(funcctx);
        }
}

References flatgeobuf_decode_ctx::ctx, flatgeobuf_check_magicbytes(), flatgeobuf_decode_row(), and flatgeobuf_decode_ctx::tupdesc.

Here is the call graph for this function: