From 02692e1a62eb922b462034663441b7825885101b Mon Sep 17 00:00:00 2001 From: Alvaro Herrera Date: Thu, 25 Sep 2014 15:55:43 -0300 Subject: [PATCH 10/44] deparse: Support CREATE TYPE AS ENUM --- src/backend/tcop/deparse_utility.c | 34 +++++++++++++++++++++++++++++++++- 1 file changed, 33 insertions(+), 1 deletion(-) diff --git a/src/backend/tcop/deparse_utility.c b/src/backend/tcop/deparse_utility.c index b21845b..50a8d10 100644 --- a/src/backend/tcop/deparse_utility.c +++ b/src/backend/tcop/deparse_utility.c @@ -886,6 +886,38 @@ deparse_CompositeTypeStmt(Oid objectId, Node *parsetree) } /* + * deparse_CreateEnumStmt + * Deparse a CreateEnumStmt (CREATE TYPE AS ENUM) + * + * Given a type OID and the parsetree that created it, return an ObjTree + * representing the creation command. + */ +static ObjTree * +deparse_CreateEnumStmt(Oid objectId, Node *parsetree) +{ + CreateEnumStmt *node = (CreateEnumStmt *) parsetree; + ObjTree *enumtype; + List *values; + ListCell *cell; + + enumtype = new_objtree_VA("CREATE TYPE %{identity}D AS ENUM (%{values:, }L)", + 0); + append_object_object(enumtype, "identity", + new_objtree_for_qualname_id(TypeRelationId, + objectId)); + values = NIL; + foreach(cell, node->vals) + { + Value *val = (Value *) lfirst(cell); + + values = lappend(values, new_string_object(strVal(val))); + } + append_array_object(enumtype, "values", values); + + return enumtype; +} + +/* * Handle deparsing of simple commands. * * This function contains a large switch that mirrors that in @@ -988,7 +1020,7 @@ deparse_simple_command(StashedCommand *cmd) break; case T_CreateEnumStmt: /* CREATE TYPE AS ENUM */ - elog(ERROR, "unimplemented deparse of %s", CreateCommandTag(parsetree)); + command = deparse_CreateEnumStmt(objectId, parsetree); break; case T_CreateRangeStmt: /* CREATE TYPE AS RANGE */ -- 2.1.4