@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();
}
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;
}
}
