@GET
  @Path("/verification")
  @Produces({MediaType.APPLICATION_JSON})
  public Response getVerificationServices() {
    MongoDBSingleton dbSingleton = MongoDBSingleton.getInstance();
    DB db = dbSingleton.getTestdb();
    Gson gson = new Gson();

    DBCollection coll = db.getCollection("verificationCode");
    DBCursor cursor = coll.find();
    List<String> list = new ArrayList<String>();
    while (cursor.hasNext()) {
      DBObject o = cursor.next();
      String verifiactionCode = o.get("code") + "";
      list.add(verifiactionCode);
    }
    return Response.ok(gson.toJson(list)).build();
  }
Пример #2
0
public class TriggerDao {
  private MongoDBSingleton dbSingleton = MongoDBSingleton.getInstance();
  private DB db = dbSingleton.getTestdb();
  private Gson gson = new Gson();

  public List<Trigger> getAllTriggers() {
    DBCollection coll = db.getCollection("trigger");
    DBCursor cursor = coll.find();
    List<Trigger> list = new ArrayList<Trigger>();
    while (cursor.hasNext()) {
      DBObject o = cursor.next();
      Trigger drug = gson.fromJson(o.toString(), Trigger.class);

      list.add(drug);
    }
    return list;
  }

  public boolean addTrigger(Trigger drug) {

    DBCollection collection = db.getCollection("trigger");
    // convert JSON to DBObject directly
    BasicDBObject bdbo = new BasicDBObject();
    bdbo.put("name", drug.getName());
    DBCursor curs = collection.find(bdbo);
    if (curs.count() > 0) return false;
    Genson genson = new Genson();
    DBObject dbObject = (DBObject) JSON.parse(genson.serialize(drug));
    collection.insert(dbObject);

    DBCursor cursorDoc = collection.find();
    while (cursorDoc.hasNext()) {
      System.out.println(cursorDoc.next());
    }

    System.out.println("Done");
    return true;
  }

  public int getNewTriggerID() {
    DBCollection coll = db.getCollection("trigger");
    BasicDBObject whereQuery = new BasicDBObject();
    DBCursor cursor = coll.find(whereQuery);
    int max = 0;
    while (cursor.hasNext()) {
      BasicDBObject next = (BasicDBObject) cursor.next();
      if (max < Double.parseDouble("" + next.get("triggerID"))) {
        max = (int) Double.parseDouble("" + next.get("triggerID"));
      }
    }
    return max + 1;
  }

  public int getTriggerID(Trigger trigger) {
    DBCollection coll = db.getCollection("trigger");
    BasicDBObject whereQuery = new BasicDBObject();
    whereQuery.put("name", trigger.getName());
    DBCursor cursor = coll.find(whereQuery);
    int id = -1;
    while (cursor.hasNext()) {
      BasicDBObject next = (BasicDBObject) cursor.next();
      id = (int) Double.parseDouble("" + next.get("triggerID"));
    }
    return id;
  }

  public boolean changeTrigger(Trigger trigger) {

    DBCollection collection = db.getCollection("trigger");
    // convert JSON to DBObject directly
    BasicDBObject bdbo = new BasicDBObject();
    bdbo.put("triggerID", trigger.getTriggerID());
    DBCursor curs = collection.find(bdbo);
    if (curs.count() > 1) return false;
    Gson genson = new Gson();
    DBObject dbObject = (DBObject) JSON.parse(genson.toJson(trigger));
    collection.update(bdbo, dbObject);

    DBCursor cursorDoc = collection.find();
    while (cursorDoc.hasNext()) {
      System.out.println(cursorDoc.next());
    }

    System.out.println("Done");
    return true;
  }

  public Trigger getTrigger(String name) {
    DBCollection coll = db.getCollection("trigger");
    BasicDBObject whereQuery = new BasicDBObject();
    whereQuery.put("name", name);
    DBCursor cursor = coll.find(whereQuery);
    Trigger trigger = null;
    while (cursor.hasNext()) {
      DBObject o = cursor.next();
      trigger = gson.fromJson(o.toString(), Trigger.class);
    }
    return trigger;
  }

  public boolean deleteTrigger(Trigger trigger) {
    DBCollection collection = db.getCollection("trigger");
    // convert JSON to DBObject directly
    BasicDBObject bdbo = new BasicDBObject();
    bdbo.put("triggerID", trigger.getTriggerID());
    DBCursor curs = collection.find(bdbo);
    if (curs.count() > 1) {
      System.err.println("Multiple triggers in the database have the same id");
      return false;
    }

    BasicDBObject document = new BasicDBObject();
    document.put("triggerID", trigger.getTriggerID());
    collection.remove(document);
    System.out.println("Done");
    return true;
  }
}