1
zj
2024-04-24 29446dbdaea7d11eb28a4210ec69a186c54e705e
src/main/java/com/nq/service/impl/StockSubscribeServiceImpl.java
@@ -1,9 +1,12 @@
package com.nq.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.github.pagehelper.PageInfo;
import com.nq.common.ServerResponse;
import com.nq.dao.StockMapper;
import com.nq.pojo.Stock;
import com.nq.pojo.StockSubscribe;
import com.nq.service.IStockSubscribeService;
import com.nq.dao.StockSubscribeMapper;
@@ -11,6 +14,7 @@
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.List;
@@ -23,8 +27,11 @@
@Slf4j
public class StockSubscribeServiceImpl extends ServiceImpl<StockSubscribeMapper, StockSubscribe>
        implements IStockSubscribeService {
    @Autowired
    @Resource
    StockSubscribeMapper stockSubscribeMapper;
    @Autowired
    StockMapper stockMapper;
    /**
     * @Description: 用户新股列表
@@ -72,15 +79,10 @@
    public ServerResponse add(StockSubscribe model, HttpServletRequest request) {
        //判断是否已经存在
        log.info("model:{}", model);
        List<StockSubscribe> stockSubscribeList = this.stockSubscribeMapper.selectList(new QueryWrapper<StockSubscribe>().eq("code", model.getCode()));
        if (stockSubscribeList != null && stockSubscribeList.size() > 0) {
            return ServerResponse.createByErrorMsg("新股已经存在,不要重复添加");
        Stock stock = stockMapper.selectOne(new LambdaQueryWrapper<Stock>().eq(Stock::getStockSpell, model.getCode()).last(" limit 1"));
        if(null == stock){
            return ServerResponse.createByErrorMsg("新股代码不存在");
        }
//        String sinaStock = SinaStockApi.getInStockDetail(model.getCode());
//        String[] arrayOfString = sinaStock.split(",");
//        if (arrayOfString.length < 10)
//        return ServerResponse.createByErrorMsg("数据源无该新股");
        //添加新股
        int resultCount = this.stockSubscribeMapper.insert(model);
        if (resultCount > 0) {
            return ServerResponse.createBySuccessMsg("添加新股成功");