private boolean isHistoryUpdateCompleted(final long guestId, final Connector connector) {
   if (connector.isAutonomous()) return isHistoryUpdateCompleted(guestId, connector.getName(), 0);
   final int[] connectorObjectTypeValues = connector.objectTypeValues();
   for (int connectorObjectTypeValue : connectorObjectTypeValues)
     if (!isHistoryUpdateCompleted(guestId, connector.getName(), connectorObjectTypeValue))
       return false;
   return true;
 }
 /**
  * delete pending tasks for a guest's connector
  *
  * @param guestId
  * @param connector
  * @param wipeOutHistory wether to delete everything including the initial history update that we
  *     use to track wether we need to everything from scratch or just do so incrementally
  */
 @Transactional(readOnly = false)
 @Override
 public void flushUpdateWorkerTasks(long guestId, Connector connector, boolean wipeOutHistory) {
   List<AbstractUpdater> toStop = new ArrayList<AbstractUpdater>();
   if (!wipeOutHistory)
     JPAUtils.execute(
         em,
         "updateWorkerTasks.delete.byApi",
         guestId,
         connector.getName(),
         UpdateType.INITIAL_HISTORY_UPDATE);
   else JPAUtils.execute(em, "updateWorkerTasks.deleteAll.byApi", guestId, connector.getName());
 }
 @Override
 public Collection<UpdateWorkerTask> getUpdatingUpdateTasks(
     final long guestId, final Connector connector) {
   List<UpdateWorkerTask> tasks =
       JPAUtils.find(
           em,
           UpdateWorkerTask.class,
           "updateWorkerTasks.isInProgressOrScheduledBefore",
           System.currentTimeMillis(),
           guestId,
           connector.getName());
   HashMap<Integer, UpdateWorkerTask> seen = new HashMap<Integer, UpdateWorkerTask>();
   for (UpdateWorkerTask task : tasks) {
     if (hasStalled(task)) {
       task.status = Status.STALLED;
       em.merge(task);
     } else {
       if (seen.containsKey(task.objectTypes)) {
         if (seen.get(task.objectTypes).timeScheduled < task.timeScheduled)
           seen.put(task.objectTypes, task);
       } else {
         seen.put(task.objectTypes, task);
       }
     }
   }
   return seen.values();
 }
  /**
   * Update all the facet types for a given user and connector.
   *
   * @param guestId the user for whom the connector is to be updated
   * @param connector the connector to be updated
   * @param force force an update (sync now); if false, it means it was called by the background
   *     "cron" task
   * @return
   */
  @Override
  public List<ScheduleResult> updateConnector(
      final long guestId, Connector connector, boolean force) {
    System.out.println("updateConnector");
    List<ScheduleResult> scheduleResults = new ArrayList<ScheduleResult>();
    StringBuilder messageRoot =
        new StringBuilder(
            "module=updateQueue component=connectorUpdateService" + " action=updateConnector");
    if (isShuttingDown) {
      logger.warn(messageRoot.append(" message=\"Service is shutting down... Refusing updates\""));
      return scheduleResults;
    }

    // if forcing an update (sync now), we actually want to flush the update requests
    // that have stacked up in the queue
    if (force) flushUpdateWorkerTasks(guestId, connector, false);

    // some connectors (e.g. the fitbit) need to decide what objectTypes to update by themselves;
    // for those, we pass 0 for the objectType parameter, which will be overridden by the
    // connector's updater
    final boolean historyUpdateCompleted = isHistoryUpdateCompleted(guestId, connector);
    if (connector.isAutonomous()) {
      scheduleObjectTypeUpdate(
          guestId,
          connector,
          0,
          scheduleResults,
          historyUpdateCompleted
              ? UpdateType.INCREMENTAL_UPDATE
              : UpdateType.INITIAL_HISTORY_UPDATE);
    } else {
      int[] objectTypeValues = connector.objectTypeValues();
      for (int objectTypes : objectTypeValues) {
        scheduleObjectTypeUpdate(
            guestId,
            connector,
            objectTypes,
            scheduleResults,
            historyUpdateCompleted
                ? UpdateType.INCREMENTAL_UPDATE
                : UpdateType.INITIAL_HISTORY_UPDATE);
      }
    }
    return scheduleResults;
  }
 @Override
 @Transactional(readOnly = false)
 public void addApiNotification(Connector connector, long guestId, String content) {
   ApiNotification notification = new ApiNotification();
   notification.api = connector.value();
   notification.guestId = guestId;
   notification.ts = System.currentTimeMillis();
   notification.content = content;
   em.persist(notification);
 }
 @Override
 public ApiUpdate getLastSuccessfulUpdate(long guestId, Connector api, int objectTypes) {
   if (objectTypes == -1) return getLastSuccessfulUpdate(guestId, api);
   return JPAUtils.findUnique(
       em,
       ApiUpdate.class,
       "apiUpdates.last.successful.byApiAndObjectTypes",
       guestId,
       api.value(),
       objectTypes);
 }
  @RequestMapping(value = "/upgradeToken")
  public String upgradeToken(HttpServletRequest request, HttpServletResponse response)
      throws OAuthException, OAuthMessageSignerException, OAuthNotAuthorizedException,
          OAuthExpectationFailedException, OAuthCommunicationException {
    OAuthConsumer consumer =
        (OAuthConsumer) request.getSession().getAttribute(FRESHBOOKS_OAUTH_CONSUMER);
    OAuthProvider provider =
        (OAuthProvider) request.getSession().getAttribute(FRESHBOOKS_OAUTH_PROVIDER);
    String verifier = request.getParameter("oauth_verifier");
    provider.retrieveAccessToken(consumer, verifier);
    Guest guest = AuthHelper.getGuest();

    final Connector connector = Connector.getConnector("freshbooks");
    final ApiKey apiKey = guestService.createApiKey(guest.getId(), connector);

    guestService.setApiKeyAttribute(apiKey, "accessToken", consumer.getToken());
    guestService.setApiKeyAttribute(apiKey, "tokenSecret", consumer.getTokenSecret());

    return "redirect:/app/from/" + connector.getName();
  }
 /**
  * Schedules a new update if there is no update for the user for this ObjectType and <code>force
  * </code> is false
  *
  * @param guestId The user's id
  * @param connector The Connector that is to be updated
  * @param objectTypes the integer bitmask of object types to be updated
  * @param scheduleResults The result of adding the update will be added to the list. \result.type
  *     will be of type ScheduleResult.ResultType. If there was a previously existing \result.type
  *     will be ALREADY_SCHEDULED
  */
 private void scheduleObjectTypeUpdate(
     long guestId,
     Connector connector,
     int objectTypes,
     List<ScheduleResult> scheduleResults,
     UpdateType updateType) {
   UpdateWorkerTask updateWorkerTask =
       getScheduledUpdateTask(guestId, connector.getName(), objectTypes);
   if (updateWorkerTask != null)
     scheduleResults.add(
         new ScheduleResult(
             connector.getName(),
             objectTypes,
             ScheduleResult.ResultType.ALREADY_SCHEDULED,
             updateWorkerTask.timeScheduled));
   else {
     final ScheduleResult scheduleResult =
         scheduleUpdate(
             guestId, connector.getName(), objectTypes, updateType, System.currentTimeMillis());
     scheduleResults.add(scheduleResult);
   }
 }
 @Override
 @Transactional(readOnly = false)
 public List<UpdateWorkerTask> getScheduledOrInProgressUpdateTasks(
     long guestId, Connector connector) {
   List<UpdateWorkerTask> updateWorkerTask =
       JPAUtils.find(
           em,
           UpdateWorkerTask.class,
           "updateWorkerTasks.isScheduledOrInProgress",
           guestId,
           connector.getName());
   for (UpdateWorkerTask workerTask : updateWorkerTask) {
     if (hasStalled(workerTask)) {
       workerTask.status = Status.STALLED;
       em.merge(workerTask);
     }
   }
   return updateWorkerTask;
 }
 @Override
 @Transactional(readOnly = false)
 public void addApiUpdate(
     long guestId,
     Connector api,
     int objectTypes,
     long ts,
     long elapsed,
     String query,
     boolean success) {
   ApiUpdate updt = new ApiUpdate();
   updt.guestId = guestId;
   updt.api = api.value();
   updt.ts = System.currentTimeMillis();
   updt.query = query;
   updt.objectTypes = objectTypes;
   updt.elapsed = elapsed;
   updt.success = success;
   em.persist(updt);
 }
 @Override
 public List<ScheduleResult> updateConnectorObjectType(
     final long guestId, final Connector connector, int objectTypes, boolean force) {
   List<ScheduleResult> scheduleResults = new ArrayList<ScheduleResult>();
   if (isShuttingDown) {
     StringBuilder sb =
         new StringBuilder(
                 "module=updateQueue component=connectorUpdateService"
                     + " action=updateConnectorObjectType")
             .append(" message=\"Service is shutting down... Refusing updates\"");
     logger.warn(sb.toString());
     return scheduleResults;
   }
   getScheduledUpdateTask(guestId, connector.getName(), objectTypes);
   // if forcing an update (sync now), we actually want to flush the update requests
   // that have stacked up in the queue
   if (force) flushUpdateWorkerTasks(guestId, connector, objectTypes, false);
   scheduleObjectTypeUpdate(
       guestId, connector, objectTypes, scheduleResults, UpdateType.INCREMENTAL_UPDATE);
   return scheduleResults;
 }
 @Override
 public Collection<UpdateWorkerTask> getLastFinishedUpdateTasks(
     final long guestId, final Connector connector) {
   List<UpdateWorkerTask> tasks =
       JPAUtils.find(
           em,
           UpdateWorkerTask.class,
           "updateWorkerTasks.getLastFinishedTask",
           System.currentTimeMillis(),
           guestId,
           connector.getName());
   HashMap<Integer, UpdateWorkerTask> seen = new HashMap<Integer, UpdateWorkerTask>();
   for (UpdateWorkerTask task : tasks) {
     if (seen.containsKey(task.objectTypes)) {
       if (seen.get(task.objectTypes).timeScheduled < task.timeScheduled)
         seen.put(task.objectTypes, task);
     } else {
       seen.put(task.objectTypes, task);
     }
   }
   return seen.values();
 }
 public boolean checkAuthorization(GuestService guestService, long guestId) {
   ApiKey apiKey = guestService.getApiKey(guestId, Connector.getConnector("toodledo"));
   return apiKey != null;
 }
 @Override
 public long getNumberOfUpdatesSince(long guestId, Connector connector, long then) {
   return JPAUtils.count(em, "apiUpdates.count.byGuest.since", guestId, connector.value(), then);
 }
 @Override
 public long getTotalNumberOfUpdatesSince(Connector connector, long then) {
   return JPAUtils.count(em, "apiUpdates.count.all.since", connector.value(), then);
 }
 @Override
 public long getNumberOfUpdates(long guestId, Connector connector) {
   return JPAUtils.count(em, "apiUpdates.count.byGuest", guestId, connector.value());
 }
 @Override
 public long getTotalNumberOfUpdates(Connector connector) {
   return JPAUtils.count(em, "apiUpdates.count.all", connector.value());
 }
 @Override
 public long getTotalNumberOfGuestsUsingConnector(Connector connector) {
   return JPAUtils.count(em, "apiKey.count.byApi", connector.value());
 }
Example #19
0
 public static ConnectorModel connectorModel(Connector api) {
   ConnectorModel vo = new ConnectorModel();
   vo.connectorName = api.getName();
   vo.prettyName = api.prettyName();
   return vo;
 }
 @Override
 public List<ApiUpdate> getUpdates(
     long guestId, final Connector connector, final int pageSize, final int page) {
   return JPAUtils.findPaged(
       em, ApiUpdate.class, "apiUpdates.last.paged", pageSize, page, guestId, connector.value());
 }
 @Override
 public ApiUpdate getLastSuccessfulUpdate(long guestId, Connector api) {
   return JPAUtils.findUnique(
       em, ApiUpdate.class, "apiUpdates.last.successful.byApi", guestId, api.value());
 }
 @Override
 public AbstractUpdater getUpdater(Connector connector) {
   return beanFactory.getBean(connector.getUpdaterClass());
 }