mirror of
https://github.com/gnosygnu/xowa.git
synced 2024-10-27 20:34:16 +00:00
60 lines
2.6 KiB
Java
60 lines
2.6 KiB
Java
/*
|
|
XOWA: the XOWA Offline Wiki Application
|
|
Copyright (C) 2012 gnosygnu@gmail.com
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as
|
|
published by the Free Software Foundation, either version 3 of the
|
|
License, or (at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package gplx.dbs; import gplx.*;
|
|
import gplx.lists.*; /*GfoNde*/
|
|
class TdbInsertWkr implements Db_qryWkr {
|
|
public Object Exec(Db_engine engineObj, Db_qry cmdObj) {
|
|
TdbEngine engine = TdbEngine.cast_(engineObj); Db_qry_insert cmd = (Db_qry_insert)cmdObj;
|
|
|
|
TdbTable tbl = engine.FetchTbl(cmd.BaseTable());
|
|
tbl.IsDirty_set(true);
|
|
return cmd.Select() == null
|
|
? InsertRowsByVals(engine, tbl, cmd)
|
|
: InsertRowsBySelect(engine, tbl, cmd);
|
|
}
|
|
int InsertRowsBySelect(TdbEngine engine, TdbTable tbl, Db_qry_insert insert) {
|
|
int count = 0;
|
|
DataRdr rdr = (DataRdr)TdbSelectWkr._.Exec(engine, insert.Select());
|
|
Sql_select_fld_list insertFlds = insert.Cols(); int insertFldsCount = insertFlds.Count();
|
|
GfoFldList selectFldsForNewRow = null;
|
|
try {selectFldsForNewRow = insertFlds.XtoGfoFldLst(tbl);}
|
|
catch (Exception e) {throw Err_.err_(e, "failed to generate flds for new row");}
|
|
if (insertFldsCount > selectFldsForNewRow.Count()) throw Err_.new_("insert flds cannot exceed selectFlds").Add("insertFlds", insertFlds.XtoStr()).Add("selectFlds", selectFldsForNewRow.XtoStr());
|
|
while (rdr.MoveNextPeer()) {
|
|
count++;
|
|
GfoNde row = GfoNde_.vals_(selectFldsForNewRow, new Object[insertFldsCount]);
|
|
for (int i = 0; i < insertFldsCount; i++) {
|
|
row.WriteAt(i, rdr.ReadAt(i)); // NOTE: SELECT and INSERT flds are in same order; ex: INSERT INTO (a, b) SELECT (b, a)
|
|
}
|
|
tbl.Rows().Add(row);
|
|
}
|
|
return count;
|
|
}
|
|
int InsertRowsByVals(TdbEngine engine, TdbTable tbl, Db_qry_insert insert) {
|
|
GfoNde row = GfoNde_.vals_(tbl.Flds(), new Object[tbl.Flds().Count()]);
|
|
for (int i = 0; i < insert.Args().Count(); i++) {
|
|
KeyVal kv = insert.Args().FetchAt(i);
|
|
Db_arg arg = (Db_arg)kv.Val();
|
|
row.Write(kv.Key(), arg.Val());
|
|
}
|
|
tbl.Rows().Add(row);
|
|
return 1;
|
|
}
|
|
public static TdbInsertWkr new_() {TdbInsertWkr rv = new TdbInsertWkr(); return rv;}
|
|
}
|