Subversion Repositories SmartDukaan

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

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

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

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

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

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

import com.spice.profitmandi.dao.entity.fofo.PartnerType;

	public ChartInvestmentModel getAllStatePartnerType(Map<Integer, PartnerDetailModel> fofoIdAndallValues)

			throws ProfitMandiBusinessException {

		Map<String, Float> partnerTypeCount = new HashMap<>();

		List<PartnerType> partnerTypes = Arrays.asList(PartnerType.values());

		for (PartnerType partnerType : partnerTypes) {

			float count = fofoIdAndallValues.entrySet().stream()

					.filter(x -> x.getValue().getPartnerType().equals(partnerType)).count();

			partnerTypeCount.put(partnerType.toString(), count);

		}

		ChartInvestmentModel cm = new ChartInvestmentModel();

		HashSet<String> labels = new HashSet<String>();

		labels.addAll(partnerTypes.stream().skip(1).map(x -> x.toString()).collect(Collectors.toList()));

		List<String> labelList = new ArrayList<>(labels);

		List<String> backgroundColor = new ArrayList<>();

		List<Float> values = new ArrayList<>();

		for (String label : labelList) {

			values.add(partnerTypeCount.get(label));

			if (label.equals("BRONZE")) {

				backgroundColor.add("#FFD700");

			}

			if (label.equals("SILVER")) {

				backgroundColor.add("silver");

			}

			if (label.equals("GOLD")) {

				backgroundColor.add("yellow");

			}

			if (label.equals("DIAMOND")) {

				backgroundColor.add("#B9F2FF");

			}

			if (label.equals("PLATINUM")) {

				backgroundColor.add("#e5e4e2");

			}

			if (label.equals("NEW")) {

				backgroundColor.add("green");

			}

			if (label.equals("ALL")) {

				backgroundColor.add("blue");

			}

		}

		LOGGER.info("valuesPartnerType" + values);

		LOGGER.info("partnerTypeCount" + partnerTypeCount);

		Data data = new Data();

		data.setData(values);

		data.setBackgroundColor(backgroundColor);

		data.setLabel("DataSet 1");

		PieLables label = new PieLables();

		label.setFontColor("black");

		label.setFontSize(22);

		Legend legend = new Legend();

		legend.setLabels(label);

		legend.setPosition("left");

		List<Data> dataList = new ArrayList<>();

		dataList.add(data);

		DataInvestmentModel datasets = new DataInvestmentModel();

		datasets.setDatasets(dataList);

		datasets.setLabels(labels);

		OptionModel om = new OptionModel();

		om.setLegend(legend);

		cm.setType("pie");

		cm.setData(datasets);

		cm.setOptions(om);

		return cm;

	}