CrazyAirhead

package com.goldsyear.ray.converter;

import static java.io.File.separator;

import com.jfinal.kit.Kv;
import java.io.File;
import java.nio.charset.StandardCharsets;
import java.util.*;
import java.util.regex.Matcher;
import java.util.stream.Collectors;
import lombok.extern.slf4j.Slf4j;
import org.dromara.hutool.core.collection.set.SetUtil;
import org.dromara.hutool.core.io.file.FileTypeUtil;
import org.dromara.hutool.core.io.file.FileUtil;

/**
 * SpringBoot 项目迁移 Solon 项目工具，并不能迁移完就直接启动，只是减少迁移工作量
 *
 * @author Airhead
 */
@Slf4j
public class RayConverterApp {

  /** 白名单文件，不进行重写，避免出问题 */
  private static final Set<String> WHITE_FILE_TYPES =
      SetUtil.of("gif", "jpg", "svg", "png", "eot", "woff2", "ttf", "woff");

  public static void main(String[] args) {
    long start = System.currentTimeMillis();

    // 修改路径
    String oldProjectDir = "旧项目路";
    String oldPackageName = "旧包名";
    String newProjectDir = "新项目路";
    String newPackageName = "新包名";

    log.info("原项目目录：{}, 原基础包名：{}", oldProjectDir, oldPackageName);
    log.info("新项目目录：{}, 新基础包名：{}", newProjectDir, newPackageName);

    // ========== 配置，需要你手动修改 ==========
    List<Kv> mappingList = new ArrayList<>();
    mappingList.add(Kv.by("old", oldPackageName).set("new", newPackageName));

    // Spring 包调整
    mappingSpringBoot(mappingList);

    // Hutool 包调整
    mappingHutool(mappingList);

    // FastJson
    mappingFastJson(mappingList);

    // Mybatis 包调整
    mappingMybatis(mappingList);

    // TODO: 根据自己依赖的项目模块进行替换

    // 业务包调整
    mappingBusiness(mappingList);

    // 获得需要复制的文件
    log.info("开始获得需要重写的文件，预计需要 10-20 秒");
    Collection<File> files = listFiles(oldProjectDir);
    log.info("需要重写的文件数量：{}，预计需要 15-30 秒", files.size());

    // 写入文件
    files.forEach(
        file -> {
          // 如果是白名单的文件类型，不进行重写，直接拷贝
          String fileType = FileTypeUtil.getType(file);
          if (WHITE_FILE_TYPES.contains(fileType)) {
            copyFile(file, oldProjectDir, oldPackageName, newProjectDir, newPackageName);
            return;
          }

          // 如果非白名单的文件类型，重写内容，在生成文件
          String content = replaceFileContent(file, mappingList);

          writeFile(file, content, oldProjectDir, oldPackageName, newProjectDir, newPackageName);
        });

    log.info("重写完成，共耗时：{} 秒", (System.currentTimeMillis() - start) / 1000);
  }

  private static void mappingBusiness(List<Kv> mappingList) {
    // TODO
  }

  private static void mappingFastJson(List<Kv> mappingList) {
    // TODO
  }

  private static void mappingMybatis(List<Kv> mappingList) {
    // Page
    mappingList.add(
        Kv.by("old", "com.baomidou.mybatisplus.extension.plugins.pagination.Page")
            .set("new", "com.baomidou.mybatisplus.solon.plugins.pagination.Page"));
    // Model
    mappingList.add(
        Kv.by("old", "com.baomidou.mybatisplus.extension.activerecord.Model")
            .set("new", "com.baomidou.mybatisplus.solon.activerecord.Model"));

    // TenantLineHandler
    mappingList.add(
        Kv.by("old", "com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler")
            .set("new", "com.baomidou.mybatisplus.solon.plugins.handler.TenantLineHandler"));
    // MybatisPlusInterceptor
    mappingList.add(
        Kv.by("old", "com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor")
            .set("new", "com.baomidou.mybatisplus.solon.plugins.MybatisPlusInterceptor"));
    // PaginationInnerInterceptor
    mappingList.add(
        Kv.by("old", "com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor")
            .set("new", "com.baomidou.mybatisplus.solon.plugins.inner.PaginationInnerInterceptor"));
    // TenantLineInnerInterceptor
    mappingList.add(
        Kv.by("old", "com.baomidou.mybatisplus.extension.plugins.inner.TenantLineInnerInterceptor")
            .set("new", "com.baomidou.mybatisplus.solon.plugins.inner.TenantLineInnerInterceptor"));

    // FastjsonTypeHandler
    mappingList.add(
        Kv.by("old", "com.baomidou.mybatisplus.extension.handlers.FastjsonTypeHandler")
            .set("new", "com.baomidou.mybatisplus.solon.handlers.FastjsonTypeHandler"));
  }

