From 14b46ebd008c9f9fbaa0dd43a49783d2f54536a3 Mon Sep 17 00:00:00 2001
From: ydj <yangsink@163.com>
Date: Tue, 11 Jun 2024 15:39:40 +0800
Subject: [PATCH] 1
---
websocketSerivce/src/main/java/org/example/websocket/controller/GenerateKey.java | 50 ++++++++++++++++
websocketSerivce/src/main/java/org/example/pojo/DataServiceKey.java | 58 +++++++++++++++++++
websocketSerivce/src/main/java/org/example/controller/ApiController.java | 14 ++++
websocketSerivce/src/main/java/org/example/dao/DataServiceKeyMapper.java | 9 +++
4 files changed, 130 insertions(+), 1 deletions(-)
diff --git a/websocketSerivce/src/main/java/org/example/controller/ApiController.java b/websocketSerivce/src/main/java/org/example/controller/ApiController.java
index 5738f0c..e7f38f8 100644
--- a/websocketSerivce/src/main/java/org/example/controller/ApiController.java
+++ b/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);
diff --git a/websocketSerivce/src/main/java/org/example/dao/DataServiceKeyMapper.java b/websocketSerivce/src/main/java/org/example/dao/DataServiceKeyMapper.java
new file mode 100644
index 0000000..a89d006
--- /dev/null
+++ b/websocketSerivce/src/main/java/org/example/dao/DataServiceKeyMapper.java
@@ -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> {
+}
diff --git a/websocketSerivce/src/main/java/org/example/pojo/DataServiceKey.java b/websocketSerivce/src/main/java/org/example/pojo/DataServiceKey.java
new file mode 100644
index 0000000..9bdce6b
--- /dev/null
+++ b/websocketSerivce/src/main/java/org/example/pojo/DataServiceKey.java
@@ -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() {}
+}
diff --git a/websocketSerivce/src/main/java/org/example/websocket/controller/GenerateKey.java b/websocketSerivce/src/main/java/org/example/websocket/controller/GenerateKey.java
new file mode 100644
index 0000000..34c4205
--- /dev/null
+++ b/websocketSerivce/src/main/java/org/example/websocket/controller/GenerateKey.java
@@ -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);
+ }
+
+}
--
Gitblit v1.9.3