Subversion Repositories SmartDukaan

package com.hotspotstore.storage;

import java.net.UnknownHostException;

import java.util.ArrayList;

import java.util.Calendar;

import java.util.List;

import org.json.JSONArray;

import org.json.JSONObject;

import com.google.gson.Gson;

import com.hotspotstore.model.Banner;

import com.hotspotstore.model.Snippet;

import com.mongodb.BasicDBObject;

import com.mongodb.DB;

import com.mongodb.DBCollection;

import com.mongodb.DBCursor;

import com.mongodb.DBObject;

import com.mongodb.MongoClient;

import com.mongodb.util.JSON;

public class Mongo {

	private static MongoClient mongo;

	private static final String HOTSPOT_DB = "Hotspot";

	private static final String BANNERS = "Banners";

	private static final String CONTENT ="CONTENT";

	private static final String SITE_CONTENT = "siteContent";

	private static final String HOTSPOT_ENTITES = "HotspotEntity";

	private static final String SNIPPETS = "Snippets";

	static {

		try {

			mongo = new MongoClient( "localhost" , 27017 );

		} catch (UnknownHostException e) {

			e.printStackTrace();

		}

	}

	public static void addBanner(Banner b){

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(BANNERS);

		BasicDBObject orderBy = new BasicDBObject();

		orderBy.put("_id", -1);

		DBCursor cursor = collection.find().sort(orderBy).limit(1);

		long id = 1l;

		while (cursor.hasNext()) {

			Gson gson = new Gson();

			Banner bannerObj = gson.fromJson(cursor.next().toString(), Banner.class);

			id = bannerObj.get_id() + 1;

		}

		b.set_id(id);

		Gson gs = new Gson();

		DBObject dbObject = (DBObject) JSON.parse(gs.toJson(b));

		collection.insert(dbObject);

	}

	public static Banner getBannerById(long id) throws Exception{

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(BANNERS);

		BasicDBObject obj = new BasicDBObject();

		obj.append("_id", id);

		DBObject result = collection.findOne(obj);

		if (result==null){

			throw new Exception();

		}

		Gson gson = new Gson();

		Banner banner = gson.fromJson(result.toString(), Banner.class);

		return banner;

	}

	public static ArrayList<Banner> getAllBanners(){

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(BANNERS);

		ArrayList<Banner> banners = new ArrayList<Banner>();

		DBCursor result = collection.find();

		Gson gson = new Gson();

		while (result.hasNext()) {

			Banner banner = gson.fromJson(result.next().toString(), Banner.class);

			banners.add(banner);

		}

		return banners;

	}

	public static ArrayList<Banner> getAllActiveBanners(){

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(BANNERS);

		ArrayList<Banner> banners = new ArrayList<Banner>();

		BasicDBObject obj = new BasicDBObject();

		obj.append("active", true);

		BasicDBObject orderBy = new BasicDBObject();

		orderBy.put("priority", 1);

		DBCursor result = collection.find(obj).sort(orderBy);

		Gson gson = new Gson();

		while (result.hasNext()) {

			Banner banner = gson.fromJson(result.next().toString(), Banner.class);

			banners.add(banner);

		}

		return banners;

	}

	public static void deleteBanner(long id){

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(BANNERS);

		BasicDBObject obj = new BasicDBObject();

		obj.append("_id", id);

		collection.remove(obj);

	}

	public static void updateBanner(Banner b){

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(BANNERS);

		Gson gs = new Gson();

		DBObject dbObject = (DBObject) JSON.parse(gs.toJson(b));

		collection.update(new BasicDBObject().append("_id", b.get_id()), dbObject);

	}

	public static JSONObject getEntityById(long id) throws Exception{

		DB db = mongo.getDB(CONTENT);

		DBCollection collection = db.getCollection(SITE_CONTENT);

		BasicDBObject obj = new BasicDBObject();

		obj.append("_id", id);

		DBObject result = collection.findOne(obj);

		if (result==null){

			throw new Exception();

		}

		return new JSONObject(JSON.serialize(result));

	}

	public static void addEntityToHotspot(long entityId, String title){

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(HOTSPOT_ENTITES);

		BasicDBObject obj = new BasicDBObject();

		obj.append("_id", entityId);

		obj.append("title", title);

		collection.insert(obj);

	}

	public static void addSnippetInfo(Snippet s){

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(SNIPPETS);

		Gson gs = new Gson();

		DBObject dbObject = (DBObject) JSON.parse(gs.toJson(s));

		collection.insert(dbObject);

	}

	public static JSONObject getHotspotEntity(Long entityId) throws Exception{

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(HOTSPOT_ENTITES);

		BasicDBObject obj = new BasicDBObject();

		obj.append("_id", entityId);

		DBObject result = collection.findOne(obj);

		if (result==null){

			throw new Exception();

		}

		return new JSONObject(JSON.serialize(result));

	}

	public static List<Snippet> getSnippets(){

		DB db = mongo.getDB(HOTSPOT_DB);

		DBCollection collection = db.getCollection(SNIPPETS);

		ArrayList<Snippet> snippets = new ArrayList<Snippet>();

		DBCursor result = collection.find();

		Gson gson = new Gson();

		while (result.hasNext()) {

			Snippet snippet = gson.fromJson(result.next().toString(), Snippet.class);

			snippets.add(snippet);

		}

		return snippets;

	}

	public static JSONArray getTermsForAutoSuggest(){

		DB db = mongo.getDB(CONTENT);

		DBCollection collection = db.getCollection(SITE_CONTENT);

		BasicDBObject obj = new BasicDBObject();

		obj.append("title", 1);

		JSONArray jsonArray = new JSONArray(); 

		DBCursor result = collection.find(new BasicDBObject(),obj);

		System.out.println(result.getQuery());

		while (result.hasNext()) {

			jsonArray.put(result);

		}

		return jsonArray;

	}

	public static void main(String[] args) throws Exception{

		JSONArray x = getTermsForAutoSuggest();

		System.out.println(x);

	}

}