pg11: pg_proc, type of function is stored differently from pg11 forward

Followed the more structured approach of pg11 in combining the different types into a kind field
when reading from older versions we migrate the old fields to the new field. Change in 11
is because of PROCEDURE support. However full PROCEDURE support will be its own change and is
registered as issue #37
This commit is contained in:
eelke 2018-12-23 08:48:45 +01:00
parent a88af1ac11
commit 43f8117bbd
3 changed files with 60 additions and 9 deletions

View file

@ -73,7 +73,25 @@ namespace {
return res; return res;
} }
}
void operator<<(ProcKind &s, const Pgsql::Value &v)
{
const char *c = v.c_str();
switch (*c) {
case 'f':
s = ProcKind::Function;
break;
case 'p':
s = ProcKind::Procedure;
break;
case 'a':
s = ProcKind::Aggregate;
break;
case 'w':
s = ProcKind::Window;
break;
}
} }
void PgProc::setArgs( void PgProc::setArgs(
@ -239,7 +257,7 @@ QString PgProc::createSql() const
sql += "\n LANGUAGE " % language % " "; sql += "\n LANGUAGE " % language % " ";
// if (GetConnection()->BackendMinimumVersion(8, 4) && GetIsWindow()) // if (GetConnection()->BackendMinimumVersion(8, 4) && GetIsWindow())
// sql += wxT("WINDOW "); // sql += wxT("WINDOW ");
if (iswindow) if (isWindow())
sql += "WINDOW "; sql += "WINDOW ";
// sql += GetVolatility(); // sql += GetVolatility();
sql += volatility(); sql += volatility();

View file

@ -25,6 +25,15 @@ public:
}; };
enum class ProcKind {
Function, // f Normal function
Procedure, // p stored procedure, new in 11 can use transactions executed with CALL
Aggregate, // a
Window // w
};
void operator<<(ProcKind &s, const Pgsql::Value &v);
class PgProc: public PgNamespaceObject, public PgOwnedObject { class PgProc: public PgNamespaceObject, public PgOwnedObject {
public: public:
using PgNamespaceObject::PgNamespaceObject; using PgNamespaceObject::PgNamespaceObject;
@ -38,8 +47,9 @@ public:
float rows = 0.f; // float4 float rows = 0.f; // float4
Oid variadic = InvalidOid; // oid Oid variadic = InvalidOid; // oid
QString transform; // regproc QString transform; // regproc
bool isagg = false; // bool // bool isagg = false; // bool < 11
bool iswindow = false; // bool // bool iswindow = false; // bool << 11
ProcKind kind = ProcKind::Function; // >= 11
bool secdef = false; // bool bool secdef = false; // bool
bool leakproof = false; // bool bool leakproof = false; // bool
bool isstrict = false; // bool bool isstrict = false; // bool
@ -77,6 +87,11 @@ public:
QString volatility() const; QString volatility() const;
bool isFunction() const { return kind == ProcKind::Function; }
bool isProcedure() const { return kind == ProcKind::Procedure; }
bool isAggregate() const { return kind == ProcKind::Aggregate; }
bool isWindow() const { return kind == ProcKind::Window; }
// bool isTrigger() const // bool isTrigger() const
// { // {
// return typname == wxT("\"trigger\"") || typname == wxT("trigger") || typname == wxT("event_trigger") || typname == wxT("\"event_trigger\"")) // return typname == wxT("\"trigger\"") || typname == wxT("trigger") || typname == wxT("event_trigger") || typname == wxT("\"event_trigger\""))

View file

@ -9,16 +9,18 @@ std::string PgProcContainer::getLoadQuery() const
{ {
std::string column_list = std::string column_list =
"oid,proname,pronamespace,proowner,prolang,procost,prorows," "oid,proname,pronamespace,proowner,prolang,procost,prorows,"
"provariadic,protransform,proisagg,proiswindow,prosecdef,proleakproof," "provariadic,protransform,prosecdef,proleakproof,"
"proisstrict,proretset,provolatile,pronargs,pronargdefaults,prorettype," "proisstrict,proretset,provolatile,pronargs,pronargdefaults,prorettype,"
"proargtypes,proallargtypes,proargmodes,proargnames,proargdefaults," "proargtypes,proallargtypes,proargmodes,proargnames,proargdefaults,"
"prosrc,probin,proconfig,proacl"; "prosrc,probin,proconfig,proacl";
if (minimumVersion(90500)) { if (minimumVersion(90500))
column_list += ",protrftypes"; column_list += ",protrftypes";
} if (minimumVersion(90600))
if (minimumVersion(90600)) {
column_list += ",proparallel"; column_list += ",proparallel";
} if (minimumVersion(110000))
column_list += ",prokind";
else
column_list += ",proisagg,proiswindow";
return "SELECT " + column_list + " FROM pg_proc"; return "SELECT " + column_list + " FROM pg_proc";
} }
@ -31,10 +33,12 @@ PgProc PgProcContainer::loadElem(const Pgsql::Row &row)
Oid namespace_oid = col.nextValue(); Oid namespace_oid = col.nextValue();
Oid owner_oid = col.nextValue(); Oid owner_oid = col.nextValue();
//ProcKind
PgProc v(m_catalog, oid, name, namespace_oid); PgProc v(m_catalog, oid, name, namespace_oid);
v.setOwnerOid(m_catalog, owner_oid); v.setOwnerOid(m_catalog, owner_oid);
col >> v.lang >> v.cost >> v.rows col >> v.lang >> v.cost >> v.rows
>> v.variadic >> v.transform >> v.isagg >> v.iswindow >> v.secdef >> v.leakproof >> v.variadic >> v.transform >> v.secdef >> v.leakproof
>> v.isstrict >> v.retset >> v.provolatile >> v.nargs >> v.nargdefaults >> v.isstrict >> v.retset >> v.provolatile >> v.nargs >> v.nargdefaults
>> v.rettype; >> v.rettype;
@ -55,6 +59,20 @@ PgProc PgProcContainer::loadElem(const Pgsql::Row &row)
if (minimumVersion(90600)) { if (minimumVersion(90600)) {
col >> v.parallel; col >> v.parallel;
} }
if (minimumVersion(110000))
col >> v.kind;
else {
bool agg, window;
col >> agg >> window;
if (agg)
v.kind = ProcKind::Aggregate;
else if (window)
v.kind = ProcKind::Window;
else {
v.kind = ProcKind::Function;
}
// Only 11 and higher has ProcKind::Procedure!
}
return v; return v;
} }