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