Subversion Repositories SmartDukaan

package in.shop2020.util;


import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import in.shop2020.metamodel.core.Entity;
import in.shop2020.metamodel.core.EntityState;
import in.shop2020.metamodel.core.FreeformContent;
import in.shop2020.metamodel.core.Media;
import in.shop2020.metamodel.core.Slide;
import in.shop2020.metamodel.definitions.OldEntityContainer;
import in.shop2020.metamodel.util.CreationUtils;
import in.shop2020.metamodel.util.ExpandedBullet;
import in.shop2020.metamodel.util.OldCreationUtils;
import in.shop2020.storage.bdb.StorageManager;


/*
 * @author rajveer 
 * 
 *
 */

public class ContentMigratorToBDB {
        private String sourceDbPath;
        private String destinationDbPath;
        
        public static void main(String[] args) throws Exception {
                String usage = "Usage: ContentMigrator {source db path}{destination db path}";
                
                String sourceDbPath = null, destinationDbPath = null;
                if(args.length < 2) {
                        System.out.println(usage);
                        System.exit(-1);
                }else{
                        sourceDbPath = args[0];
                        destinationDbPath = args[1];
                }
                
                StorageManager.getStorageManager().close();
                
                ContentMigratorToBDB contentmigrator = new ContentMigratorToBDB(sourceDbPath, destinationDbPath);
                //boolean result = contentmigrator.addMedia();
                //boolean result = contentmigrator.addLearnedFields();
                //boolean result = contentmigrator.migrate();
                //System.out.println("Content migration status:  "+ result);
        }

        public ContentMigratorToBDB(String sourceDbPath, String destinationDbPath) {
                this.sourceDbPath = sourceDbPath;
                this.destinationDbPath = destinationDbPath;
        }
        
        public boolean addLearnedFields() throws Exception{
                OldEntityContainer oldEnts = new OldEntityContainer(sourceDbPath);
                Map<Long, List<ExpandedBullet>> learnedBullets = oldEnts.getLearnedBullets();
                CreationUtils.storeLearnedBullets(learnedBullets);
                return true;
                
        }
        
        public boolean addMedia() throws NumberFormatException, IOException, Exception{
            FileInputStream fstream = new FileInputStream("/home/rajveer/Desktop/test1");
            DataInputStream in = new DataInputStream(fstream);
            BufferedReader br = new BufferedReader(new InputStreamReader(in));
            String strLine;
            
            while ((strLine = br.readLine()) != null)   {
                String[] parts  = strLine.split(":");
                long entityID = Long.parseLong(parts[0]);
                long slideID = Long.parseLong(parts[1]);
                Entity entity = StorageManager.getStorageManager().getEntity(entityID);
                Map<String, Media> medias =  OldCreationUtils.getRawMedia(entityID);
                String imageLabels = parts[2];
                String videoLabels = parts[3];
                if(!imageLabels.equalsIgnoreCase("null")){
                        String[] labels = imageLabels.split(",");
                        for(int i = 0; i < labels.length; i++){
                                if(medias.containsKey(labels[i])){
                                        Media media = medias.get(labels[i]);
                                        entity.getSlide(slideID).getFreeformContent().addMedia(media);
                                }
                        }
                }
                if(!videoLabels.equalsIgnoreCase("null")){
                        String[] labels = videoLabels.split(",");
                        for(int i = 0; i < labels.length; i++){
                                if(medias.containsKey(labels[i])){
                                        Media media = medias.get(labels[i]);
                                        entity.getSlide(slideID).getFreeformContent().addMedia(media);
                                }
                        }
                }
                StorageManager.getStorageManager().updateEntity(entity);
            }
            in.close();
            return true;
}

        /**
         * this function will read source definition and source entities, and will convert
         * source entities to destination entities according to destination definitions.  
         * 
         * @return boolean
         * @throws Exception 
         */
        public boolean migrate() throws Exception{
                OldEntityContainer sourceEnts = new OldEntityContainer(sourceDbPath);
                
                StringBuilder sb = new StringBuilder();
                Map<Long, Entity> entities = sourceEnts.getEntities();
                for(Long entityId: entities.keySet()){
                        Entity entity = OldCreationUtils.getEntity(entityId);
                        if(entity.getID() == 0){
                                continue;
                        }
                        Entity newEntity = new Entity(entity.getID(), entity.getCategoryID());
                        newEntity.setBrand(entity.getBrand());
                        newEntity.setModelName(entity.getModelName());
                        newEntity.setModelNumber(entity.getModelNumber());
                        
                        newEntity.setSlideSequence(OldCreationUtils.getSlideSequence(entity.getID()));
                        
                        List<Slide> slides = entity.getSlides();
                        if(slides!=null){
                        for(Slide slide: slides){
                                FreeformContent ffc = slide.getFreeformContent();
                                //sb.append(entityId+ ":" + slide.getSlideDefinitionID() + ":"+ ffc.getFreeformTexts()+"\n");
                                FreeformContent newFfc = new FreeformContent();
                                newFfc.setFreeformTexts(ffc.getFreeformTexts());
                                slide.setFreeformContent(newFfc);
                                //if(ffc.getAllLabels()!=null && !ffc.getAllLabels().isEmpty()){
                                        //sb.append(entity.getID() + ":" + slide.getSlideDefinitionID() + ":" + ffc.getAllLabels() + "\n");
                                        //sb.append(entity.getID() + ":" + slide.getSlideDefinitionID() + ":" + ffc.getImageLabels() + ":" + ffc.getYoutubeLabels() + "\n");
                                        //System.out.println(entity.getID() + ":" + slide.getSlideDefinitionID() + ":" + ffc.getImageLabels() + ":" + ffc.getYoutubeLabels());
//                                      System.out.println("some image found");
//                                      System.out.println(ffc.getImageLabels());
//                                      System.out.println(ffc.getYoutubeLabels());
                                //}
                                //System.out.println(ffc.getImageLabels());
                                //System.out.println(ffc.getYoutubeLabels());
                                //FreeformContent newFfc = new FreeformContent();
                                //newFfc.setFreeformTexts(ffc.getFreeformTexts());
                        }
                        }
                        newEntity.setSlides(slides);
                        EntityState entityMetadata = new EntityState(entityId, "admin");
                        StorageManager.getStorageManager().createEntity(newEntity, entityMetadata);
                        
                        //System.out.println(StorageManager.getStorageManager().getStorageManager().getEntity(entityId).getSlide(130033).getFreeformContent().getFreeformTexts());
                        
                        //StorageManager.getStorageManager().close();
                        //System.out.println("Entity id is" + entity.getID());
                }
                
                StorageManager.getStorageManager().close();
                //System.out.println(sb.toString());
                return true;
        }
}