package com.yami.trading.admin.task;

import com.alibaba.fastjson.JSONObject;

import com.yami.trading.api.websocket.WebSocketServer;
import com.yami.trading.api.websocket.WebSocketSession;
import com.yami.trading.bean.data.domain.Trade;
import com.yami.trading.bean.data.domain.TradeEntry;
import com.yami.trading.bean.item.domain.Item;
import com.yami.trading.common.util.Arith;
import com.yami.trading.common.util.RandomUtil;
import com.yami.trading.common.util.ThreadUtils;
import com.yami.trading.common.web.ResultObject;
import com.yami.trading.huobi.data.DataCache;
import com.yami.trading.huobi.data.internal.TradeTimeObject;
import com.yami.trading.huobi.data.job.DataQueue;
import com.yami.trading.huobi.data.job.HandleObject;
import com.yami.trading.service.item.ItemService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.compress.utils.Lists;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;


import java.math.RoundingMode;
import java.text.DecimalFormat;
import java.util.*;

/**
 * 近期交易记录
 *
 */
@Component
@Slf4j
public class TradePushJob implements Runnable {

	@Autowired
	private ItemService itemService;

	public void start() {
		new Thread(this, "tradePushJob").start();
		if (log.isInfoEnabled())
			log.info("启动tradePushJob！");
	}

	public void run() {

		while (true) {
			try {
				this.realtimeHandle();
			} catch (Exception e) {
				log.error("run fail", e);
			} finally {
				ThreadUtils.sleep(500);
			}
		}

	}

	private void realtimeHandle() {
		try {
			// 数据处理
			ResultObject tradeResult = new ResultObject();
			HashMap<Object, Object> data = new HashMap<>();
			data.put("data", Lists.newArrayList());
			tradeResult.setData(data);
			Map<String, String> tradeResultMap = new HashMap<>();
			
			if (!WebSocketServer.tradeMap.isEmpty()) {

				// 客户端请求的所有币种，去重集合
				Set<String> symbolSet = new HashSet<String>();
				for (String socketKey : WebSocketServer.tradeMap.keySet()) {
					WebSocketSession webSocketSession = WebSocketServer.tradeMap.get(socketKey);
					String symbolKey = webSocketSession.getParam();
					symbolSet.add(symbolKey);
				}
				
				for (String symbol : symbolSet) {
					data.put("symbol", symbol);
					 TradeTimeObject trade = DataCache.getTrade().get(symbol);
					if (null != trade && null != trade.getTrade()) {
						Trade tradeData = trade.getTrade();
						if(itemService.findBySymbol(symbol).getType().equalsIgnoreCase(Item.cryptos)){
							tradeResult.setData(tradeRevise(tradeData, symbol));
						}else{
							Map<String, Object> result =  new HashMap<>();
							result.put("symbol", symbol);
							result.put("ts", tradeData.getTs());
							result.put("data", tradeData.getData());
							tradeResult.setData(result);
						}
					} else {
						Item item = this.itemService.findBySymbol(symbol);
						HandleObject handleObject = new HandleObject();
						handleObject.setType(HandleObject.type_trade);
						handleObject.setItem(item);
						DataQueue.add(handleObject);
					}
					tradeResultMap.put(symbol, JSONObject.toJSONString(tradeResult));
					// System.out.println("交易记录 推送 " + JSONObject.toJSONString(tradeResult));
				}
				
				if (tradeResultMap.isEmpty()) {
					return;
				}
				
				for (String socketKey : WebSocketServer.tradeMap.keySet()) {
//					long timeMillins = System.currentTimeMillis();
					//WebSocketServer server = WebSocketServer.tradeMap.get(socketKey);
//					if (server.getTimeStr() != 0 && timeMillins > server.getTimeStr()) {
//						server.onClose();
//						return;
//					}
					WebSocketSession webSocketSession = WebSocketServer.tradeMap.get(socketKey);

					String type = webSocketSession.getType();
					String symbolKey = webSocketSession.getParam();
					WebSocketServer.sendToMessageById(socketKey, tradeResultMap.get(symbolKey), type);
				}
			}
		} catch (Throwable e) {
			e.printStackTrace();
		}
	}
	
	/**
	 * 近期交易记录解析
	 */
	private Map<String, Object> tradeRevise(Trade data, String symbol) {
		Map<String, Object> map = new HashMap<String, Object>();
		map.put("symbol", symbol);
		map.put("ts", data.getTs());
		Item item = this.itemService.findBySymbol(data.getSymbol());
		List<Map<String, Object>> tradeEntry_list = new ArrayList<Map<String, Object>>();
		
		for (int i = 0; i < data.getData().size(); i++) {
			TradeEntry tradeEntry = data.getData().get(i);
			Map<String, Object> tradeEntry_map = new HashMap<String, Object>();
			String direction = tradeEntry.getDirection();
			Random random = new Random();
			if (random.nextBoolean()) {
				if ("buy".equals(direction)) {
					direction = "sell";
				}else {
					direction = "buy";
				}
			}
			tradeEntry_map.put("direction", direction);
			tradeEntry_map.put("ts", tradeEntry.getTs());
			tradeEntry_map.put("current_time", tradeEntry.getCurrent_time());
			
			double addPriceValue = getRandomValue((int)tradeEntry.getPrice().doubleValue());
			double addAmountValue = getRandomValue((int)tradeEntry.getAmount().doubleValue());
			
			double price = Arith.add(tradeEntry.getPrice(), addPriceValue);
			double amount = Arith.add(tradeEntry.getAmount(), addAmountValue);

			if (item.getDecimals() == null || item.getDecimals() < 0) {
				tradeEntry_map.put("price", price);
				tradeEntry_map.put("amount", amount);
			} else {
				String format = "";
				if (item.getDecimals() == 0) {
					format = "#";
				} else {
					format = "#.";
					for (int j = 0; j < item.getDecimals(); j++) {
						format = format + "#";
					}
				}

				DecimalFormat df = new DecimalFormat(format);
				df.setRoundingMode(RoundingMode.FLOOR);// 向下取整

				tradeEntry_map.put("price", df.format(price));
				tradeEntry_map.put("amount", df.format(amount));
			}
			tradeEntry_list.add(tradeEntry_map);

		}
		map.put("data", tradeEntry_list);
		return map;
	}
	
	private double getRandomValue(int value) {
		double addValue;
		if (value > 0) {
		    int count = 0;
		    while (value > 0) {
		    	value = value / 10;
		        count++;
		    }
		    // 个
		    if (count == 1) {
		    	addValue = RandomUtil.randomFloat(0.01, 0.1999, 4);
		    	return addValue;
		    } 
		    // 十
		    if (count == 2) {
		    	addValue = RandomUtil.randomFloat(0.1, 0.5999, 4);
		    	return addValue;
		    } 
		    // 百
		    if (count == 3) {
		    	addValue = RandomUtil.randomFloat(0.1, 2.9999, 4);
		    	return addValue;
		    } 
		    // 千
		    if (count == 4) {
		    	addValue = RandomUtil.randomFloat(1, 3.9999, 4);
		    	return addValue;
		    } 
		    // 万
		    if (count == 5) {
		    	addValue = RandomUtil.randomFloat(1, 5.9999, 4);
		    	return addValue;
		    } 
		    // 十万
		    else {
		    	addValue = RandomUtil.randomFloat(1, 5.9999, 4);
		    	return addValue;
		    }
		} else {
			addValue = RandomUtil.randomFloat(0.01, 0.2999, 4);
			return addValue;
		}
	}

	public void setItemService(ItemService itemService) {
		this.itemService = itemService;
	}
	
}