  private static void mappingHutool(List<Kv> mappingList) {
    // StrUtil;
    mappingList.add(
        Kv.by("old", "cn.hutool.core.util.StrUtil")
            .set("new", "org.dromara.hutool.core.text.StrUtil"));

    // Convert;
    mappingList.add(
        Kv.by("old", "cn.hutool.core.convert.Convert")
            .set("new", "org.dromara.hutool.core.convert.ConvertUtil"));
    mappingList.add(Kv.by("old", "Convert\\.").set("new", "ConvertUtil."));

    // IdUtil;
    mappingList.add(
        Kv.by("old", "cn.hutool.core.util.IdUtil")
            .set("new", "org.dromara.hutool.core.data.id.IdUtil"));

    // DateUtil;
    mappingList.add(
        Kv.by("old", "cn.hutool.core.date.DateUtil")
            .set("new", "org.dromara.hutool.core.date.DateUtil"));

    // ReUtil
    mappingList.add(
        Kv.by("old", "cn.hutool.core.util.ReUtil")
            .set("new", "org.dromara.hutool.core.regex.ReUtil"));

    // Spring CollectionUtils;
    mappingList.add(
        Kv.by("old", "org.springframework.util.CollectionUtils")
            .set("new", "org.dromara.hutool.core.collection.CollUtil"));
    mappingList.add(Kv.by("old", "CollectionUtils").set("new", "CollUtil"));

    // CollectionUtil;
    mappingList.add(
        Kv.by("old", "cn.hutool.core.collection.CollectionUtil")
            .set("new", "org.dromara.hutool.core.collection.CollUtil"));
    mappingList.add(Kv.by("old", "CollectionUtil").set("new", "CollUtil"));

    // LocalDateTimeUtil
    mappingList.add(
        Kv.by("old", "cn.hutool.core.date.LocalDateTimeUtil")
            .set("new", "org.dromara.hutool.core.date.TimeUtil"));
    mappingList.add(Kv.by("old", "LocalDateTimeUtil").set("new", "TimeUtil"));

    // FileUtil
    mappingList.add(
        Kv.by("old", "cn.hutool.core.io.FileUtil")
            .set("new", "org.dromara.hutool.core.io.file.FileUtil"));

    // IoUtil
    mappingList.add(
        Kv.by("old", "cn.hutool.core.io.IoUtil").set("new", "org.dromara.hutool.core.io.IoUtil"));
  }

  private static void mappingSpringBoot(List<Kv> mappingList) {
    // annotation
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation\\.\\*")
            .set("new", "org.noear.solon.annotation.*"));

    // Component
    mappingList.add(
        Kv.by("old", "org.springframework.stereotype.Component")
            .set("new", "org.noear.solon.annotation.Component"));

    // Service
    mappingList.add(
        Kv.by("old", "org.springframework.stereotype.Service")
            .set("new", "org.noear.solon.annotation.Component"));
    mappingList.add(Kv.by("old", "@Service").set("new", "@Component"));

    // Autowired
    mappingList.add(
        Kv.by("old", "org.springframework.beans.factory.annotation.Autowired")
            .set("new", "org.noear.solon.annotation.Inject"));
    mappingList.add(Kv.by("old", "@Autowired").set("new", "@Inject"));

