Exemple #1
0
std::pair<std::shared_ptr<te::da::DataSetType>, std::shared_ptr<te::da::DataSet> >
terrama2::core::getDCPPositionsTable(std::shared_ptr<te::da::DataSource> datasource, const std::string& dataSetName)
{
  terrama2::core::OpenClose< std::shared_ptr<te::da::DataSource> > openClose(datasource); Q_UNUSED(openClose);

  std::shared_ptr<te::da::DataSourceTransactor> transactorDestination(datasource->getTransactor());
  te::da::ScopedTransaction scopedTransaction(*transactorDestination);

  std::shared_ptr<te::da::DataSet> teDataset = transactorDestination->getDataSet(dataSetName);
  std::shared_ptr<te::da::DataSetType> teDataSetType = transactorDestination->getDataSetType(dataSetName);

  return {teDataSetType, teDataset};
}
void terrama2::core::DataStoragerPostGis::store(DataSetSeries series, DataSetPtr outputDataSet) const
{
  QUrl url(dataProvider_->uri.c_str());

  std::shared_ptr<te::da::DataSource> datasourceDestination(te::da::DataSourceFactory::make("POSTGIS"));
  std::map<std::string, std::string> connInfo{{"PG_HOST", url.host().toStdString()},
                                              {"PG_PORT", std::to_string(url.port())},
                                              {"PG_USER", url.userName().toStdString()},
                                              {"PG_PASSWORD", url.password().toStdString()},
                                              {"PG_DB_NAME", url.path().section("/", 1, 1).toStdString()},
                                              {"PG_CONNECT_TIMEOUT", "4"},
                                              {"PG_CLIENT_ENCODING", "UTF-8"}};
  datasourceDestination->setConnectionInfo(connInfo);

  OpenClose< std::shared_ptr<te::da::DataSource> > openClose(datasourceDestination); Q_UNUSED(openClose);
  if(!datasourceDestination->isOpened())
  {
    QString errMsg = QObject::tr("Could not connect to database");
    TERRAMA2_LOG_ERROR() << errMsg;
    throw DataProviderException() << ErrorDescription(errMsg);
  }

  std::string destinationDataSetName = getDataSetTableName(outputDataSet);

  std::shared_ptr<te::da::DataSourceTransactor> transactorDestination(datasourceDestination->getTransactor());
  te::da::ScopedTransaction scopedTransaction(*transactorDestination);

  std::shared_ptr<te::da::DataSetType> datasetType = series.teDataSetType;


  std::map<std::string, std::string> options;
  std::shared_ptr<te::da::DataSetType> newDataSetType;
  if (!transactorDestination->dataSetExists(destinationDataSetName))
  {
    // create and save datasettype in the datasource destination
    newDataSetType = std::shared_ptr<te::da::DataSetType>(static_cast<te::da::DataSetType*>(datasetType->clone()));
    if(!newDataSetType->getPrimaryKey())
    {
      std::string pkName = "\""+newDataSetType->getName()+"_pk\"";
      auto pk = new te::da::PrimaryKey(pkName, newDataSetType.get());

      te::dt::SimpleProperty* serialPk = new te::dt::SimpleProperty("pid", te::dt::INT32_TYPE, true);
      serialPk->setAutoNumber(true);
      newDataSetType->add(serialPk);
      pk->add(serialPk);
    }

    newDataSetType->setName(destinationDataSetName);
    transactorDestination->createDataSet(newDataSetType.get(),options);

    //Get original geometry to get srid
    te::gm::GeometryProperty* geom = GetFirstGeomProperty(datasetType.get());
    //configure if there is a geometry property
    if(geom)
    {
      GetFirstGeomProperty(newDataSetType.get())->setSRID(geom->getSRID());
      GetFirstGeomProperty(newDataSetType.get())->setGeometryType(te::gm::GeometryType);
    }

  }
  else
  {
    newDataSetType = transactorDestination->getDataSetType(destinationDataSetName);
  }

  const auto& oldPropertiesList = newDataSetType->getProperties();
  for(const auto & property : datasetType->getProperties())
  {
    auto it = std::find_if(oldPropertiesList.cbegin(), oldPropertiesList.cend(), [property]
                                                                                 (te::dt::Property* oldMember)
                                                                                 {
                                                                                   return property->getName() == oldMember->getName()
                                                                                          && property->getType() == oldMember->getType();
                                                                                 });
    if(it == oldPropertiesList.cend())
      transactorDestination->addProperty(newDataSetType->getName(), property);
  }

  series.syncDataSet->dataset()->moveBeforeFirst();
  transactorDestination->add(newDataSetType->getName(), series.syncDataSet->dataset().get(), options);

  scopedTransaction.commit();
}