mirror of
https://github.com/ultimatepp/ultimatepp.git
synced 2026-05-17 14:16:10 -06:00
133 lines
2.7 KiB
C++
133 lines
2.7 KiB
C++
#include "Sql.h"
|
|
|
|
NAMESPACE_UPP
|
|
|
|
SqlMassInsert::~SqlMassInsert()
|
|
{
|
|
Flush();
|
|
}
|
|
|
|
SqlMassInsert& SqlMassInsert::operator()(SqlId col, const Value& val)
|
|
{
|
|
if(pos == 0) {
|
|
cache.Add().nulls = 0;
|
|
cache.Top();
|
|
}
|
|
if(cache.GetCount() == 1)
|
|
column.Add(~col);
|
|
else
|
|
ASSERT(column[pos] == col.Quoted() || column[pos] == ~col);
|
|
Row& r = cache.Top();
|
|
r.value.Add(val);
|
|
if(IsNull(val))
|
|
r.nulls |= (1 << pos);
|
|
pos++;
|
|
ASSERT(pos < 30);
|
|
return *this;
|
|
}
|
|
|
|
SqlMassInsert& SqlMassInsert::EndRow(SqlBool remove)
|
|
{
|
|
cache.Top().remove = remove;
|
|
if(cache.GetCount() && cache[0].value.GetCount() * cache.GetCount() > 5000)
|
|
Flush();
|
|
ASSERT(column.GetCount() == pos);
|
|
pos = 0;
|
|
return *this;
|
|
}
|
|
|
|
void SqlMassInsert::Flush()
|
|
{
|
|
const dword DONE = 0xffffffff;
|
|
if(cache.GetCount() == 0)
|
|
return;
|
|
if(use_transaction)
|
|
sql.GetSession().Begin();
|
|
SqlBool remove;
|
|
bool doremove = false;
|
|
for(int ii = 0; ii < cache.GetCount(); ii++) {
|
|
SqlBool rm = cache[ii].remove;
|
|
if(!rm.IsEmpty()) {
|
|
doremove = true;
|
|
remove = remove || rm;
|
|
}
|
|
}
|
|
if(doremove)
|
|
sql * Delete(table).Where(remove);
|
|
String insert;
|
|
int dialect = sql.GetDialect();
|
|
if(dialect == MY_SQL || dialect == PGSQL) {
|
|
insert << "insert into " + ~table + '(';
|
|
for(int i = 0; i < column.GetCount(); i++) {
|
|
if(i)
|
|
insert << ", ";
|
|
insert << column[i];
|
|
}
|
|
insert << ") values ";
|
|
for(int i = 0; i < cache.GetCount(); i++) {
|
|
Row& r = cache[i];
|
|
if(i)
|
|
insert << ", ";
|
|
insert << "(";
|
|
for(int i = 0; i < r.value.GetCount(); i++) {
|
|
if(i)
|
|
insert << ", ";
|
|
insert << SqlCompile(dialect, SqlFormat(r.value[i]));
|
|
}
|
|
insert << ")";
|
|
}
|
|
}
|
|
else
|
|
for(int ii = 0; ii < cache.GetCount(); ii++) {
|
|
dword nulls = cache[ii].nulls;
|
|
if(nulls != DONE) {
|
|
insert << "insert into " + ~table + '(';
|
|
bool nextcol = false;
|
|
for(int i = 0; i < column.GetCount(); i++) {
|
|
if(!(nulls & (1 << i))) {
|
|
if(nextcol)
|
|
insert << ", ";
|
|
nextcol = true;
|
|
insert << column[i];
|
|
}
|
|
}
|
|
insert << ')';
|
|
bool nextsel = false;
|
|
for(int i = ii; i < cache.GetCount(); i++) {
|
|
Row& r = cache[i];
|
|
if(r.nulls == nulls) {
|
|
r.nulls = DONE;
|
|
if(nextsel)
|
|
insert << " union all";
|
|
nextsel = true;
|
|
insert << " select ";
|
|
bool nextval = false;
|
|
for(int i = 0; i < r.value.GetCount(); i++)
|
|
if(!(nulls & (1 << i))) {
|
|
if(nextval)
|
|
insert << ", ";
|
|
nextval = true;
|
|
insert << SqlCompile(dialect, SqlFormat(r.value[i]));
|
|
}
|
|
if(dialect == ORACLE)
|
|
insert << " from dual";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
sql.Execute(insert);
|
|
if(sql.WasError()) {
|
|
error = true;
|
|
if(use_transaction)
|
|
sql.GetSession().Rollback();
|
|
}
|
|
else
|
|
if(use_transaction)
|
|
sql.GetSession().Commit();
|
|
cache.Clear();
|
|
column.Clear();
|
|
pos = 0;
|
|
}
|
|
|
|
END_UPP_NAMESPACE
|
|
|