style: 代码格式化

MySQL 8.0+ 默认使用 caching_sha2_password 认证插件，这种认证方式有两种工作模式：
- 如果使用SSL连接，直接通过安全通道传输密码
- 如果不使用SSL连接，客户端需要从服务器获取RSA公钥来加密密码
当设置 useSSL=false 但未明确允许公钥检索时，JDBC驱动出于安全考虑会阻止这种操作（报错：Public Key Retrieval is not allowed）。
生产环境中，应优先考虑：
1. 启用 SSL/TLS 加密连接
2. 如需禁用 SSL，改用 mysql_native_password 认证
3. 仅在受控环境（如开发环境）中使用 allowPublicKeyRetrieval=true

config/shardingsphere-jdbc.yml

@@ -11,13 +11,13 @@ dataSources:
   ds_1:
     dataSourceClassName: com.zaxxer.hikari.HikariDataSource
     driverClassName: com.mysql.cj.jdbc.Driver
-    jdbcUrl: jdbc:mysql://localhost:3306/novel_plus?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghai
+    jdbcUrl: jdbc:mysql://localhost:3306/novel_plus?allowPublicKeyRetrieval=true&useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghai
     username: root
     password: test123456
   ds_2:
     dataSourceClassName: com.zaxxer.hikari.HikariDataSource
     driverClassName: com.mysql.cj.jdbc.Driver
-    url: jdbc:mysql://localhost:3306/information_schema?useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghai
+    url: jdbc:mysql://localhost:3306/information_schema?allowPublicKeyRetrieval=true&useUnicode=true&characterEncoding=utf-8&useSSL=false&serverTimezone=Asia/Shanghai
     username: root
     password: test123456
 # 规则配置

novel-common/src/main/java/com/java2nb/novel/core/advice/CustomResponseBodyAdvice.java

@@ -0,0 +1,53 @@
+package com.java2nb.novel.core.advice;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.databind.module.SimpleModule;
+import com.fasterxml.jackson.databind.ser.std.ToStringSerializer;
+import org.springframework.core.MethodParameter;
+import org.springframework.http.MediaType;
+import org.springframework.http.converter.HttpMessageConverter;
+import org.springframework.http.converter.json.Jackson2ObjectMapperBuilder;
+import org.springframework.http.server.ServerHttpRequest;
+import org.springframework.http.server.ServerHttpResponse;
+import org.springframework.web.bind.annotation.RestControllerAdvice;
+import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
+
+import java.util.Objects;
+
+/**
+ * 在对 RestController 返回对象 json 序列化时，将所有 Long 类型转为 String 类型返回，避免前端数据精度丢失的问题
+ * 取代 spring.jackson.generator.write-numbers-as-strings=true 配置，避免影响全局的 ObjectMapper
+ *
+ * @author xiongxiaoyang
+ * */
+@RestControllerAdvice
+public class CustomResponseBodyAdvice implements ResponseBodyAdvice<Object> {
+
+    private final ObjectMapper customObjectMapper;
+
+    public CustomResponseBodyAdvice(Jackson2ObjectMapperBuilder builder) {
+        customObjectMapper = builder.createXmlMapper(false).build();
+        SimpleModule simpleModule = new SimpleModule();
+        simpleModule.addSerializer(Long.class, ToStringSerializer.instance);
+        simpleModule.addSerializer(Long.TYPE, ToStringSerializer.instance);
+        customObjectMapper.registerModule(simpleModule);
+    }
+
+    @Override
+    public boolean supports(MethodParameter returnType, Class<? extends HttpMessageConverter<?>> converterType) {
+        // 返回 true 表示对所有 Controller 的响应都生效
+        return true;
+    }
+
+    @Override
+    public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class<? extends HttpMessageConverter<?>> selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
+        // 使用自定义的 ObjectMapper 序列化响应体
+        if(Objects.nonNull(body)) {
+            return customObjectMapper.valueToTree(body);
+        }else{
+            return null;
+        }
+    }
+
+}
+

novel-common/src/main/java/com/java2nb/novel/core/cache/CacheKey.java

@@ -69,4 +69,8 @@ public interface CacheKey {
      * 测试爬虫规则缓存
      */
     String BOOK_TEST_PARSE = "testParse";
+    /**
+     * AI生成图片
+     * */
+    String AI_GEN_PIC = "aiGenPic";
 }

novel-common/src/main/resources/application-common.yml

@@ -6,11 +6,6 @@ spring:
     mode: LEGACYHTML5 #去除thymeleaf的html严格校验thymeleaf.mode=LEGACYHTML5
     cache: false # 是否开启模板缓存，默认true,建议在开发时关闭缓存,不然没法看到实时
 
