Sql: GetSchColumn now guaranteed to be uppercase

git-svn-id: svn://ultimatepp.org/upp/trunk@4345 f0d560ea-af0d-0410-9eb7-867de7ffcac7
This commit is contained in:
cxl 2011-12-26 10:38:19 +00:00
parent 7dca0d0701
commit 79226efa5e
4 changed files with 135 additions and 139 deletions

View file

@ -1,133 +1,133 @@
#include "Sql.h"
NAMESPACE_UPP
struct SchTableInfo {
Vector<String> column;
Vector<String> ref_table;
Vector<String> ref_column;
String primary_key;
String prefix;
};
ArrayMap<String, SchTableInfo>& sSchTableInfo()
{
static ArrayMap<String, SchTableInfo> x;
return x;
}
void SchDbInfoTable(const char *table)
{
sSchTableInfo().GetAdd(table);
}
void SchDbInfoColumn(const char *name)
{
SchTableInfo& f = sSchTableInfo().Top();
f.column.Add(f.prefix + name);
f.ref_table.Add();
f.ref_column.Add();
if(IsNull(f.primary_key))
f.primary_key = name;
}
void SchDbInfoVar(void (*fn)(), const char *name)
{
SchTableInfo& f = sSchTableInfo().Top();
String h = f.prefix;
f.prefix << name << '$';
(*fn)();
f.prefix = h;
}
void SchDbInfoReferences(const char *table)
{
sSchTableInfo().Top().ref_table.Top() = table;
}
void SchDbInfoReferences(const char *table, const char *column)
{
SchDbInfoReferences(table);
sSchTableInfo().Top().ref_column.Top() = column;
}
void SchDbInfoPrimaryKey()
{
SchTableInfo& f = sSchTableInfo().Top();
f.primary_key = f.column.Top();
}
void SchDbInfoColumnArray(const char *name, int items)
{
for(int i = 0; i < items; i++)
SchDbInfoColumn(String().Cat() << name << i);
}
const SchTableInfo& GetSchTableInfo(const String& table)
{
static SchTableInfo sSchTableInfoZero;
return sSchTableInfo().Get(~table, sSchTableInfoZero);
}
SqlBool Join(const String& tab1, const String& as1, const String& tab2, const String& as2)
{
const SchTableInfo& t1 = GetSchTableInfo(tab1);
const SchTableInfo& t2 = GetSchTableInfo(tab2);
for(int i = 0; i < t1.ref_table.GetCount(); i++)
if(t1.ref_table[i] == tab2)
return SqlId(t1.column[i]).Of(SqlId(as1)) == SqlId(t2.primary_key).Of(SqlId(as2));
for(int i = 0; i < t2.ref_table.GetCount(); i++)
if(t2.ref_table[i] == tab1)
return SqlId(t2.column[i]).Of(SqlId(as1)) == SqlId(t1.primary_key).Of(SqlId(as2));
return SqlBool::False();
}
int sChrf(int c)
{
return c < 32 ? ' ' : c;
}
StaticMutex sM;
Vector<String> GetSchColumns(const String& table)
{
Mutex::Lock __(sM);
return Vector<String>(GetSchTableInfo(table).column, 0);
}
SqlBool FindSchJoin(const String& tables)
{
Mutex::Lock __(sM);
static VectorMap<String, SqlBool> cache;
if(cache.GetCount() > 20000) // Defend against unlikely dynamic Join permutations
cache.Clear();
int q = cache.Find(tables);
if(q >= 0)
return cache[q];
Vector<String> s = Split(Filter(tables, sChrf), ',');
Vector<String> as;
Vector<String> table;
for(int i = 0; i < s.GetCount(); i++) {
Vector<String> ss = Split(s[i], ' ');
if(ss.GetCount()) {
table.Add(ss[0]);
as.Add(ss.Top());
}
}
if(table.GetCount() >= 2) {
String tab1 = table.Top();
String as1 = as.Top();
for(int i = 0; i < table.GetCount() - 1; i++) {
SqlBool b = Join(tab1, as1, table[i], as[i]);
if(!b.IsFalse()) {
cache.Add(tables, b);
return b;
}
}
}
NEVER_("Schema join not found");
return SqlBool::False();
}
END_UPP_NAMESPACE
#include "Sql.h"
NAMESPACE_UPP
struct SchTableInfo {
Vector<String> column;
Vector<String> ref_table;
Vector<String> ref_column;
String primary_key;
String prefix;
};
ArrayMap<String, SchTableInfo>& sSchTableInfo()
{
static ArrayMap<String, SchTableInfo> x;
return x;
}
void SchDbInfoTable(const char *table)
{
sSchTableInfo().GetAdd(table);
}
void SchDbInfoColumn(const char *name)
{
SchTableInfo& f = sSchTableInfo().Top();
f.column.Add(ToUpper(f.prefix + name));
f.ref_table.Add();
f.ref_column.Add();
if(IsNull(f.primary_key))
f.primary_key = name;
}
void SchDbInfoVar(void (*fn)(), const char *name)
{
SchTableInfo& f = sSchTableInfo().Top();
String h = f.prefix;
f.prefix << name << '$';
(*fn)();
f.prefix = h;
}
void SchDbInfoReferences(const char *table)
{
sSchTableInfo().Top().ref_table.Top() = table;
}
void SchDbInfoReferences(const char *table, const char *column)
{
SchDbInfoReferences(table);
sSchTableInfo().Top().ref_column.Top() = column;
}
void SchDbInfoPrimaryKey()
{
SchTableInfo& f = sSchTableInfo().Top();
f.primary_key = f.column.Top();
}
void SchDbInfoColumnArray(const char *name, int items)
{
for(int i = 0; i < items; i++)
SchDbInfoColumn(String().Cat() << name << i);
}
const SchTableInfo& GetSchTableInfo(const String& table)
{
static SchTableInfo sSchTableInfoZero;
return sSchTableInfo().Get(~table, sSchTableInfoZero);
}
SqlBool Join(const String& tab1, const String& as1, const String& tab2, const String& as2)
{
const SchTableInfo& t1 = GetSchTableInfo(tab1);
const SchTableInfo& t2 = GetSchTableInfo(tab2);
for(int i = 0; i < t1.ref_table.GetCount(); i++)
if(t1.ref_table[i] == tab2)
return SqlId(t1.column[i]).Of(SqlId(as1)) == SqlId(t2.primary_key).Of(SqlId(as2));
for(int i = 0; i < t2.ref_table.GetCount(); i++)
if(t2.ref_table[i] == tab1)
return SqlId(t2.column[i]).Of(SqlId(as1)) == SqlId(t1.primary_key).Of(SqlId(as2));
return SqlBool::False();
}
int sChrf(int c)
{
return c < 32 ? ' ' : c;
}
StaticMutex sM;
Vector<String> GetSchColumns(const String& table)
{
Mutex::Lock __(sM);
return Vector<String>(GetSchTableInfo(table).column, 0);
}
SqlBool FindSchJoin(const String& tables)
{
Mutex::Lock __(sM);
static VectorMap<String, SqlBool> cache;
if(cache.GetCount() > 20000) // Defend against unlikely dynamic Join permutations
cache.Clear();
int q = cache.Find(tables);
if(q >= 0)
return cache[q];
Vector<String> s = Split(Filter(tables, sChrf), ',');
Vector<String> as;
Vector<String> table;
for(int i = 0; i < s.GetCount(); i++) {
Vector<String> ss = Split(s[i], ' ');
if(ss.GetCount()) {
table.Add(ss[0]);
as.Add(ss.Top());
}
}
if(table.GetCount() >= 2) {
String tab1 = table.Top();
String as1 = as.Top();
for(int i = 0; i < table.GetCount() - 1; i++) {
SqlBool b = Join(tab1, as1, table[i], as[i]);
if(!b.IsFalse()) {
cache.Add(tables, b);
return b;
}
}
}
NEVER_("Schema join not found");
return SqlBool::False();
}
END_UPP_NAMESPACE

