Subversion Repositories SmartDukaan

package com.spice.profitmandi.service;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

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

import org.springframework.stereotype.Component;

import com.spice.profitmandi.common.model.ProfitMandiConstants;

import com.spice.profitmandi.dao.entity.logistics.PostOffice;

import com.spice.profitmandi.dao.repository.logistics.PostOfficeRepository;

@Component

public class PostOfficeServiceImpl implements PostOfficeService {

	@Autowired

	private PostOfficeRepository postOfficeRepository;

	private static final Map<Integer, String> pincodeCitiesMap = new HashMap<Integer, String>();

	static {

		pincodeCitiesMap.put(160104, "Ludhiana");

	}

	@Override

	public Map<String, Object> getPOPincode(int pincode) {

		List<PostOffice> postOffices = postOfficeRepository.selectByPinCode(pincode);

		Set<String> cities = new HashSet<>();

		//Add custom pincodes to support the major cities;

		if(pincodeCitiesMap.containsKey(pincode)) {

			cities.add(pincodeCitiesMap.get(pincode));

		}

		Map<String, Object> map = new HashMap<>(3);

		map.put(ProfitMandiConstants.PIN_CODE, pincode);

		for (PostOffice postOffice : postOffices) {

			String city = postOffice.getCity();

			if (city.equals("NA")) {

				cities.add(postOffice.getDistrict());

			} else {

				cities.add(postOffice.getCity());

			}

			map.put(ProfitMandiConstants.STATE, postOffice.getState());

		}

		map.put(ProfitMandiConstants.CITIES, cities);

		return map;

	}

}