-  # 将所有数字转为 String 类型返回，避免前端数据精度丢失的问题
-  jackson:
-    generator:
-      write-numbers-as-strings: true
-
   #上传文件的最大值（100M）
   servlet:
     multipart:

novel-front/src/main/java/com/java2nb/novel/controller/AuthorController.java

@@ -17,7 +17,14 @@ import jakarta.servlet.http.HttpServletRequest;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.messages.UserMessage;
+import org.springframework.ai.chat.model.ChatModel;
+import org.springframework.ai.chat.model.ChatResponse;
+import org.springframework.ai.chat.prompt.Prompt;
+import org.springframework.ai.openai.OpenAiChatModel;
+import org.springframework.http.MediaType;
 import org.springframework.web.bind.annotation.*;
+import reactor.core.publisher.Flux;
 
 import java.util.Date;
 
@@ -28,7 +35,7 @@ import java.util.Date;
 @RestController
 @Slf4j
 @RequiredArgsConstructor
-public class AuthorController extends BaseController{
+public class AuthorController extends BaseController {
 
     private final AuthorService authorService;
 
@@ -36,62 +43,64 @@ public class AuthorController extends BaseController{
 
     private final ChatClient chatClient;
 
+    private final OpenAiChatModel chatModel;
+
     /**
      * 校验笔名是否存在
-     * */
+     */
     @GetMapping("checkPenName")
-    public RestResult<Boolean> checkPenName(String penName){
+    public RestResult<Boolean> checkPenName(String penName) {
 
         return RestResult.ok(authorService.checkPenName(penName));
     }
 
     /**
      * 作家发布小说分页列表查询
-     * */
+     */
     @GetMapping("listBookByPage")
-    public RestResult<PageBean<Book>> listBookByPage(@RequestParam(value = "curr", defaultValue = "1") int page, @RequestParam(value = "limit", defaultValue = "10") int pageSize , HttpServletRequest request){
+    public RestResult<PageBean<Book>> listBookByPage(@RequestParam(value = "curr", defaultValue = "1") int page,
+        @RequestParam(value = "limit", defaultValue = "10") int pageSize, HttpServletRequest request) {
 
-        return RestResult.ok(bookService.listBookPageByUserId(getUserDetails(request).getId(),page,pageSize));
+        return RestResult.ok(bookService.listBookPageByUserId(getUserDetails(request).getId(), page, pageSize));
     }
 
     /**
      * 发布小说
-     * */
+     */
     @PostMapping("addBook")
-    public RestResult<Void> addBook(@RequestParam("bookDesc") String bookDesc, Book book, HttpServletRequest request){
+    public RestResult<Void> addBook(@RequestParam("bookDesc") String bookDesc, Book book, HttpServletRequest request) {
 
         Author author = checkAuthor(request);
 
         //bookDesc不能使用book对象来接收，否则会自动去掉前面的空格
         book.setBookDesc(bookDesc
-                .replaceAll("\\n","<br>")
-                .replaceAll("\\s","&nbsp;"));
+            .replaceAll("\\n", "<br>")
+            .replaceAll("\\s", "&nbsp;"));
         //发布小说
-        bookService.addBook(book,author.getId(),author.getPenName());
+        bookService.addBook(book, author.getId(), author.getPenName());
 
         return RestResult.ok();
     }
 
     /**
      * 更新小说状态,上架或下架
-     * */
+     */
     @PostMapping("updateBookStatus")
-    public RestResult<Void> updateBookStatus(Long bookId,Byte status,HttpServletRequest request){
+    public RestResult<Void> updateBookStatus(Long bookId, Byte status, HttpServletRequest request) {
         Author author = checkAuthor(request);
 
         //更新小说状态,上架或下架
-        bookService.updateBookStatus(bookId,status,author.getId());
+        bookService.updateBookStatus(bookId, status, author.getId());
 
         return RestResult.ok();
     }
 
 
-
     /**
      * 删除章节
      */
     @DeleteMapping("deleteIndex/{indexId}")
-    public RestResult<Void> deleteIndex(@PathVariable("indexId") Long indexId,  HttpServletRequest request) {
+    public RestResult<Void> deleteIndex(@PathVariable("indexId") Long indexId, HttpServletRequest request) {
 
         Author author = checkAuthor(request);
 
@@ -105,7 +114,7 @@ public class AuthorController extends BaseController{
      * 更新章节名
      */
     @PostMapping("updateIndexName")
-    public RestResult<Void> updateIndexName(Long indexId,  String indexName, HttpServletRequest request) {
+    public RestResult<Void> updateIndexName(Long indexId, String indexName, HttpServletRequest request) {
 
         Author author = checkAuthor(request);
 
@@ -116,19 +125,18 @@ public class AuthorController extends BaseController{
     }
 
 
-
-
     /**
      * 发布章节内容
      */
     @PostMapping("addBookContent")
-    public RestResult<Void> addBookContent(Long bookId, String indexName, String content,Byte isVip, HttpServletRequest request) {
+    public RestResult<Void> addBookContent(Long bookId, String indexName, String content, Byte isVip,
+        HttpServletRequest request) {
         Author author = checkAuthor(request);
 
         content = content.replaceAll("\\n", "<br>")
-                .replaceAll("\\s", "&nbsp;");
+            .replaceAll("\\s", "&nbsp;");
         //发布章节内容
-        bookService.addBookContent(bookId, indexName, content,isVip, author.getId());
+        bookService.addBookContent(bookId, indexName, content, isVip, author.getId());
 
         return RestResult.ok();
     }
@@ -137,14 +145,14 @@ public class AuthorController extends BaseController{
      * 查询章节内容
      */
     @GetMapping("queryIndexContent/{indexId}")
-    public RestResult<String> queryIndexContent(@PathVariable("indexId") Long indexId,  HttpServletRequest request) {
+    public RestResult<String> queryIndexContent(@PathVariable("indexId") Long indexId, HttpServletRequest request) {
 
         Author author = checkAuthor(request);
 
         String content = bookService.queryIndexContent(indexId, author.getId());
 
         content = content.replaceAll("<br>", "\n")
-                .replaceAll("&nbsp;", " ");
+            .replaceAll("&nbsp;", " ");
 
         return RestResult.ok(content);
     }
@@ -153,11 +161,12 @@ public class AuthorController extends BaseController{
      * 更新章节内容
      */
     @PostMapping("updateBookContent")
-    public RestResult<Void> updateBookContent(Long indexId, String indexName, String content, HttpServletRequest request) {
+    public RestResult<Void> updateBookContent(Long indexId, String indexName, String content,
+        HttpServletRequest request) {
         Author author = checkAuthor(request);
 
         content = content.replaceAll("\\n", "<br>")
-                .replaceAll("\\s", "&nbsp;");
+            .replaceAll("\\s", "&nbsp;");
         //更新章节内容
         bookService.updateBookContent(indexId, indexName, content, author.getId());
 
@@ -168,38 +177,44 @@ public class AuthorController extends BaseController{
      * 修改小说封面
      */
     @PostMapping("updateBookPic")
-    public RestResult<Void> updateBookPic(@RequestParam("bookId") Long bookId,@RequestParam("bookPic") String bookPic,HttpServletRequest request) {
+    public RestResult<Void> updateBookPic(@RequestParam("bookId") Long bookId, @RequestParam("bookPic") String bookPic,
+        HttpServletRequest request) {
         Author author = checkAuthor(request);
-        bookService.updateBookPic(bookId,bookPic, author.getId());
+        bookService.updateBookPic(bookId, bookPic, author.getId());
         return RestResult.ok();
     }
 
 
     /**
      * 作家日收入统计数据分页列表查询
-     * */
+     */
     @GetMapping("listIncomeDailyByPage")
-    public RestResult<PageBean<AuthorIncomeDetail>> listIncomeDailyByPage(@RequestParam(value = "curr", defaultValue = "1") int page,
-                                                                          @RequestParam(value = "limit", defaultValue = "10") int pageSize ,
-                                                                          @RequestParam(value = "bookId", defaultValue = "0") Long bookId,
-                                                                          @RequestParam(value = "startTime",defaultValue = "2020-05-01") Date startTime,
-                                                                          @RequestParam(value = "endTime",defaultValue = "2030-01-01") Date endTime,
-                                                                          HttpServletRequest request){
+    public RestResult<PageBean<AuthorIncomeDetail>> listIncomeDailyByPage(
+        @RequestParam(value = "curr", defaultValue = "1") int page,
+        @RequestParam(value = "limit", defaultValue = "10") int pageSize,
+        @RequestParam(value = "bookId", defaultValue = "0") Long bookId,
+        @RequestParam(value = "startTime", defaultValue = "2020-05-01") Date startTime,
+        @RequestParam(value = "endTime", defaultValue = "2030-01-01") Date endTime,
+        HttpServletRequest request) {
 
-        return RestResult.ok(authorService.listIncomeDailyByPage(page,pageSize,getUserDetails(request).getId(),bookId,startTime,endTime));
+        return RestResult.ok(
+            authorService.listIncomeDailyByPage(page, pageSize, getUserDetails(request).getId(), bookId, startTime,
+                endTime));
     }
 
 
     /**
      * 作家月收入统计数据分页列表查询
-     * */
+     */
     @GetMapping("listIncomeMonthByPage")
-    public RestResult<PageBean<AuthorIncome>> listIncomeMonthByPage(@RequestParam(value = "curr", defaultValue = "1") int page,
-                                                                    @RequestParam(value = "limit", defaultValue = "10") int pageSize ,
-                                                                    @RequestParam(value = "bookId", defaultValue = "0") Long bookId,
-                                                                    HttpServletRequest request){
+    public RestResult<PageBean<AuthorIncome>> listIncomeMonthByPage(
+        @RequestParam(value = "curr", defaultValue = "1") int page,
+        @RequestParam(value = "limit", defaultValue = "10") int pageSize,
+        @RequestParam(value = "bookId", defaultValue = "0") Long bookId,
+        HttpServletRequest request) {
 
-        return RestResult.ok(authorService.listIncomeMonthByPage(page,pageSize,getUserDetails(request).getId(),bookId));
+        return RestResult.ok(
+            authorService.listIncomeMonthByPage(page, pageSize, getUserDetails(request).getId(), bookId));
     }
 
     private Author checkAuthor(HttpServletRequest request) {
@@ -218,22 +233,29 @@ public class AuthorController extends BaseController{
             throw new BusinessException(ResponseStatus.AUTHOR_STATUS_FORBIDDEN);
         }
 
-
         return author;
 
 
     }
 
+    /**
+     * 查询AI生成图片
+     */
+    @GetMapping("queryAiGenPic")
+    public RestResult<String> queryAiGenPic(@RequestParam("bookId") Long bookId) {
+        return RestResult.ok(bookService.queryAiGenPic(bookId));
+    }
+
     /**
      * AI扩写
      */
     @PostMapping("ai/expand")
     public RestResult<String> expandText(@RequestParam("text") String text, @RequestParam("ratio") Double ratio) {
-        String prompt = "请将以下文本扩写为原长度的" + ratio/100 + "倍：" + text;
+        String prompt = "请将以下文本扩写为原长度的" + ratio / 100 + "倍：" + text;
         return RestResult.ok(chatClient.prompt()
-                .user(prompt)
-                .call()
-                .content());
+            .user(prompt)
+            .call()
+            .content());
     }
 
     /**
@@ -241,11 +263,11 @@ public class AuthorController extends BaseController{
      */
     @PostMapping("ai/condense")
     public RestResult<String> condenseText(@RequestParam("text") String text, @RequestParam("ratio") Integer ratio) {
-        String prompt = "请将以下文本缩写为原长度的" + 100/ratio + "分之一：" + text;
+        String prompt = "请将以下文本缩写为原长度的" + 100 / ratio + "分之一：" + text;
         return RestResult.ok(chatClient.prompt()
-                .user(prompt)
-                .call()
-                .content());
+            .user(prompt)
+            .call()
+            .content());
     }
 
     /**
@@ -255,9 +277,9 @@ public class AuthorController extends BaseController{
     public RestResult<String> continueText(@RequestParam("text") String text, @RequestParam("length") Integer length) {
         String prompt = "请续写以下文本，续写长度约为" + length + "字：" + text;
         return RestResult.ok(chatClient.prompt()
-                .user(prompt)
-                .call()
-                .content());
+            .user(prompt)
+            .call()
+            .content());
     }
 
     /**
@@ -267,12 +289,57 @@ public class AuthorController extends BaseController{
     public RestResult<String> polishText(@RequestParam("text") String text) {
         String prompt = "请润色优化以下文本，保持原意：" + text;
         return RestResult.ok(chatClient.prompt()
-                .user(prompt)
-                .call()
-                .content());
+            .user(prompt)
+            .call()
+            .content());
     }
 
+    /**
+     * AI扩写
+     */
+    @GetMapping(value = "ai/stream/expand", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<String> streamExpandText(@RequestParam("text") String text, @RequestParam("ratio") Double ratio) {
+        String prompt = "请将以下文本扩写为原长度的" + ratio / 100 + "倍：" + text;
+        return chatClient.prompt()
+            .user(prompt)
+            .stream()
+            .content();
+    }
 
+    /**
+     * AI缩写
+     */
+    @GetMapping(value = "ai/stream/condense", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<String> streamCondenseText(@RequestParam("text") String text, @RequestParam("ratio") Integer ratio) {
+        String prompt = "请将以下文本缩写为原长度的" + 100 / ratio + "分之一：" + text;
+        return chatClient.prompt()
+            .user(prompt)
+            .stream()
+            .content();
+    }
 
+    /**
+     * AI续写
+     */
+    @GetMapping(value = "ai/stream/continue", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<String> streamContinueText(@RequestParam("text") String text, @RequestParam("length") Integer length) {
+        String prompt = "请续写以下文本，续写长度约为" + length + "字：" + text;
+        return chatClient.prompt()
+            .user(prompt)
+            .stream()
+            .content();
+    }
+
+    /**
+     * AI润色
+     */
+    @GetMapping(value = "/ai/stream/polish", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<String> streamPolishText(@RequestParam("text") String text) {
+        String prompt = "请润色优化以下文本，保持原意：" + text;
+        return chatClient.prompt()
+            .user(prompt)
+            .stream()
+            .content();
+    }
 
 }

novel-front/src/main/java/com/java2nb/novel/core/config/AiConfig.java

@@ -18,16 +18,7 @@ import org.springframework.web.client.RestClient;
 public class AiConfig {
 
     /**
-     * 目的：配置自定义的 RestClientBuilder 对象
-     * <p>
-     * 原因：Spring AI 框架的 ChatClient 内部通过 RestClient（Spring Framework 6 和 Spring Boot 3 中引入） 发起 HTTP REST 请求与远程的大模型服务进行通信，
-     * 如果项目中没有配置自定义的 RestClientBuilder 对象， 那么在 RestClient 的自动配置类 org.springframework.boot.autoconfigure.web.client.RestClientAutoConfiguration
-     * 中配置的 RestClientBuilder 对象会使用 Spring 容器中提供的 HttpMessageConverters， 由于本项目中配置了 spring.jackson.generator.write-numbers-as-strings
-     * = true， 所以 Spring 容器中的 HttpMessageConverters 在 RestClient 发起 HTTP REST 请求转换 Java 对象为 JSON 字符串时会自动将 Number 类型的
-     * Java 对象属性转换为字符串而导致请求参数错误
-     * <p>
-     * 示例："temperature": 0.7 =》"temperature": "0.7"
-     * {"code":20015,"message":"The parameter is invalid. Please check again.","data":null}
+     * 配置自定义的 RestClientBuilder 对象
      */
     @Bean
     public RestClient.Builder restClientBuilder() {

novel-front/src/main/java/com/java2nb/novel/service/BookService.java

@@ -290,4 +290,9 @@ public interface BookService {
      * @param authorId
      */
     void updateBookPic(Long bookId, String bookPic, Long authorId);
+
+    /**
+     * 查询AI生成图片
+     */
+    String queryAiGenPic(Long bookId);
 }

novel-front/src/main/java/com/java2nb/novel/service/impl/BookServiceImpl.java

@@ -116,7 +116,7 @@ public class BookServiceImpl implements BookService {
             }
             result = new ObjectMapper().writeValueAsString(
                 list.stream().collect(Collectors.groupingBy(BookSettingVO::getType)));
-            cacheService.set(CacheKey.INDEX_BOOK_SETTINGS_KEY, result);
+            cacheService.set(CacheKey.INDEX_BOOK_SETTINGS_KEY, result, 3600 * 24);
         }
         return new ObjectMapper().readValue(result, Map.class);
     }
@@ -559,6 +559,7 @@ public class BookServiceImpl implements BookService {
                     .where(id, isEqualTo(book.getId()))
                     .build()
                     .render(RenderingStrategies.MYBATIS3));
+                cacheService.set(CacheKey.AI_GEN_PIC + book.getId(), picUrl, 60 * 60);
             });
         }
     }
@@ -882,5 +883,10 @@ public class BookServiceImpl implements BookService {
             .render(RenderingStrategies.MYBATIS3));
     }
 
+    @Override
+    public String queryAiGenPic(Long bookId) {
+        return cacheService.get(CacheKey.AI_GEN_PIC + bookId);
+    }
 
-}
+
+}

novel-front/src/main/resources/templates/author/content_add.html

@@ -15,7 +15,7 @@
                 padding: 20px;
                 background: #fff;
                 border-radius: 8px;
-                box-shadow: 0 2px 12px 0 rgba(0,0,0,.1);
+                box-shadow: 0 2px 12px 0 rgba(0, 0, 0, .1);
             }
 
             /* 文本域样式 */
@@ -124,10 +124,10 @@
                                 <li id="contentLi" style="width: 500px">
                                     <div class="editor-container">
                                         <div class="ai-toolbar">
-                                            <a class="ai-link expand" data-type="expand">AI扩写</a>
-                                            <a class="ai-link condense" data-type="condense">AI缩写</a>
-                                            <a class="ai-link continue" data-type="continue">AI续写</a>
-                                            <a class="ai-link polish" data-type="polish">AI润色</a>
+                                            <a class="ai-link expand" data-type="stream/expand">AI扩写</a>
+                                            <a class="ai-link condense" data-type="stream/condense">AI缩写</a>
+                                            <a class="ai-link continue" data-type="stream/continue">AI续写</a>
+                                            <a class="ai-link polish" data-type="stream/polish">AI润色</a>
                                         </div>
                                         <textarea id="bookContent" name="bookContent"
                                                   placeholder="请输入文本内容..."></textarea>
@@ -268,7 +268,7 @@
         }, speed);
     }
 
-    $('.ai-toolbar .ai-link').click(function(e){
+    $('.ai-toolbar .ai-link').click(function (e) {
         e.preventDefault(); // 阻止默认链接行为
         const type = $(this).data('type');
         const textarea = $('#bookContent');
@@ -284,27 +284,27 @@
 
         // 参数配置
         let params = {text: selectedText};
-        if(type === 'expand' || type === 'condense'){
+        if (type === 'stream/expand' || type === 'stream/condense') {
             layer.prompt({
                 title: '请输入比例',
                 value: 2,
                 btn: ['确定', '取消'],
-                btn2: function(){
+                btn2: function () {
                     layer.close(loading);
                 },
-                cancel: function(){
+                cancel: function () {
                     layer.close(loading);
                 }
-            }, function(value, index){
-                if(isNaN(Number(value)) || isNaN(parseFloat(value))){
+            }, function (value, index) {
+                if (isNaN(Number(value)) || isNaN(parseFloat(value))) {
                     layer.msg('请输入正确的比例');
                     return;
                 }
-                if(type === 'expand' && value <= 1){
+                if (type === 'stream/expand' && value <= 1) {
                     layer.msg('请输入正确的比例');
                     return;
                 }
-                if(type === 'condense' && (value <=0 || value >= 1)){
+                if (type === 'stream/condense' && (value <= 0 || value >= 1)) {
                     layer.msg('请输入正确的比例');
                     return;
                 }
@@ -313,19 +313,19 @@
                 sendRequest(type, params, loading, textarea);
             });
             return;
-        }else if(type === 'continue'){
+        } else if (type === 'stream/continue') {
             layer.prompt({
                 title: '请输入续写长度（字数）',
                 value: 200,
                 btn: ['确定', '取消'],
-                btn2: function(){
+                btn2: function () {
                     layer.close(loading);
                 },
-                cancel: function(){
+                cancel: function () {
                     layer.close(loading);
                 }
-            }, function(value, index){
-                if(!Number.isInteger(Number(value)) || value <= 0){
+            }, function (value, index) {
+                if (!Number.isInteger(Number(value)) || value <= 0) {
                     layer.msg('请输入正确的长度');
                     return;
                 }
@@ -339,26 +339,28 @@
         sendRequest(type, params, loading, textarea);
     });
 
-    function sendRequest(type, params, loading, textarea){
-        $.ajax({
-            url: '/author/ai/' + type,
-            type: 'POST',
-            data: params,
-            success: function(res){
-                layer.close(loading);
-                // 将生成的内容追加到文本末尾
-                if(res.code == '200'){
-                    const newText = "\n\n【AI生成内容】" + res.data; // 添加换行符分隔
-                    typeWriter(textarea, newText); // 使用打字机效果
-                }else{
-                    layer.msg('AI内容生成失败，请稍后重试');
-                }
-            },
-            error: function(){
-                layer.msg('请求失败，请稍后重试');
-                layer.close(loading);
-            }
-        });
+    function sendRequest(type, params, loading, textarea) {
+        const url = `/author/ai/${type}?text=${encodeURIComponent(params.text)}&ratio=${params.ratio}&length=${params.length}`;
+        const eventSource = new EventSource(url);
+
+        // 监听消息事件
+        eventSource.onmessage = function (event) {
+            layer.close(loading);
+            const data = event.data;
+            console.log('Received data:', data);
+
+            textarea.val(textarea.val() + data);
+            // 滚动到底部
+            textarea.scrollTop(textarea[0].scrollHeight);
+        };
+
+        // 监听错误事件
+        eventSource.onerror = function (error) {
+            layer.close(loading);
+            console.error('EventSource failed:', error);
+            eventSource.close(); // 关闭连接
+        };
+
     }
 
 </script>

novel-front/src/main/resources/templates/author/index.html

@@ -51,8 +51,7 @@
         </div>
         <div class="my_r">
             <div id="noContentDiv">
-                <div class="tc" style="margin-top: 200px"><a href="/author/book_add.html" class="btn_red">创建作品</a>
-                </div>
+                <div class="tc" style="margin-top: 200px"><a href="/author/book_add.html" class="btn_red">创建作品</a></div>
 
             </div>
             <div class="my_bookshelf" id="hasContentDiv" style="display: none">
@@ -143,13 +142,12 @@
 <script src="/javascript/common.js" type="text/javascript"></script>
 
 <script language="javascript" type="text/javascript">
-    var searchCount = 0;
-    var timeout;
+    var coverUpdateInterval;
+
     search(1, 5);
 
     function search(curr, limit) {
-        searchCount++;
-        clearTimeout(timeout);
+        clearInterval(coverUpdateInterval);
         $.ajax({
             type: "get",
             url: "/author/listBookByPage",
@@ -159,7 +157,25 @@
                 if (data.code == 200) {
                     var bookList = data.data.list;
                     if (bookList.length > 0) {
-                        var aiPicGenerating = bookList[0].picUrl == '/images/default.gif'
+                        if(curr == 1 && bookList[0].picUrl == '/images/default.gif'){
+                            coverUpdateInterval = setInterval(function(){
+                                $.ajax({
+                                    type: "get",
+                                    url: "/author/queryAiGenPic",
+                                    data: {'bookId': bookList[0].id},
+                                    dataType: "json",
+                                    success: function (data) {
+                                        if(data.code == 200 && data.data){
+                                            $("#cover"+bookList[0].id).attr("src", data.data);
+                                            clearInterval(coverUpdateInterval);
+                                        }
+                                    }
+                                  });
+                            }, 3000);
+                            setTimeout(() => {
+                                clearInterval(coverUpdateInterval);
+                            }, 10000);
+                        }
                         $("#hasContentDiv").css("display", "block");
                         $("#noContentDiv").css("display", "none");
                         var bookListHtml = "";
@@ -171,15 +187,12 @@
                                  "                            </td>\n" +*/
 
                                 "                            <td style=\"position: relative\" class=\"goread\">\n" +
-                                "<input class=\"opacity\" onchange=\"picChange('" + book.id + "'," + i + ")\"\n" +
-                                "                                           type=\"file\" id=\"file" + i + "\" name=\"file\"\n" +
+                                "<input class=\"opacity\" onchange=\"picChange('" + book.id + "')\"\n" +
+                                "                                           type=\"file\" id=\"file" + book.id + "\" name=\"file\"\n" +
                                 "                                           title=\"点击上传图片\"\n" +
                                 "                                           style=\"z-index: 100;cursor: pointer;left: 30px; top: 0px; width: 60px; height: 80px; opacity: 0; position: absolute;  \"\n" +
                                 "                                    />" +
-                                "<img width='50' height='70' src='" + book.picUrl + "'/><br/>" +
-                                "                                " + book.bookName + "</td>\n" +
-
-
+                                "<img id=\"cover" + book.id + "\" width='50' height='70' src='" + book.picUrl + "'/><br/>" +                                "                                " + book.bookName + "</td>\n" +
                                 "                            <td class=\"goread\" >"
                                 + book.catName + "</td>\n" +
 
@@ -231,12 +244,6 @@
                             });
                         });
 
-                        if (curr === 1 && aiPicGenerating && searchCount < 10) {
-                            timeout = setTimeout(function () {
-                                search(curr, limit);
-                            }, 3000);
-                        }
-
                     }
 
 
@@ -288,19 +295,20 @@
     }
 
 
-    function picChange(bookId, i) {
-        var file = $("#file" + i).val(); //文件名称
+   function picChange(bookId) {
+        var file = $("#file" + bookId).val(); //文件名称
         if (file != "") {
-            if (checkPicUpload($("#file" + i)[0])) {
+            if (checkPicUpload($("#file" + bookId)[0])) {
 
                 $.ajaxFileUpload({
                     url: "/file/picUpload", //用于文件上传的服务器端请求地址
                     secureuri: false, //是否需要安全协议，一般设置为false
-                    fileElementId: "file" + i, //文件上传域的ID
+                    fileElementId: "file" + bookId, //文件上传域的ID
                     dataType: "json", //返回值类型 一般设置为json
                     type: "post",
                     success: function (data) { //服务器成功响应处理函数
                         if (data.code == 200) {
+                            let picUrl = data.data;
                             $.ajax({
                                 type: "POST",
                                 url: "/author/updateBookPic",
@@ -308,17 +316,13 @@
                                 dataType: "json",
                                 success: function (data) {
                                     if (data.code == 200) {
-
-                                        location.reload();
-
+                                        $("#cover"+bookId).attr("src", picUrl);
                                     } else {
-                                        lock = false;
                                         layer.alert(data.msg);
                                     }
 
                                 },
                                 error: function () {
-                                    lock = false;
                                     layer.alert('网络异常');
                                 }
                             })

templates/green/html/author/content_add.html

@@ -15,7 +15,7 @@
                 padding: 20px;
                 background: #fff;
                 border-radius: 8px;
-                box-shadow: 0 2px 12px 0 rgba(0,0,0,.1);
+                box-shadow: 0 2px 12px 0 rgba(0, 0, 0, .1);
             }
 
             /* 文本域样式 */
@@ -124,10 +124,10 @@
                                 <li id="contentLi" style="width: 500px">
                                     <div class="editor-container">
                                         <div class="ai-toolbar">
-                                            <a class="ai-link expand" data-type="expand">AI扩写</a>
-                                            <a class="ai-link condense" data-type="condense">AI缩写</a>
-                                            <a class="ai-link continue" data-type="continue">AI续写</a>
-                                            <a class="ai-link polish" data-type="polish">AI润色</a>
+                                            <a class="ai-link expand" data-type="stream/expand">AI扩写</a>
+                                            <a class="ai-link condense" data-type="stream/condense">AI缩写</a>
+                                            <a class="ai-link continue" data-type="stream/continue">AI续写</a>
+                                            <a class="ai-link polish" data-type="stream/polish">AI润色</a>
                                         </div>
                                         <textarea id="bookContent" name="bookContent"
                                                   placeholder="请输入文本内容..."></textarea>
@@ -268,7 +268,7 @@
         }, speed);
     }
 
-    $('.ai-toolbar .ai-link').click(function(e){
+    $('.ai-toolbar .ai-link').click(function (e) {
         e.preventDefault(); // 阻止默认链接行为
         const type = $(this).data('type');
         const textarea = $('#bookContent');
@@ -284,27 +284,27 @@
 
         // 参数配置
         let params = {text: selectedText};
-        if(type === 'expand' || type === 'condense'){
+        if (type === 'stream/expand' || type === 'stream/condense') {
             layer.prompt({
                 title: '请输入比例',
                 value: 2,
                 btn: ['确定', '取消'],
-                btn2: function(){
+                btn2: function () {
                     layer.close(loading);
                 },
-                cancel: function(){
+                cancel: function () {
                     layer.close(loading);
                 }
-            }, function(value, index){
-                if(isNaN(Number(value)) || isNaN(parseFloat(value))){
+            }, function (value, index) {
+                if (isNaN(Number(value)) || isNaN(parseFloat(value))) {
                     layer.msg('请输入正确的比例');
                     return;
                 }
-                if(type === 'expand' && value <= 1){
+                if (type === 'stream/expand' && value <= 1) {
                     layer.msg('请输入正确的比例');
                     return;
                 }
-                if(type === 'condense' && (value <=0 || value >= 1)){
+                if (type === 'stream/condense' && (value <= 0 || value >= 1)) {
                     layer.msg('请输入正确的比例');
                     return;
                 }
@@ -313,19 +313,19 @@
                 sendRequest(type, params, loading, textarea);
             });
             return;
-        }else if(type === 'continue'){
+        } else if (type === 'stream/continue') {
             layer.prompt({
                 title: '请输入续写长度（字数）',
                 value: 200,
                 btn: ['确定', '取消'],
-                btn2: function(){
+                btn2: function () {
                     layer.close(loading);
                 },
-                cancel: function(){
+                cancel: function () {
                     layer.close(loading);
                 }
-            }, function(value, index){
-                if(!Number.isInteger(Number(value)) || value <= 0){
+            }, function (value, index) {
+                if (!Number.isInteger(Number(value)) || value <= 0) {
                     layer.msg('请输入正确的长度');
                     return;
                 }
@@ -339,26 +339,28 @@
         sendRequest(type, params, loading, textarea);
     });
 
-    function sendRequest(type, params, loading, textarea){
-        $.ajax({
-            url: '/author/ai/' + type,
-            type: 'POST',
-            data: params,
-            success: function(res){
-                layer.close(loading);
-                // 将生成的内容追加到文本末尾
-                if(res.code == '200'){
-                    const newText = "\n\n【AI生成内容】" + res.data; // 添加换行符分隔
-                    typeWriter(textarea, newText); // 使用打字机效果
-                }else{
-                    layer.msg('AI内容生成失败，请稍后重试');
-                }
-            },
-            error: function(){
-                layer.msg('请求失败，请稍后重试');
-                layer.close(loading);
-            }
-        });
+    function sendRequest(type, params, loading, textarea) {
+        const url = `/author/ai/${type}?text=${encodeURIComponent(params.text)}&ratio=${params.ratio}&length=${params.length}`;
+        const eventSource = new EventSource(url);
+
+        // 监听消息事件
+        eventSource.onmessage = function (event) {
+            layer.close(loading);
+            const data = event.data;
+            console.log('Received data:', data);
+
+            textarea.val(textarea.val() + data);
+            // 滚动到底部
+            textarea.scrollTop(textarea[0].scrollHeight);
+        };
+
+        // 监听错误事件
+        eventSource.onerror = function (error) {
+            layer.close(loading);
+            console.error('EventSource failed:', error);
+            eventSource.close(); // 关闭连接
+        };
+
     }
 
 </script>