123 lines
3.3 KiB
C++
123 lines
3.3 KiB
C++
#ifndef PGPROC_H
|
|
#define PGPROC_H
|
|
|
|
#include "PgNamespaceObject.h"
|
|
#include "PgOwnedObject.h"
|
|
#include <QString>
|
|
#include <libpq-fe.h>
|
|
#include "Pgsql_Value.h"
|
|
#include <vector>
|
|
|
|
class Arg {
|
|
public:
|
|
enum Mode {
|
|
In, Out, InOut, Variadic, Table
|
|
};
|
|
|
|
Oid type;
|
|
Mode mode;
|
|
QString name;
|
|
QString def;
|
|
|
|
Arg(Oid t, Mode m, QString n, QString d)
|
|
: type(t), mode(m), name(n), def(d)
|
|
{}
|
|
};
|
|
|
|
|
|
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);
|
|
|
|
enum class ParallelMode {
|
|
Unsafe, // u, default
|
|
Restricted, // r,
|
|
Safe // s
|
|
};
|
|
|
|
void operator<<(ParallelMode &s, const Pgsql::Value &v);
|
|
|
|
/**
|
|
* @brief parallelModeToSql
|
|
* @param pm The parallel mode
|
|
* @param explicit_unsafe When false an empty string is return for unsage when true
|
|
* a "PARALLEL UNSAFE" is returned.
|
|
* @return
|
|
*/
|
|
QString parallelModeToSql(ParallelMode pm, bool explicit_unsafe = false);
|
|
|
|
|
|
class PgProc: public PgNamespaceObject, public PgOwnedObject {
|
|
public:
|
|
using PgNamespaceObject::PgNamespaceObject;
|
|
|
|
// Oid oid = InvalidOid; // oid
|
|
// QString name; // name
|
|
// Oid pronamespace = InvalidOid; // oid, namespace
|
|
// Oid owner = InvalidOid; // oid
|
|
Oid lang = InvalidOid; // oid
|
|
float cost = 0.f; // float4
|
|
float rows = 0.f; // float4
|
|
Oid variadic = InvalidOid; // oid
|
|
QString transform; // regproc
|
|
// bool isagg = false; // bool < 11
|
|
// bool iswindow = false; // bool << 11
|
|
ProcKind kind = ProcKind::Function; // >= 11
|
|
bool secdef = false; // bool
|
|
bool leakproof = false; // bool
|
|
bool isstrict = false; // bool
|
|
bool retset = false; // bool
|
|
char provolatile = '\0'; // char
|
|
ParallelMode parallel = ParallelMode::Unsafe; // char, version >= 9.6
|
|
int16_t nargs = 0; // int2
|
|
int16_t nargdefaults; // = 0; // int2
|
|
Oid rettype = InvalidOid; // oid
|
|
std::vector<Oid> trftypes; // oid[], version >= 9.5
|
|
QString src; // text
|
|
QString bin; // text
|
|
std::vector<QString> config; // text[]
|
|
QString acl; // aclitem[]
|
|
|
|
/// Converts the collection of arrays about the arguments to a single list of arguments
|
|
/// making it much easier to work with them correctly
|
|
void setArgs(
|
|
std::vector<Oid> argtypes,
|
|
std::vector<Oid> allargtypes,
|
|
std::vector<char> argmodes,
|
|
std::vector<QString> argnames,
|
|
std::optional<QString> argdefaults
|
|
);
|
|
const std::vector<Arg>& args() const;
|
|
|
|
// bool operator==(Oid _oid) const { return oid == _oid; }
|
|
// bool operator==(const QString &n) const { return name == n; }
|
|
// bool operator<(Oid _oid) const { return oid < _oid; }
|
|
// bool operator<(const PgProc &rhs) const { return oid < rhs.oid; }
|
|
|
|
const QString& createSql() const;
|
|
QString argListWithNames(bool multiline = false) const;
|
|
QString argSigList(const bool forScript = false) 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
|
|
// {
|
|
// return typname == wxT("\"trigger\"") || typname == wxT("trigger") || typname == wxT("event_trigger") || typname == wxT("\"event_trigger\""))
|
|
// }
|
|
private:
|
|
mutable QString createSqlCache;
|
|
|
|
std::vector<Arg> m_args;
|
|
};
|
|
|
|
#endif // PGPROC_H
|