View file

@ -286,7 +286,6 @@ bool Sql::Fetch(ValueMap& row) {
}
struct sReadFields : public FieldOperator {
int pi;
Sql *sql;
void Field(const char *name, Ref f) {
@ -298,7 +297,6 @@ void Sql::Get(Fields fo)
{
sReadFields ff;
ff.sql = this;
ff.pi = 0;
fo(ff);
}

View file

@ -69,8 +69,6 @@ void SqlCtrls::Add(SqlId id, Ctrl& ctrl) {
void SqlCtrls::Table(Ctrl& dlg, SqlId table)
{
Vector<String> col = GetSchColumns(~table);
for(int i = 0; i < col.GetCount(); i++)
col[i] = ToUpper(col[i]);
for(Ctrl *q = dlg.GetFirstChild(); q; q = q->GetNext()) {
String id = ToUpper(q->GetLayoutId());
if(!dynamic_cast<Button *>(q) && !dynamic_cast<Label *>(q) && FindIndex(col, id) >= 0)

View file

@ -5,9 +5,9 @@ NAMESPACE_UPP
int SqlError(
#ifdef PLATFORM_WIN32
HWND parent,
HWND parent,
#endif
const char *text, const char *error, const char *statement, bool retry) {
const char *text, const char *error, const char *statement, bool retry) {
WithSqlErrorLayout<TopWindow> r;
CtrlLayoutCancel(r, t_("Database error"));
r.text = text;