Subversion Repositories SmartDukaan

package in.shop2020.dtrapi.Storage;

import java.lang.reflect.Type;

import java.net.UnknownHostException;

import java.util.List;

import org.apache.log4j.Logger;

import java.util.ArrayList;

import com.google.gson.Gson;

import com.google.gson.reflect.TypeToken;

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;

import in.shop2020.dtrapi.models.FofoForm;

import in.shop2020.dtrapi.models.MasterData;

public class Mongo {

	private static MongoClient mongo;

	private static final String CATALOG_DB = "Catalog";

	private static final String MASTER_DATA = "MasterData";

	private static final String FOFO_DB = "Fofo";

	private static final String FOFO_FORM_COLLECTION = "RegistrationForm";

	private static Logger log = Logger.getLogger(Mongo.class);

	static {

		try {

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

		} catch (UnknownHostException e) {

			e.printStackTrace();

		}

	}

	public static MasterData getItemByID(long id) throws Exception{

		DB db = mongo.getDB(CATALOG_DB);

		DBCollection collection = db.getCollection(MASTER_DATA);

		BasicDBObject obj = new BasicDBObject();

		obj.append("_id", id);

		DBObject result = collection.findOne(obj);

		if (result==null){

			throw new Exception();

		}

		Gson gson = new Gson();

		MasterData masterData = gson.fromJson(result.toString(), MasterData.class);

		return masterData;

	}

	public static ArrayList<MasterData> getItemsByBundleId(long bundleId){

		DB db = mongo.getDB(CATALOG_DB);

		DBCollection collection = db.getCollection(MASTER_DATA);

		BasicDBObject obj = new BasicDBObject();

		BasicDBObject in_query = new BasicDBObject();

		obj.append("skuBundleId", bundleId);

		obj.append("in_stock",1);

		in_query.append("$in",new int[] {1,2,3,4});

		obj.append("source_id", in_query);

		ArrayList<MasterData> items = new ArrayList<MasterData>();

		DBCursor result = collection.find(obj);

		Gson gson = new Gson();

		while (result.hasNext()) {

			MasterData masterData = gson.fromJson(result.next().toString(), MasterData.class);

			items.add(masterData);

		}

		return items;

	}

	public static void persistFofoRegInfo(FofoForm ff){

	    DB db = mongo.getDB(FOFO_DB);

		DBCollection collection = db.getCollection(FOFO_FORM_COLLECTION);

		if(ff.get_id()==0) {

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

				FofoForm existingFofo = gson.fromJson(cursor.next().toString(), FofoForm.class);

				id = existingFofo.get_id() + 1;

			}

			ff.set_id(id);

		}

		Gson gs = new Gson();

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

		collection.save(dbObject);

	}

	public static List<FofoForm> getFofoForms(int offset, int limit) {

		List<FofoForm> ffList = new ArrayList<FofoForm>(); 

		DB db = mongo.getDB(FOFO_DB);

		DBCollection collection = db.getCollection(FOFO_FORM_COLLECTION);

		BasicDBObject orderBy = new BasicDBObject();

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

		DBCursor dbc = collection.find().sort(orderBy).limit(limit).skip(offset);

		while (dbc.hasNext()) {

			ffList.add(convertJSONToPojo(dbc.next().toString()));

		}

		return ffList;

	}

	public static String getFofoFormJsonStringByFofoId(int fofoId){

		DB db = mongo.getDB(FOFO_DB);

		BasicDBObject filter = new BasicDBObject();

		filter.append("_id", fofoId);

		DBCollection collection = db.getCollection(FOFO_FORM_COLLECTION);

		DBObject fofoDbOject = collection.find(filter).one();

		return fofoDbOject.toString();

	}

	public static FofoForm getFofoForm(int fofoId) {

		String fofoFormJsonString = getFofoFormJsonStringByFofoId(fofoId);

		System.out.println(fofoFormJsonString);

		return new Gson().fromJson(fofoFormJsonString, FofoForm.class);

		//return convertJSONToPojo(fofoDbOject.toString());

	}

	private static FofoForm convertJSONToPojo(String json){

	    Type type = new TypeToken< FofoForm >(){}.getType();

	    return new Gson().fromJson(json, type);

	}

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

		System.out.println(getFofoForms(0, 10));

	}

}