    // Controller
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.RestController")
            .set("new", "org.noear.solon.annotation.Controller"));
    mappingList.add(Kv.by("old", "@RestController").set("new", "@Controller"));

    // RequestMapping
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.RequestMapping")
            .set("new", "org.noear.solon.annotation.Mapping"));
    mappingList.add(Kv.by("old", "@RequestMapping").set("new", "@Mapping"));

    // MediaType;
    mappingList.add(Kv.by("old", "import org.springframework.http.MediaType;").set("new", ""));
    mappingList.add(
        Kv.by("old", ", produces = MediaType.APPLICATION_JSON_UTF8_VALUE").set("new", ""));

    mappingList.add(
        Kv.by("old", "import org.springframework.core.env.Environment;").set("new", ""));

    // PostMapping
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.PostMapping")
            .set("new", "org.noear.solon.annotation.Post"));
    mappingList.add(Kv.by("old", "@PostMapping").set("new", "@Post\n@Mapping"));

    // GetMapping
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.GetMapping")
            .set("new", "org.noear.solon.annotation.Get"));
    mappingList.add(Kv.by("old", "@GetMapping").set("new", "@Post\n@Mapping"));

    // PutMapping
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.PutMapping")
            .set("new", "org.noear.solon.annotation.Put"));
    mappingList.add(Kv.by("old", "@PutMapping").set("new", "@Put\n@Mapping"));

    // DeleteMapping
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.DeleteMapping")
            .set("new", "org.noear.solon.annotation.Delete"));
    mappingList.add(Kv.by("old", "@DeleteMapping").set("new", "@Delete\n@Mapping"));

    // @RequestBody
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.RequestBody")
            .set("new", "org.noear.solon.annotation.Body"));
    mappingList.add(Kv.by("old", "@RequestBody").set("new", "@Body"));

    // @RequestParam
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.RequestParam")
            .set("new", "org.noear.solon.annotation.Param"));
    mappingList.add(Kv.by("old", "@RequestParam").set("new", "@Param"));

    //    RequestPart
    mappingList.add(
        Kv.by("old", "org.springframework.web.bind.annotation.RequestPart")
            .set("new", "org.noear.solon.annotation.Param"));
    mappingList.add(Kv.by("old", "@RequestPart").set("new", "@Param"));

    // Repository;
    mappingList.add(
        Kv.by("old", "import org.springframework.stereotype.Repository;").set("new", ""));
    mappingList.add(Kv.by("old", "@Repository").set("new", ""));

    // Validated
    mappingList.add(
        Kv.by("old", "org.springframework.validation.annotation.Validated")
            .set("new", "org.noear.solon.validation.annotation.Validated"));
    mappingList.add(Kv.by("old", "@Validated\n").set("new", ""));

    // Configuration;
    mappingList.add(
        Kv.by("old", "org.springframework.context.annotation.Configuration")
            .set("new", "org.noear.solon.annotation.Configuration"));

    // RefreshScope;
    mappingList.add(
        Kv.by("old", "import org.springframework.cloud.context.config.annotation.RefreshScope;")
            .set("new", ""));
    mappingList.add(Kv.by("old", "@RefreshScope\n").set("new", ""));

    // EnableAsync;
    mappingList.add(
        Kv.by("old", "import org.springframework.scheduling.annotation.EnableAsync;")
            .set("new", ""));
    mappingList.add(Kv.by("old", "@EnableAsync").set("new", ""));

    // Bean;
    mappingList.add(
        Kv.by("old", "org.springframework.context.annotation.Bean")
            .set("new", "org.noear.solon.annotation.Bean"));

    //  ConfigurationProperties;
    mappingList.add(
        Kv.by("old", "org.springframework.boot.context.properties.ConfigurationProperties")
            .set("new", "org.noear.solon.annotation.Inject"));
    mappingList.add(Kv.by("old", "@ConfigurationProperties").set("new", "@Inject"));

    // CommandLineRunner
    mappingList.add(
        Kv.by("old", "import org.springframework.boot.CommandLineRunner;")
            .set(
                "new",
                "import org.noear.solon.core.event.AppLoadEndEvent;\nimport org.noear.solon.core.event.EventListener;"));
    mappingList.add(Kv.by("old", "CommandLineRunner").set("new", "EventListener<AppLoadEndEvent>"));

    // Resource;
    mappingList.add(
        Kv.by("old", "javax.annotation.Resource").set("new", "org.noear.solon.annotation.Inject"));
    mappingList.add(Kv.by("old", "@Resource\\(name").set("new", "@Inject\\(value"));

    // Lazy;
    mappingList.add(
        Kv.by("old", "import org.springframework.context.annotation.Lazy;\n").set("new", ""));
    mappingList.add(Kv.by("old", "@Lazy").set("new", ""));

    // Valid
    mappingList.add(
        Kv.by("old", "javax.validation.Valid")
            .set("new", "org.noear.solon.validation.annotation.Valid"));
    mappingList.add(
        Kv.by("old", "javax.validation.constraints.NotEmpty")
            .set("new", "org.noear.solon.validation.annotation.NotEmpty"));
    mappingList.add(
        Kv.by("old", "javax.validation.constraints.NotNull")
            .set("new", "org.noear.solon.validation.annotation.NotNull"));
    // TODO: 补充更多

    // HttpServletResponse;
    mappingList.add(
        Kv.by("old", "javax.servlet.http.HttpServletResponse")
            .set("new", "org.noear.solon.core.handle.Context"));
    mappingList.add(Kv.by("old", "HttpServletResponse").set("new", "Context"));

    // MultipartFile;
    mappingList.add(
        Kv.by("old", "org.springframework.web.multipart.MultipartFile")
            .set("new", "org.noear.solon.core.handle.UploadedFile"));
    mappingList.add(Kv.by("old", "MultipartFile").set("new", "UploadedFile"));

    //    consumes = {MediaType.MULTIPART_FORM_DATA_VALUE}
    mappingList.add(
        Kv.by("old", ",\n\\s+consumes = \\{MediaType.MULTIPART_FORM_DATA_VALUE\\}").set("new", ""));
    mappingList.add(
        Kv.by("old", ", consumes = \\{MediaType.MULTIPART_FORM_DATA_VALUE\\}").set("new", ""));
  }

  private static Collection<File> listFiles(String projectBaseDir) {
    Collection<File> files = FileUtil.loopFiles(new File(projectBaseDir));
    // 移除 IDEA、Git 自身的文件、Node 编译出来的文件
    files =
        files.stream()
            .filter(
                file ->
                    !file.getPath().contains(separator + "target" + separator)
                        && !file.getPath().contains(separator + "node_modules" + separator)
                        && !file.getPath().contains(separator + ".idea" + separator)
                        && !file.getPath().contains(separator + ".git" + separator)
                        && !file.getPath().contains(separator + "dist" + separator)
                        && !file.getPath().contains(separator + "build" + separator)
                        && !file.getPath().contains(separator + "out" + separator)
                        && !file.getPath().contains(".iml")
                        && !file.getPath().contains(".html.gz")
                        && !file.getPath().contains("build.gradle")
                        && !file.getPath().contains("pom.xml"))
            .collect(Collectors.toList());
    return files;
  }

  private static String replaceFileContent(File file, List<Kv> mappingList) {
    String content = FileUtil.readString(file, StandardCharsets.UTF_8);
    // 如果是白名单的文件类型，不进行重写
    String fileType = FileTypeUtil.getType(file);
    if (WHITE_FILE_TYPES.contains(fileType)) {
      return content;
    }

    for (Kv kv : mappingList) {
      content = content.replaceAll(kv.getStr("old"), kv.getStr("new"));
    }

    return content;
  }

  private static void writeFile(
      File file,
      String fileContent,
      String oldProjectDir,
      String oldPageName,
      String newProjectDir,
      String newPackageName) {
    String newPath =
        buildNewFilePath(file, oldProjectDir, oldPageName, newProjectDir, newPackageName);
    FileUtil.writeUtf8String(fileContent, newPath);
  }

  private static void copyFile(
      File file,
      String oldProjectDir,
      String oldPackageName,
      String newProjectDir,
      String newPackageName) {
    String newPath =
        buildNewFilePath(file, oldProjectDir, oldPackageName, newProjectDir, newPackageName);
    FileUtil.copy(file, new File(newPath), true);
  }

  private static String buildNewFilePath(
      File file,
      String oldProjectDir,
      String oldPackageName,
      String newProjectDir,
      String newPackageName) {
    return file.getPath()
        .replace(oldProjectDir, newProjectDir)
        .replace(
            oldPackageName.replaceAll("\\.", Matcher.quoteReplacement(separator)),
            newPackageName.replaceAll("\\.", Matcher.quoteReplacement(separator)));
  }
}

