1

ydj

2024-06-11 14b46ebd008c9f9fbaa0dd43a49783d2f54536a3

1

 websocketSerivce/src/main/java/org/example/controller/ApiController.java

@@ -5,15 +5,19 @@
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.example.common.ServerResponse;
import org.example.dao.DataServiceKeyMapper;
import org.example.dao.JournalismMapper;
import org.example.dao.StockMarketNewMapper;
import org.example.enums.EStockType;
import org.example.pojo.DataServiceKey;
import org.example.pojo.Journalism;
import org.example.pojo.StockMarketNew;
import org.example.util.HttpClientRequest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

/**
@@ -31,9 +35,17 @@
    @Autowired
    StockMarketNewMapper stockMarketNewMapper;

    @Autowired
    DataServiceKeyMapper dataServiceKeyMapper;


    @GetMapping("/stock-markets")
    public List JournalismAll(){
    public List JournalismAll(@RequestParam(value="key",required = true)String key){
        DataServiceKey data = dataServiceKeyMapper.selectOne(
                new LambdaQueryWrapper<DataServiceKey>().eq(DataServiceKey::getTokenKey,key));
        if(data == null){
            return Collections.singletonList(new ArrayList().add("key invalid"));
        }
        QueryWrapper<Journalism> queryWrapper = new QueryWrapper<>();
        queryWrapper.orderByDesc("time");
        Page<Journalism> page = new Page<>(1,15);

 websocketSerivce/src/main/java/org/example/dao/DataServiceKeyMapper.java

New file
@@ -0,0 +1,9 @@
package org.example.dao;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import org.apache.ibatis.annotations.Mapper;
import org.example.pojo.DataServiceKey;

@Mapper
public interface DataServiceKeyMapper extends BaseMapper<DataServiceKey> {
}

 websocketSerivce/src/main/java/org/example/pojo/DataServiceKey.java

New file
@@ -0,0 +1,58 @@
package org.example.pojo;

import com.baomidou.mybatisplus.annotation.*;
import lombok.Data;

import java.io.Serializable;
import java.util.Date;

/**
 * @program: webSocketProject
 * @description:
 * @create: 2024-03-27 17:01
 **/
@Data
@TableName("data_service_key")
public class DataServiceKey implements Serializable {

    private static final long serialVersionUID = 1L;


    /**
     * id
     */
    @TableId(type = IdType.AUTO)
    private Integer id;

    /**
     * key
     */
    private String tokenKey;

    /**
     * 是否可用 0：否 1： 是
     */
    private Integer isAvailable;

    /**
     * 到期时间
     */
    private Date expirationTime;

    /**
     * 添加时间
     */
    private Date startTime;

    /**
     * 修改时间
     */
    private Date updateTime;

    /**
     * 备注
     */
    private String remark;

    public DataServiceKey() {}
}

 websocketSerivce/src/main/java/org/example/websocket/controller/GenerateKey.java

New file
@@ -0,0 +1,50 @@
package org.example.websocket.controller;

import cn.hutool.core.date.DateUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.example.common.ServerResponse;
import org.example.dao.DataServiceKeyMapper;
import org.example.pojo.DataServiceKey;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.util.Date;
import java.util.UUID;


/**
 * @program: webSocketProject
 * @description: 生成key
 * @create: 2024-03-27 19:38
 **/
@RestController
@RequestMapping("/api")
public class GenerateKey {

    @Autowired
    DataServiceKeyMapper dataServiceKeyMapper;

    @PostMapping("/creationKey")
    public ServerResponse sendNotification(@RequestParam("time") Date time) {
        String randomKey = UUID.randomUUID().toString();
        try {
            Long count = dataServiceKeyMapper.selectCount(new LambdaQueryWrapper<DataServiceKey>().eq(DataServiceKey::getTokenKey, randomKey));
            if(count > 0){
                return ServerResponse.createByErrorMsg("请重新生成");
            }
            System.out.println(randomKey);
            DataServiceKey dataServiceKey = new DataServiceKey();
            dataServiceKey.setTokenKey(randomKey);
            dataServiceKey.setExpirationTime(time);
            dataServiceKey.setStartTime(DateUtil.date());
            dataServiceKeyMapper.insert(dataServiceKey);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return ServerResponse.createBySuccessMsg(randomKey);
    }

}