@@ -170,40 +170,32 @@ void DbStructureModel::reloadData()
170170 typeToParentItem.insert (" trigger" , itemTriggers);
171171
172172 // Get all database objects and sort them by their name
173- QMultiMap<QString, DBBrowserObject > dbobjs;
173+ QMultiMap<QString, sqlb::ObjectPtr > dbobjs;
174174 for (auto it=m_db.objMap .constBegin (); it != m_db.objMap .constEnd (); ++it)
175- dbobjs.insert ((*it). getname (), (*it));
175+ dbobjs.insert ((*it)-> name (), (*it));
176176
177177 // Add the actual table objects
178178 for (auto it=dbobjs.constBegin ();it!=dbobjs.constEnd ();++it)
179179 {
180180 // Object node
181- QString type;
182- switch ((*it).gettype ())
183- {
184- case sqlb::Object::Types::Table: type = " table" ; break ;
185- case sqlb::Object::Types::Index: type = " index" ; break ;
186- case sqlb::Object::Types::Trigger: type = " trigger" ; break ;
187- case sqlb::Object::Types::View: type = " view" ; break ;
188- }
189- QTreeWidgetItem* item = addNode (typeToParentItem.value (type), *it);
181+ QTreeWidgetItem* item = addNode (typeToParentItem.value (sqlb::Object::typeToString ((*it)->type ())), *it);
190182
191183 // If it is a table or view add the field nodes
192- if ((*it). gettype () == sqlb::Object::Types::Table || (*it). gettype () == sqlb::Object::Types::View)
184+ if ((*it)-> type () == sqlb::Object::Types::Table || (*it)-> type () == sqlb::Object::Types::View)
193185 {
194186 // Add extra node for browsable section
195187 addNode (typeToParentItem.value (" browsable" ), *it);
196188
197189 // Add field nodes
198190 QStringList pk_columns;
199- if (it-> gettype () == sqlb::Object::Types::Table)
191+ if ((*it)-> type () == sqlb::Object::Types::Table)
200192 {
201- sqlb::FieldVector pk = it-> object .dynamicCast <sqlb::Table>()->primaryKey ();
193+ sqlb::FieldVector pk = (*it) .dynamicCast <sqlb::Table>()->primaryKey ();
202194 foreach (sqlb::FieldPtr pk_col, pk)
203195 pk_columns.push_back (pk_col->name ());
204196
205197 }
206- sqlb::FieldInfoList fieldList = it-> object ->fieldInformation ();
198+ sqlb::FieldInfoList fieldList = (*it) ->fieldInformation ();
207199 foreach (const sqlb::FieldInfo& field, fieldList)
208200 {
209201 QTreeWidgetItem *fldItem = new QTreeWidgetItem (item);
@@ -293,22 +285,15 @@ bool DbStructureModel::dropMimeData(const QMimeData* data, Qt::DropAction action
293285 }
294286}
295287
296- QTreeWidgetItem* DbStructureModel::addNode (QTreeWidgetItem* parent, const DBBrowserObject & object)
288+ QTreeWidgetItem* DbStructureModel::addNode (QTreeWidgetItem* parent, const sqlb::ObjectPtr & object)
297289{
298- QString type;
299- switch (object.gettype ())
300- {
301- case sqlb::Object::Types::Table: type = " table" ; break ;
302- case sqlb::Object::Types::Index: type = " index" ; break ;
303- case sqlb::Object::Types::Trigger: type = " trigger" ; break ;
304- case sqlb::Object::Types::View: type = " view" ; break ;
305- }
290+ QString type = sqlb::Object::typeToString (object->type ());
306291
307292 QTreeWidgetItem *item = new QTreeWidgetItem (parent);
308293 item->setIcon (0 , QIcon (QString (" :/icons/%1" ).arg (type)));
309- item->setText (0 , object. getname ());
294+ item->setText (0 , object-> name ());
310295 item->setText (1 , type);
311- item->setText (3 , object. getsql ());
296+ item->setText (3 , object-> originalSql ());
312297
313298 return item;
314299}
0 commit comments