Exemplo n.º 1
0
  /**
   * Insert transportation routes
   *
   * @param transportationRouteList List of transportation routes to insert (id == null)
   * @throws Exception
   */
  public void insertTransportationRoutes(Collection<TransportationRoute> transportationRouteList)
      throws Exception {
    try {
      String query =
          String.format(
              "INSERT INTO %s (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
              table,
              id,
              name,
              validFrom,
              validUntil,
              transportationID,
              operator,
              network,
              extRef,
              description,
              descriptionFrom,
              descriptionTo,
              routeNo);
      PreparedStatement ps = ConnectionManager.getInstance().prepareStatement(query);

      for (TransportationRoute transportationRoute : transportationRouteList) {
        if (transportationRoute.getId() != null || transportationRoute.getType() == null) {
          continue;
        }

        transportationRoute.setId(UUID.randomUUID());
        PGobject toInsertUUID = new PGobject();
        toInsertUUID.setType("uuid");
        toInsertUUID.setValue(String.valueOf(transportationRoute.getId()));

        ps.setObject(1, toInsertUUID);
        ps.setString(2, transportationRoute.getName());
        ps.setTimestamp(3, transportationRoute.getValidFrom());
        ps.setTimestamp(4, transportationRoute.getValidUntil());
        ps.setObject(5, transportationRoute.getType().getId());
        ps.setString(6, transportationRoute.getOperator());
        ps.setString(7, transportationRoute.getNetwork());
        ps.setString(8, transportationRoute.getExtRef());
        ps.setString(9, transportationRoute.getDescription());
        ps.setString(10, transportationRoute.getDescriptionFrom());
        ps.setString(11, transportationRoute.getDescriptionTo());
        ps.setString(12, transportationRoute.getRouteNo());

        ps.addBatch();
      }

      ps.executeBatch();
      ps.close();

      ScheduleDAO scheduleDAO = new ScheduleDAO();
      scheduleDAO.insertOrUpdateSchedules(transportationRouteList);
    } catch (Exception e) {
      ConnectionManager.getInstance().closeConnection(true);

      throw new Exception("Transportation route insert failed: " + e.getMessage());
    }
  }
Exemplo n.º 2
0
  /**
   * Get all transportation routes
   *
   * @return List
   */
  public List<TransportationRoute> getAllTransportationRoutes() throws Exception {
    // Get all transportation types
    TransportationTypeDAO transportationTypeDAO = new TransportationTypeDAO();
    HashMap<String, TransportationType> transportationTypes =
        ListConverter.getEntityListAsHashMap(transportationTypeDAO.getAllTransportationTypes());

    // Fetch transportation routes
    ArrayList<TransportationRoute> transportationRoutes = new ArrayList<>();
    try {
      String query = String.format("Select * from %s", table);
      ResultSet result = ConnectionManager.getInstance().createStatement().executeQuery(query);

      while (result.next()) {
        TransportationRoute transportationRoute = new TransportationRoute();

        transportationRoute.setId(UUID.fromString(result.getString(id)));
        transportationRoute.setName(result.getString(name));
        transportationRoute.setValidFrom(result.getTimestamp(validFrom));
        transportationRoute.setValidUntil(result.getTimestamp(validUntil));
        transportationRoute.setType(transportationTypes.get(result.getString(transportationID)));
        transportationRoute.setOperator(result.getString(operator));
        transportationRoute.setNetwork(result.getString(network));
        transportationRoute.setExtRef(result.getString(extRef));
        transportationRoute.setDescriptionFrom(result.getString(descriptionFrom));
        transportationRoute.setDescriptionTo(result.getString(descriptionTo));
        transportationRoute.setDescription(result.getString(description));
        transportationRoute.setRouteNo(result.getString(routeNo));

        transportationRoutes.add(transportationRoute);
      }

      ScheduleDAO scheduleDAO = new ScheduleDAO();
      for (TransportationRoute transportationRoute : transportationRoutes) {
        transportationRoute.setSchedules(scheduleDAO.getValidSchedules(transportationRoute));
      }
    } catch (Exception e) {
      ConnectionManager.getInstance().closeConnection(true);

      throw new Exception("Transportation route fetch failed: " + e.getMessage());
    }

    return transportationRoutes;
  }