package com.yami.trading.service.item;
|
|
import cn.hutool.core.bean.BeanUtil;
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
import com.yami.trading.bean.data.domain.Realtime;
|
import com.yami.trading.bean.item.domain.Item;
|
import com.yami.trading.bean.item.domain.ItemUserOptional;
|
import com.yami.trading.bean.item.domain.ItemUserOptionalItem;
|
import com.yami.trading.bean.item.dto.ItemUserOptionalDTO;
|
import com.yami.trading.bean.item.dto.ItemUserOptionalItemDTO;
|
import com.yami.trading.service.data.DataService;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.beans.factory.annotation.Qualifier;
|
import org.springframework.stereotype.Service;
|
import org.springframework.transaction.annotation.Transactional;
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
import com.yami.trading.bean.item.domain.ItemUserOptionalList;
|
import com.yami.trading.dao.item.ItemUserOptionalListMapper;
|
|
import javax.naming.ldap.HasControls;
|
import java.util.HashSet;
|
import java.util.List;
|
import java.util.Set;
|
import java.util.stream.Collectors;
|
|
/**
|
* 自选分组Service
|
* @author lucas
|
* @version 2023-03-10
|
*/
|
@Service
|
@Transactional
|
public class ItemUserOptionalListService extends ServiceImpl<ItemUserOptionalListMapper, ItemUserOptionalList> {
|
@Autowired
|
@Qualifier("dataService")
|
private DataService dataService;
|
|
@Autowired
|
private ItemUserOptionalItemService itemUserOptionalItemService;
|
|
@Autowired
|
private ItemUserOptionalService itemUserOptionalService;
|
@Autowired
|
private ItemService itemService;
|
/**
|
* 查询用户拥有的分组
|
* @param partyId
|
* @return
|
*/
|
public List<ItemUserOptionalList> findListByPartyId(String partyId){
|
QueryWrapper queryWrapper = new QueryWrapper();
|
queryWrapper.eq("PARTY_ID", partyId);
|
List<ItemUserOptionalList> list = list(queryWrapper);
|
return list;
|
}
|
|
/**
|
* 查询用户拥有的分组
|
* @param partyId
|
* @return
|
*/
|
public List<ItemUserOptionalItemDTO> findListItemsByPartyId(String partyId, String listId){
|
QueryWrapper<ItemUserOptionalItem> queryWrapper = new QueryWrapper();
|
queryWrapper.eq("PARTY_ID", partyId);
|
queryWrapper.eq("LIST_ID", listId);
|
List<ItemUserOptionalItem> list = itemUserOptionalItemService.list(queryWrapper);
|
List<ItemUserOptionalItemDTO> models = BeanUtil.copyToList(list, ItemUserOptionalItemDTO.class);
|
|
for(ItemUserOptionalItemDTO dto: models){
|
dto.setName(itemService.findBySymbol(dto.getSymbol()).getName());
|
}
|
return models;
|
|
}
|
|
/**
|
* 查询用户拥有的分组
|
* @param partyId
|
* @return
|
*/
|
public List<ItemUserOptionalItemDTO> findListItemsByPartyIdAndType(String partyId, String type){
|
List<String> byType = itemService.findByType(type).stream().map(Item::getSymbol).collect(Collectors.toList());
|
byType.add("-1");
|
|
|
QueryWrapper<ItemUserOptionalItem> queryWrapper = new QueryWrapper();
|
queryWrapper.select("distinct SYMBOL,PARTY_ID");
|
queryWrapper.eq("PARTY_ID", partyId);
|
queryWrapper.in("SYMBOL", byType);
|
List<ItemUserOptionalItem> list = itemUserOptionalItemService.list(queryWrapper);
|
List<String> optionalSymbols = itemUserOptionalService.getOptionalSymbols(partyId);
|
|
List<ItemUserOptionalItemDTO> models = BeanUtil.copyToList(list, ItemUserOptionalItemDTO.class);
|
Set<String> allSymbos = new HashSet<>();
|
|
for(ItemUserOptionalItemDTO dto: models){
|
dto.setName(itemService.findBySymbol(dto.getSymbol()).getName());
|
Realtime realtime = dataService.realtime(dto.getSymbol()).get(0);
|
dto.setClose(realtime.getClose());
|
dto.setChangeRatio(realtime.getChangeRatio());
|
dto.setChangeRatioStr(realtime.getChangeRatio());
|
dto.setTurnoverRate(realtime.getTurnoverRate());
|
dto.setVolumeRatio(realtime.getVolumeRatio());
|
allSymbos.add(dto.getSymbol());
|
}
|
|
for (String symbol : optionalSymbols) {
|
if (allSymbos.contains(symbol)) {
|
continue;
|
}
|
ItemUserOptionalItemDTO dto = new ItemUserOptionalItemDTO();
|
dto.setSymbol(symbol);
|
Item bySymbol = itemService.findBySymbol(dto.getSymbol());
|
if (!bySymbol.getType().equalsIgnoreCase(type)) {
|
continue;
|
}
|
dto.setName(bySymbol.getName());
|
|
// 获取实时数据
|
List<Realtime> realtimeList = dataService.realtime(dto.getSymbol());
|
|
// 检查实时数据列表是否为空
|
if (realtimeList.isEmpty()) {
|
continue; // 如果列表为空，跳过当前循环
|
}
|
|
// 获取第一个实时数据
|
Realtime realtime = realtimeList.get(0);
|
|
// 设置 DTO 属性
|
dto.setClose(realtime.getClose());
|
dto.setChangeRatio(realtime.getChangeRatio());
|
dto.setChangeRatioStr(realtime.getChangeRatio());
|
dto.setTurnoverRate(realtime.getTurnoverRate());
|
dto.setVolumeRatio(realtime.getVolumeRatio());
|
|
allSymbos.add(dto.getSymbol());
|
models.add(dto);
|
}
|
return models;
|
|
}
|
/**
|
*
|
* @param partyId
|
* @param name
|
* @return
|
*/
|
public ItemUserOptionalList findOne(String partyId, String name){
|
QueryWrapper<ItemUserOptionalList> queryWrapper = new QueryWrapper<>();
|
queryWrapper.eq("PARTY_ID", partyId);
|
queryWrapper.eq("NAME", name);
|
return baseMapper.selectOne(queryWrapper);
|
}
|
|
|
}
|
