Subversion Repositories SmartDukaan

package in.shop2020.dtrapi.Storage;

import in.shop2020.dtrapi.models.FofoForm;

import in.shop2020.dtrapi.models.MasterData;

import java.awt.List;

import java.net.UnknownHostException;

import java.util.ArrayList;

import com.google.gson.Gson;

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 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";

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

		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.insert(dbObject);

	}

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

		System.out.println(getItemsByBundleId(100000));

	}

}