Subversion Repositories SmartDukaan

package com.spice.profitmandi.dao.util;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.lang.reflect.Type;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import com.google.gson.Gson;

import com.google.gson.reflect.TypeToken;

import com.spice.profitmandi.common.enumuration.ContentType;

import com.spice.profitmandi.common.exception.ProfitMandiBusinessException;

import com.spice.profitmandi.common.util.FileUtil;

import com.spice.profitmandi.dao.entity.dtr.Document;

import com.spice.profitmandi.dao.repository.dtr.DocumentRepository;

import com.spice.profitmandi.dao.repository.dtr.Mongo;

@Component

public class MongoMigrationUtil {

	@Autowired

	DocumentRepository documentRepository;

	@Autowired

	Mongo mongoClient;

	private static final Logger LOGGER = LoggerFactory.getLogger(MongoMigrationUtil.class);

	public void migrate() throws ProfitMandiBusinessException{

		String fofoFormsJsonString = mongoClient.getFofoFormsJsonString();

		Map<Integer, Map<String, Integer>> fofoIdPathMap = new HashMap<>();

		Gson gson = new Gson();

        Type paths = new TypeToken<List<Map<String, String>>>(){}.getType();

        List<Map<String, String>> maps = gson.fromJson(fofoFormsJsonString, paths);

        System.out.println(maps);

        for(Map<String, String> map : maps){

        	int id = 0;

        	Map<String, Integer> pathMap = new HashMap<>();

        	for(Map.Entry<String, String> entry : map.entrySet()){

        		if(entry.getKey().equals("_id")){

        			id = Integer.parseInt(entry.getValue());

        		}

        		if(entry.getValue().startsWith("/hsps-docs/")){

        			Document document = new Document();

        			try{

        				File file = new File(entry.getValue());

        				FileInputStream fileInputStream = new FileInputStream(file);

        				ContentType contentType = FileUtil.detectFileType(file);

        				document.setContentType(contentType);

        				document.setName(entry.getValue().substring(entry.getValue().lastIndexOf('/') + 1));

        				document.setPath(entry.getValue());

        				document.setPersisted(true);

        				document.setSize(file.length());

        				documentRepository.persist(document);

        				pathMap.put(entry.getKey(), document.getId());

        			}catch(FileNotFoundException fileNotFoundException){

        			}

        		}

        	}

        	if(!pathMap.isEmpty()){

        		fofoIdPathMap.put(id, pathMap);

        	}

        }

        LOGGER.info("fofoIdPathMap {}", fofoIdPathMap);

        for(Map.Entry<Integer, Map<String, Integer>> entry : fofoIdPathMap.entrySet()){

        	mongoClient.updateColumnsById(entry.getValue(), entry.getKey());

        }

	}

}