应用管理

package nexus.io.mosskb.controller;

import java.time.LocalDate;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import nexus.io.annotation.Delete;
import nexus.io.annotation.Get;
import nexus.io.annotation.Post;
import nexus.io.annotation.Put;
import nexus.io.annotation.RequestPath;
import nexus.io.db.TableInput;
import nexus.io.jfinal.aop.Aop;
import nexus.io.mosskb.service.MossKbApplicationAccessTokenService;
import nexus.io.mosskb.service.MossKbApplicationService;
import nexus.io.mosskb.vo.MossKbApplicationVo;
import nexus.io.model.result.ResultVo;
import nexus.io.tio.boot.http.TioRequestContext;
import nexus.io.tio.http.common.HttpRequest;
import nexus.io.tio.utils.json.JsonUtils;

@RequestPath("/api/application")
public class ApiApplicationController {

  @Post
  public ResultVo create(MossKbApplicationVo application) {
    Long userId = TioRequestContext.getUserIdLong();
    return Aop.get(MossKbApplicationService.class).create(userId, application);
  }

  @Put("/{applicationId}")
  public ResultVo update(Long applicationId, HttpRequest request) {
    String bodyString = request.getBodyString();
    MossKbApplicationVo application = JsonUtils.parse(bodyString, MossKbApplicationVo.class);
    Long userId = TioRequestContext.getUserIdLong();
    application.setId(applicationId);
    return Aop.get(MossKbApplicationService.class).update(userId, application);
  }

  @Delete("/{applicationId}")
  public ResultVo delete(Long applicationId) {
    Long userId = TioRequestContext.getUserIdLong();
    return Aop.get(MossKbApplicationService.class).delete(userId, applicationId);
  }

  @Get("/{applicationId}")
  public ResultVo getById(Long applicationId) {
    Long userId = TioRequestContext.getUserIdLong();
    return Aop.get(MossKbApplicationService.class).get(userId, applicationId);
  }

  @Get
  public ResultVo list() {
    Long userId = TioRequestContext.getUserIdLong();
    return Aop.get(MossKbApplicationService.class).list(userId);
  }

  @Get("/{pageNo}/{pageSize}")
  public ResultVo page(Integer pageNo, Integer pageSize, String name) {
    TableInput tableInput = new TableInput();
    tableInput.setPageNo(pageNo).setPageSize(pageSize);
    tableInput.setSearchKey(name);
    return Aop.get(MossKbApplicationService.class).page(tableInput);
  }

  @Get("/{applicationId}/access_token")
  public ResultVo getAccessToken(Long applicationId) {
    return Aop.get(MossKbApplicationAccessTokenService.class).getById(applicationId);
  }

  @Get("/{applicationId}/statistics/chat_record_aggregate_trend")
  public ResultVo statisticsOfCharRecordAggreagteTrend(Long applicationId, String start_time, String end_time) {
    LocalDate startDate = LocalDate.parse(start_time);
    LocalDate endDate = LocalDate.parse(end_time);
    List<Map<String, Object>> dataList = new ArrayList<>();
    LocalDate currentDate = startDate;
    while (!currentDate.isAfter(endDate)) {
      Map<String, Object> dailyData = new HashMap<>();
      dailyData.put("star_num", 0);
      dailyData.put("trample_num", 0);
      dailyData.put("tokens_num", 0);
      dailyData.put("chat_record_count", 0);
      dailyData.put("customer_num", 0);
      dailyData.put("day", currentDate.toString());
      dailyData.put("customer_added_count", 0);
      dataList.add(dailyData);
      currentDate = currentDate.plusDays(1);
    }
    return ResultVo.ok(dataList);
  }

  @Get("/{applicationId}/model")
  public ResultVo listApplicaionModel(Long applicationId) {
    Long userId = TioRequestContext.getUserIdLong();
    return Aop.get(MossKbApplicationService.class).listApplicaionModel(userId, applicationId);
  }

  @Get("/{applicationId}/model_params_form/{modelId}")
  public ResultVo setModelId(Long applicationId, Long modelId) {
    Long userIdLong = TioRequestContext.getUserIdLong();
    return Aop.get(MossKbApplicationService.class).setModelId(userIdLong, applicationId, modelId);
  }

  @Get("/{applicationId}/list_dataset")
  public ResultVo listApplicaionDataset(Long applicationId) {
    Long userId = TioRequestContext.getUserIdLong();
    return Aop.get(MossKbApplicationService.class).listApplicaionDataset(userId, applicationId);
  }
}

package nexus.io.mosskb.service;

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

import com.jfinal.kit.Kv;
import nexus.io.db.TableInput;
import nexus.io.db.TableResult;
import nexus.io.db.activerecord.Db;
import nexus.io.db.activerecord.Row;
import nexus.io.jfinal.aop.Aop;
import nexus.io.kit.PGJsonUtils;
import nexus.io.mosskb.constant.TableNames;
import nexus.io.mosskb.model.MossKbApplication;
import nexus.io.mosskb.model.MossKbApplicationDatasetMapping;
import nexus.io.mosskb.model.MossKbModel;
import nexus.io.mosskb.vo.MossKbApplicationVo;
import nexus.io.mosskb.vo.MossKbDatasetSettingVo;
import nexus.io.mosskb.vo.MossKbModelParamsSetting;
import nexus.io.mosskb.vo.MossKbModelSetting;
import nexus.io.model.page.Page;
import nexus.io.model.result.ResultVo;
import nexus.io.table.services.ApiTable;
import nexus.io.tio.utils.snowflake.SnowflakeIdUtils;

import lombok.extern.slf4j.Slf4j;