docker run -d --name redis-stack -p 6379:6379 -p 8001:8001 swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/redis/redis-stack:latest

ollama run deepseek-r1:32b
ollama run qwen2.5:7b
ollama run nomic-embed-text

package com.example.demo.ai.llm.service;

import com.jfinal.kit.Kv;
import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject;
import org.noear.solon.flow.ChainContext;
import org.noear.solon.flow.FlowEngine;

/**
 * @author airhead
 */
@Component
public class LlmDiagService {
  @Inject private FlowEngine flowEngine;

  public Kv diag(String prompt) {
    try {
      ChainContext chainContext = new ChainContext();
      chainContext.put("prompt", prompt);
      Kv kv = Kv.create();
      chainContext.put("result", kv);
      flowEngine.eval("ai-flow-01", chainContext);

      return kv;
    } catch (Throwable e) {
      throw new RuntimeException(e);
    }
  }
}

package com.example.demo.ai.llm.service;

import com.jfinal.kit.Kv;
import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject;
import org.noear.solon.flow.ChainContext;
import org.noear.solon.flow.FlowEngine;

/**
 * @author airhead
 */
@Component
public class LlmService {
  @Inject private FlowEngine flowEngine;

  public Kv flow(String prompt) {

    try {
      ChainContext chainContext = new ChainContext();
      chainContext.put("prompt", prompt);
      Kv kv = Kv.create();
      chainContext.put("result", kv);
      flowEngine.eval("c1", chainContext);

      return kv;
    } catch (Throwable e) {
      throw new RuntimeException(e);
    }
  }

