dump_module.c

#include <postgres.h>
#include <fmgr.h>
#include <utils/builtins.h>

#include <plpgsql.h>

#include "string_helper.h"
#include "dump_plpgsql_function.h"
#include "dump_sql_parse_tree.h"


PG_MODULE_MAGIC;

PG_FUNCTION_INFO_V1(dump_sql_parse_tree);

Datum dump_sql_parse_tree( PG_FUNCTION_ARGS )
{
  char * query = text_to_cstring( PG_GETARG_TEXT_P( 0 ) );
  DumpContext * context = new_dump_context();
//  debug_query_string = query;

  xml_pi( context, "xml", "version", "%.1f", 1.0, NULL );
  dump_sql_parse_tree_internal( context, query );

  if ( *context->output )
    PG_RETURN_TEXT_P( cstring_to_text( *context->output ) );
  else 
    PG_RETURN_NULL();
}


PG_FUNCTION_INFO_V1(dump_plpgsql_function);

Datum dump_plpgsql_function( PG_FUNCTION_ARGS )
{
  DumpContext * context = new_dump_context();
  xml_pi( context, "xml", "version", "%.1f", 1.0, NULL );
  const char * result = dump_plpgsql_function_internal( context, PG_GETARG_OID( 0 ) );

  PG_RETURN_TEXT_P( cstring_to_text( result ) );
}

void _PG_init()
{
  string_helper_init();
  plpgsql_HashTableInit();
}