Esempio n. 1
0
 @Override
 public Alert createAlert(String checkId, Alert alert) {
   alert.setId(ObjectId.get().toString());
   alert.setCheckId(checkId);
   getAlertsCollection().insert(mapper.alertToDBObject(alert));
   return alert;
 }
Esempio n. 2
0
 @Override
 public Check getCheck(String checkId) {
   DBObject dbo = getChecksCollection().findOne(object("_id", checkId));
   if (dbo == null) {
     return null;
   }
   return mapper.checkFrom(dbo);
 }
Esempio n. 3
0
 @Override
 public void updateSubscription(String checkId, Subscription subscription) {
   DBObject subscriptionObject = mapper.subscriptionToDBObject(subscription);
   DBObject subscriptionFindObject = forId(subscription.getId());
   DBObject checkFindObject =
       forId(checkId).with("subscriptions", object("$elemMatch", subscriptionFindObject));
   DBObject updateObject = object("$set", object("subscriptions.$", subscriptionObject));
   getChecksCollection().update(checkFindObject, updateObject);
 }
Esempio n. 4
0
 @Override
 public Subscription createSubscription(String checkId, Subscription subscription) {
   subscription.setId(ObjectId.get().toString());
   DBObject check = forId(checkId);
   DBObject query =
       object("$push", object("subscriptions", mapper.subscriptionToDBObject(subscription)));
   getChecksCollection().update(check, query);
   return subscription;
 }
Esempio n. 5
0
 @Override
 public Alert getLastAlertForTargetOfCheck(String target, String checkId) {
   DBObject query = object("checkId", checkId).with("target", target);
   DBCursor cursor = getAlertsCollection().find(query).sort(object("timestamp", -1)).limit(1);
   try {
     while (cursor.hasNext()) {
       return mapper.alertFrom(cursor.next());
     }
   } finally {
     cursor.close();
   }
   return null;
 }
Esempio n. 6
0
 @Override
 public SeyrenResponse<Check> getChecks(Boolean enabled) {
   List<Check> checks = new ArrayList<Check>();
   DBCursor dbc;
   if (enabled != null) {
     dbc = getChecksCollection().find(object("enabled", enabled));
   } else {
     dbc = getChecksCollection().find();
   }
   while (dbc.hasNext()) {
     checks.add(mapper.checkFrom(dbc.next()));
   }
   return new SeyrenResponse<Check>().withValues(checks).withTotal(dbc.count());
 }
Esempio n. 7
0
 @Override
 public SeyrenResponse<Alert> getAlerts(int start, int items) {
   DBCursor dbc =
       getAlertsCollection().find().sort(object("timestamp", -1)).skip(start).limit(items);
   List<Alert> alerts = new ArrayList<Alert>();
   while (dbc.hasNext()) {
     alerts.add(mapper.alertFrom(dbc.next()));
   }
   dbc.close();
   return new SeyrenResponse<Alert>()
       .withValues(alerts)
       .withItems(items)
       .withStart(start)
       .withTotal(dbc.count());
 }
Esempio n. 8
0
  @Override
  public SeyrenResponse<Check> getChecksByState(Set<String> states, Boolean enabled) {
    List<Check> checks = new ArrayList<Check>();

    DBObject query = new BasicDBObject();
    query.put("state", object("$in", states.toArray()));
    if (enabled != null) {
      query.put("enabled", enabled);
    }
    DBCursor dbc = getChecksCollection().find(query);

    while (dbc.hasNext()) {
      checks.add(mapper.checkFrom(dbc.next()));
    }
    dbc.close();

    return new SeyrenResponse<Check>().withValues(checks).withTotal(dbc.count());
  }
Esempio n. 9
0
 @Override
 public Check createCheck(Check check) {
   check.setId(ObjectId.get().toString());
   getChecksCollection().insert(mapper.checkToDBObject(check));
   return check;
 }