  public Kv aiFlow(String prompt) {
    try {
      ChainContext chainContext = new ChainContext();
      chainContext.put("prompt", prompt);
      Kv kv = Kv.create();
      chainContext.put("result", kv);
      flowEngine.eval("ai-flow-01", chainContext);

      return kv;
    } catch (Throwable e) {
      throw new RuntimeException(e);
    }
  }
}

package com.example.demo.ai.llm.service;

import com.jfinal.kit.Kv;
import java.time.Duration;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Pattern;

import org.dromara.hutool.core.regex.ReUtil;
import org.dromara.hutool.core.text.StrUtil;
import org.noear.solon.ai.chat.ChatConfig;
import org.noear.solon.ai.chat.ChatModel;
import org.noear.solon.ai.chat.ChatResponse;
import org.noear.solon.ai.chat.message.ChatMessage;
import org.noear.solon.annotation.Component;
import org.noear.solon.flow.ChainContext;
import org.noear.solon.flow.Node;
import org.noear.solon.flow.TaskComponent;

/**
 * @author airhead
 */
@Component("intentionTask")
public class IntentionTask implements TaskComponent {
  @Override
  public void run(ChainContext context, Node node) throws Throwable {
    Kv meta = Kv.create().set(node.meta());
    ChatConfig chatConfig = new ChatConfig();
    chatConfig.setModel(meta.getStr("model"));
    chatConfig.setProvider(meta.getStr("provider"));
    chatConfig.setApiUrl(meta.getStr("apiUrl"));
    chatConfig.setTimeout(Duration.ofSeconds(600));
    ChatModel chatModel = ChatModel.of(chatConfig).build();

    List<ChatMessage> chatMessageList = new ArrayList<>();
    ChatMessage system = ChatMessage.ofSystem(meta.getStr("system"));
    chatMessageList.add(system);

    Kv model = Kv.create().set(context.model());
    String inputKey = meta.getStr("input");
    ChatMessage userMessage = ChatMessage.ofUser(model.getStr(inputKey));
    chatMessageList.add(userMessage);
    ChatResponse response = chatModel.prompt(chatMessageList).call();
    String content = response.getMessage().getContent();

    // 去掉think的部分。
    String pattern = "<think>.*?</think>";
    Pattern p = Pattern.compile(pattern, Pattern.DOTALL);
    content = StrUtil.trim(ReUtil.replaceAll(content, p, ""));

    String outputKey = meta.getStr("output");
    context.put(outputKey, content);

    Kv result = context.get("result");
    result.set("data", content);
  }
}

package com.example.demo.ai.llm.service;

import com.jfinal.kit.Kv;
import java.util.ArrayList;
import java.util.List;
import org.noear.solon.ai.chat.ChatConfig;
import org.noear.solon.ai.chat.ChatModel;
import org.noear.solon.ai.chat.ChatResponse;
import org.noear.solon.ai.chat.message.ChatMessage;
import org.noear.solon.annotation.Component;
import org.noear.solon.flow.ChainContext;
import org.noear.solon.flow.Node;
import org.noear.solon.flow.TaskComponent;

