Subversion Repositories SmartDukaan

package in.shop2020.dtrapi.Storage;

import in.shop2020.dtrapi.models.MasterData;

import in.shop2020.model.v1.order.AmazonFbaSalesSnapshot;

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;

public class Mongo {

	private static MongoClient mongo;

	private static final String CATALOG_DB = "Catalog";

	private static final String MASTER_DATA = "MasterData";

	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 main(String[] args) throws Exception{

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

	}

}