@Slf4j
public class MossKbApplicationService {

  public ResultVo create(Long userId, MossKbApplicationVo application) {
    Long applicationId = SnowflakeIdUtils.id();
    application.setId(applicationId);
    Row row = Row.fromBean(application);
    row.remove("dataset_id_list");
    row.set("user_id", userId);
    Db.save(MossKbApplication.tableName, row);
    Aop.get(MossKbApplicationAccessTokenService.class).create(applicationId);
    return ResultVo.ok(application);
  }

  public ResultVo update(Long userId, MossKbApplicationVo application) {
    List<Long> dataset_id_list = application.getDataset_id_list();
    List<MossKbApplicationDatasetMapping> saveRecords = new ArrayList<>();
    if (dataset_id_list != null) {
      for (Long datasetId : dataset_id_list) {
        MossKbApplicationDatasetMapping mapping = new MossKbApplicationDatasetMapping();
        mapping.setId(SnowflakeIdUtils.id()).setDatasetId(datasetId).setApplicationId(application.getId());
        saveRecords.add(mapping);
      }
    }
    Row row = Row.fromBean(application);
    row.remove("dataset_id_list");
    row.set("user_id", userId);
    Db.tx(() -> {
      if (saveRecords.size() > 0) {
        Db.batchSave(saveRecords, 2000);
      }
      Db.update(MossKbApplication.tableName, row);
      return true;
    });
    return ResultVo.ok();
  }

  public ResultVo delete(Long userId, Long applicationId) {
    boolean deleted = new MossKbApplication().setId(applicationId).setUserId(userId).delete();
    Aop.get(MossKbApplicationAccessTokenService.class).delete(userId, applicationId);
    return ResultVo.ok(deleted);
  }

  public ResultVo page(TableInput tableInput) {
    Integer pageNo = tableInput.getPageNo();
    Integer pageSize = tableInput.getPageSize();
    log.info("page:{},{}", pageNo, pageSize);
    tableInput.setFrom(TableNames.moss_kb_application);
    TableResult<Page<Row>> result = ApiTable.page(tableInput);
    Page<Row> page = result.getData();
    List<Row> records = page.getList();
    List<Kv> kvs = new ArrayList<>();
    for (Row row : records) {
      PGJsonUtils.toBean(row, "model_setting", MossKbModelSetting.class);
      PGJsonUtils.toBean(row, "model_params_setting", MossKbModelParamsSetting.class);
      PGJsonUtils.toBean(row, "dataset_setting", MossKbDatasetSettingVo.class);
      kvs.add(row.toKv());
    }
    Kv kv = Kv.by("current", pageNo).set("size", pageSize).set("total", page.getTotalRow()).set("records", kvs);
    return ResultVo.ok(kv);
  }

  public ResultVo list(Long userId) {
    Row quereyRecord = new Row();
    quereyRecord.set("user_id", userId);
    List<Row> records = Db.find(TableNames.moss_kb_application, quereyRecord);
    List<Kv> kvs = new ArrayList<>();
    for (Row row : records) {
      PGJsonUtils.toBean(row, "model_params_setting", MossKbModelParamsSetting.class);
      PGJsonUtils.toBean(row, "model_setting", MossKbModelSetting.class);
      PGJsonUtils.toBean(row, "dataset_setting", MossKbDatasetSettingVo.class);
      Kv kv = row.toKv();
      kvs.add(kv);
    }
    return ResultVo.ok(kvs);
  }

  public ResultVo get(Long userId, Long applicationId) {
    Row quereyRecord = new MossKbApplication().setId(applicationId).setUserId(userId).toRecord();
    Row row = Db.findFirst(MossKbApplication.tableName, quereyRecord);
    PGJsonUtils.toBean(row, "model_params_setting", MossKbModelParamsSetting.class);
    PGJsonUtils.toBean(row, "model_setting", MossKbModelSetting.class);
    PGJsonUtils.toBean(row, "dataset_setting", MossKbDatasetSettingVo.class);
    Long modelId = row.getLong("model_id");
    row.remove("model_id");
    String sql = String.format("select dataset_id from %s where application_id=?", MossKbApplicationDatasetMapping.tableName);
    List<Long> dataset_id_list = Db.queryListLong(sql, applicationId);
    row.set("dataset_id_list", dataset_id_list);
    row.set("model", modelId);
    Kv kv = row.toKv();
    return ResultVo.ok(kv);
  }

  public ResultVo listApplicaionModel(Long userId, Long applicationId) {
    Row queryRecord = Row.by("user_id", userId);
    String columns = "id,name,provider,model_type,model_name,status,meta,permission_type,user_id";
    List<Row> list = Db.find(MossKbModel.tableName, columns, queryRecord);
    List<Kv> kvs = new ArrayList<>();
    MossKbUserService mossKbUserService = Aop.get(MossKbUserService.class);
    for (Row row : list) {
      Long user_id = row.getLong("user_id");
      String username = mossKbUserService.queryUsername(user_id);
      row.set("username", username);
      kvs.add(row.toKv());
    }
    return ResultVo.ok(kvs);
  }

  public ResultVo setModelId(Long userIdLong, Long applicationId, Long modelId) {
    Row updateRecord = Row.by("id", modelId).set("id", applicationId).set("user_id", userIdLong).set("model_id", modelId);
    boolean update = Db.update(MossKbApplication.tableName, "id,user_id", updateRecord);
    return ResultVo.ok(update);
  }

  public ResultVo listApplicaionDataset(Long userId, Long applicationId) {
    return Aop.get(MossKbDatasetService.class).list(userId);
  }
}