/**
 * @author airhead
 */
@Component("suggestionTask")
public class SuggestionTask implements TaskComponent {
  @Override
  public void run(ChainContext context, Node node) throws Throwable {
    Kv meta = Kv.create().set(node.meta());
    ChatConfig chatConfig = new ChatConfig();
    chatConfig.setModel(meta.getStr("model"));
    chatConfig.setProvider(meta.getStr("provider"));
    chatConfig.setApiUrl(meta.getStr("apiUrl"));
    ChatModel chatModel = ChatModel.of(chatConfig).build();

    List<ChatMessage> chatMessageList = new ArrayList<>();
    ChatMessage system = ChatMessage.ofSystem(meta.getStr("system"));
    chatMessageList.add(system);

    Kv model = Kv.create().set(context.model());
    String inputKey = meta.getStr("input");
    ChatMessage userMessage = ChatMessage.ofUser(model.getStr(inputKey));
    chatMessageList.add(userMessage);
    ChatResponse response = chatModel.prompt(chatMessageList).call();
    String content = response.getMessage().getContent();

    String outputKey = meta.getStr("output");
    context.put(outputKey, content);

    Kv result = context.get("result");
    result.set("data", content);
  }
}

package com.example.demo.solon.controller;

import static org.junit.jupiter.api.Assertions.*;

import com.example.demo.solon.DemoCloudBreakerApp;
import java.util.concurrent.CountDownLatch;
import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test;
import org.noear.solon.Utils;
import org.noear.solon.test.HttpTester;
import org.noear.solon.test.SolonTest;

@Slf4j
@SolonTest(DemoCloudBreakerApp.class)
public class DemoControllerTest extends HttpTester {
  @Test
  public void test() throws Exception {
    CountDownLatch count = new CountDownLatch(100);
    for (int i = 0; i < 100; i++) {
      log.info("get hello start. i:{}", i);
      int finalI = i;
      Utils.async(
          () -> {
            try {
              path("/hello").get();
            } catch (Exception e) {
              log.error("", e);
            }

            log.info("get hello end. i:{}", finalI);
            count.countDown();
          });
    }

    count.await();
  }
}

dependencies {
    implementation platform(project(":demo-parent"))

    implementation("org.noear:solon-web")
    implementation("org.noear:solon-ai")
    implementation("org.noear:solon-logging-logback")
    implementation("org.noear:solon-openapi2-knife4j")
    implementation("org.noear:solon-web-rx")
    implementation("org.noear:solon-web-sse")

    testImplementation("org.noear:solon-test")
}

package com.example.demo.ai.llm.config;

import org.noear.solon.ai.chat.ChatConfig;
import org.noear.solon.ai.chat.ChatModel;
import org.noear.solon.annotation.Bean;
import org.noear.solon.annotation.Configuration;
import org.noear.solon.annotation.Inject;

/**
 * @author airhead
 */
@Configuration
public class LlmConfig {
  @Bean
  public ChatModel build(@Inject("${demo.llm}") ChatConfig config) {
    return ChatModel.of(config).build();
  }
}

package com.example.demo.ai.llm.controller;

import com.example.demo.ai.llm.service.LlmService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import java.io.IOException;
import org.noear.solon.annotation.*;
import org.noear.solon.core.util.MimeType;
import reactor.core.publisher.Flux;

/**
 * @author airhead
 */
@Controller
@Mapping("/llm")
@Api("聊天")
public class LlmController {
  @Inject private LlmService service;

  @ApiOperation("chat")
  @Post
  @Mapping("chat")
  public String chat(String prompt) {
    return service.chat(prompt);
  }

  @Produces(MimeType.TEXT_EVENT_STREAM_UTF8_VALUE)
  @Mapping("stream")
  public Flux<String> stream(String prompt) throws IOException {
    return service.stream(prompt);
  }

  @ApiOperation("functionCall")
  @Post
  @Mapping("functionCall")
  public String functionCall(String prompt) {
    return service.functionCall(prompt);
  }
}

package com.example.demo.ai.llm.service;

import java.io.IOException;
import org.noear.solon.ai.chat.ChatModel;
import org.noear.solon.ai.chat.ChatResponse;
import org.noear.solon.annotation.Component;
import org.noear.solon.annotation.Inject;
import reactor.core.publisher.Flux;

/**
 * @author airhead
 */
@Component
public class LlmService {
  @Inject private ChatModel chatModel;

  public String chat(String prompt) {
    try {
      ChatResponse response = chatModel.prompt(prompt).call();

      return response.getMessage().getContent();
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }

  public Flux<String> stream(String prompt) throws IOException {
    return Flux.from(chatModel.prompt(prompt).stream())
        .filter(ChatResponse::hasChoices)
        .map(resp -> resp.getMessage().getContent());
  }

  public String functionCall(String prompt) {
    try {
      ChatResponse response =
          chatModel.prompt(prompt).options(o -> o.functionAdd(new Tools())).call();

      return response.getMessage().getContent();
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }
}

package com.example.demo.solon.filter;

import io.vertx.core.buffer.Buffer;
import lombok.extern.slf4j.Slf4j;
import org.dromara.hutool.core.text.StrUtil;
import org.noear.solon.annotation.Component;
import org.noear.solon.cloud.gateway.exchange.ExContext;
import org.noear.solon.cloud.gateway.exchange.ExFilter;
import org.noear.solon.cloud.gateway.exchange.ExFilterChain;
import org.noear.solon.cloud.gateway.route.RouteFilterFactory;
import org.noear.solon.rx.Completable;

/**
 * @author airhead
 */
@Component
@Slf4j
public class AuthFilterFactory implements RouteFilterFactory {
  public static final String TOKEN = "token";
  public static final String EMPTY = "empty";

  @Override
  public String prefix() {
    return "Auth";
  }

  @Override
  public ExFilter create(String config) {
    if (TOKEN.equals(config)) {
      return new TokenFilter();
    } else {
      return new EmptyAuthFilter();
    }
  }

  /** 默认的空鉴权 */
  public static class EmptyAuthFilter implements ExFilter {
    public EmptyAuthFilter() {}

    @Override
    public Completable doFilter(ExContext ctx, ExFilterChain chain) {
      log.info("empty filter, do nothing");

      return chain.doFilter(ctx);
    }
  }

  /** token 鉴权 ，如果鉴权逻辑比较复杂的，用独立类 */
  public static class TokenFilter implements ExFilter {
    @Override
    public Completable doFilter(ExContext ctx, ExFilterChain chain) {
      String token = ctx.rawHeader("token");
      if (StrUtil.isBlank(token)) {
        ctx.newResponse().status(401);
        ctx.newResponse().body(Buffer.buffer("token is empty"));
        return Completable.complete();
      }

      return chain.doFilter(ctx);
    }
  }
}

 package com.example.demo.solon.controller;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.noear.solon.annotation.Controller;
import org.noear.solon.annotation.Mapping;
import org.noear.solon.annotation.Param;

/**
 * @author airhead
 */
@Api("Demo")
@Controller
@Mapping
public class DemoController {
  @ApiOperation("hello")
  @Mapping("/hello")
  public String hello(@Param(defaultValue = "world") String name) {
    return String.format("Hello %s!, By Service01", name);
  }
}

package com.example.demo.solon.controller;

import com.example.demo.solon.client.DemoClient;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import java.io.IOException;
import org.noear.nami.annotation.NamiClient;
import org.noear.solon.annotation.Controller;
import org.noear.solon.annotation.Mapping;
import org.noear.solon.annotation.Param;
import org.noear.solon.net.http.HttpUtils;

/**
 * @author airhead
 */
@Api("Demo")
@Controller
@Mapping
public class DemoController {
  @NamiClient private DemoClient demoClient;

  @ApiOperation("hello")
  @Mapping("/hello")
  public String hello(@Param(defaultValue = "world") String name) {
    try {
      return HttpUtils.http("demo-cloud-service01", "/hello").data("name", name).get();
    } catch (IOException e) {
      throw new RuntimeException(e);
    }
  }

  @ApiOperation("hello2")
  @Mapping("/hello2")
  public String hello2(@Param(defaultValue = "world") String name) {
    return demoClient.hello(name);
  }
}

package com.example.demo.solon.client;

import org.noear.nami.annotation.NamiClient;

/**
 * @author airhead
 */
@NamiClient(name = "demo-cloud-service01")
public interface DemoClient {
  String hello(String name);
}