1.5.3前端

AI生成SQL表信息设置
master
xs 3 months ago
parent 21df05fe66
commit 3e2f9594cc

@ -0,0 +1,126 @@
import request from '@/utils/request';
import { AxiosPromise } from 'axios';
import { AiSqlTableVO, AiSqlTableForm, AiSqlTableQuery,DbTableQuery,DbTableVO } from '@/api/ai/base/aiSqlTable/types';
/**
* AIsql
* @param query
* @returns {*}
*/
export const listAiSqlTable = (query?: AiSqlTableQuery): AxiosPromise<AiSqlTableVO[]> => {
return request({
url: '/ai/aiSqlTable/list',
method: 'get',
params: query
});
};
/**
* AIsql
* @param tableId
*/
export const getAiSqlTable = (tableId: string | number): AxiosPromise<AiSqlTableVO> => {
return request({
url: '/ai/aiSqlTable/' + tableId,
method: 'get'
});
};
/**
* AIsql
* @param data
*/
export const addAiSqlTable = (data: AiSqlTableForm) => {
return request({
url: '/ai/aiSqlTable',
method: 'post',
data: data
});
};
/**
* AIsql
* @param data
*/
export const updateAiSqlTable = (data: AiSqlTableForm) => {
return request({
url: '/ai/aiSqlTable',
method: 'put',
data: data
});
};
/**
* AIsql
* @param tableId
*/
export const delAiSqlTable = (tableId: string | number | Array<string | number>) => {
return request({
url: '/ai/aiSqlTable/' + tableId,
method: 'delete'
});
};
/**
* AIsql
* @param query
* @returns {*}
*/
export function getAiSqlTableList (query) {
return request({
url: '/ai/aiSqlTable/getAiSqlTableList',
method: 'get',
params: query
});
};
// 查询db数据库列表
export const listDbTable = (query: DbTableQuery): AxiosPromise<DbTableVO[]> => {
return request({
url: '/ai/aiSqlTable/db/list',
method: 'get',
params: query
});
};
// 导入表
export const importTable = (aiSqlTableList: AiSqlTableVO[]): AxiosPromise<AiSqlTableVO> => {
return request({
url: '/ai/aiSqlTable/importTable',
method: 'post',
data: aiSqlTableList
});
};
// 获取数据源名称
export const getDataNames = () => {
return request({
url: '/ai/aiSqlTable/getDataNames',
method: 'get'
});
};
/**
* AIsql
* @param tableId
*/
export const deleteAiSqlTable = (data) => {
return request({
url: '/ai/aiSqlTable/deleteAiSqlTable',
method: 'delete',
data: data
});
};
// 获取数据源名称
export const refreshCache = () => {
return request({
url: '/ai/aiSqlTable/refreshCache',
method: 'post'
});
};

@ -0,0 +1,128 @@
import {DbColumnVO} from "@/api/tool/gen/types";
export interface AiSqlTableVO {
/**
*
*/
tableId: string | number;
/**
*
*/
tableName: string;
/**
*
*/
tableComment: string;
/**
*
*/
subTableName: string;
/**
*
*/
subTableFkName: string;
}
export interface AiSqlTableForm extends BaseEntity {
/**
*
*/
tableId?: string | number;
/**
*
*/
tableName?: string;
/**
*
*/
tableComment?: string;
/**
*
*/
subTableName?: string;
/**
*
*/
subTableFkName?: string;
}
export interface AiSqlTableQuery extends PageQuery {
/**
*
*/
tableId?: string | number;
/**
*
*/
tableName?: string;
/**
*
*/
tableComment?: string;
/**
*
*/
subTableName?: string;
/**
*
*/
subTableFkName?: string;
/**
*
*/
params?: any;
}
export interface DbTableQuery extends PageQuery {
dataName: string;
tableName: string;
tableComment: string;
}
export interface DbTableVO {
createDept?: any;
tableId?: any;
tableName: string;
tableComment: string;
subTableName?: any;
subTableFkName?: any;
className?: any;
tplCategory?: any;
packageName?: any;
moduleName?: any;
businessName?: any;
functionName?: any;
functionAuthor?: any;
genType?: any;
genPath?: any;
pkColumn?: any;
columns: DbColumnVO[];
options?: any;
remark?: any;
treeCode?: any;
treeParentCode?: any;
treeName?: any;
menuIds?: any;
parentMenuId?: any;
parentMenuName?: any;
tree: boolean;
crud: boolean;
}

@ -69,6 +69,17 @@ export const delAiMessage = (sessionId: string) => {
};
export const generateSql = (query?: AskVO): AxiosPromise<AskVO[]> => {
return request({
url: '/ai/assistant/generateSql',
method: 'post',
params: query,
timeout: 1000000
});
};
/**
* AI
@ -92,13 +103,7 @@ export const aiStreamAsk = (query?: AskVO): AxiosPromise<AskVO[]> => {
});
};
export const aiSqlAsk = (query?: AskVO): AxiosPromise<AskVO[]> => {
return request({
url: '/ai/api/chat/sql',
method: 'post',
params: query
});
};
export const streamDeepSeek = (prompt, handlers) => {

@ -1,16 +1,19 @@
export interface AskVO {
/**
*
* prompt
*/
question: string;
text: string;
/**
* api
* AIID
*/
provider: string;
modelId: number;
/**
* AIID
*/
platformId: number;
}

@ -0,0 +1,271 @@
<template>
<div class="p-2">
<transition :enter-active-class="proxy?.animate.searchAnimate.enter"
:leave-active-class="proxy?.animate.searchAnimate.leave">
<div v-show="showSearch" class="mb-[10px]">
<el-card shadow="hover">
<el-form ref="queryFormRef" :model="queryParams" :inline="true">
<el-form-item label="表名称" prop="tableName">
<el-input v-model="queryParams.tableName" placeholder="请输入表名称" clearable
@keyup.enter="handleQuery"/>
</el-form-item>
<el-form-item label="表描述" prop="tableComment">
<el-input v-model="queryParams.tableComment" placeholder="请输入表描述" clearable
@keyup.enter="handleQuery"/>
</el-form-item>
<el-form-item>
<el-button type="primary" icon="Search" @click="handleQuery"></el-button>
<el-button icon="Refresh" @click="resetQuery"></el-button>
</el-form-item>
</el-form>
</el-card>
</div>
</transition>
<el-card shadow="never">
<template #header>
<el-row :gutter="10" class="mb8">
<el-col :span="1.5">
<el-button type="primary" plain icon="Upload" @click="handleAdd" v-hasPermi="['ai:aiSqlTable:add']">
</el-button>
</el-col>
<el-col :span="1.5">
<el-button type="danger" plain icon="Delete" :disabled="multiple" @click="handleDelete()"
v-hasPermi="['ai:aiSqlTable:remove']">删除
</el-button>
</el-col>
<el-col :span="1.5">
<el-button type="warning" plain icon="Refresh" @click="handleRefresh" v-hasPermi="['ai:aiSqlTable:refresh']">
刷新
</el-button>
</el-col>
</el-row>
</template>
<el-table v-loading="loading" :data="aiSqlTableList" @selection-change="handleSelectionChange">
<el-table-column type="selection" width="55" align="center"/>
<el-table-column label="编号" align="center" prop="tableId" v-if="columns[0].visible"/>
<el-table-column label="数据源" align="center" prop="dataName" />
<el-table-column label="表名称" align="center" prop="tableName" v-if="columns[2].visible"/>
<el-table-column label="表描述" align="center" prop="tableComment" v-if="columns[3].visible"/>
<el-table-column label="操作" align="center" class-name="small-padding fixed-width">
<template #default="scope">
<el-tooltip content="删除" placement="top">
<el-button link type="primary" icon="Delete" @click="handleDelete(scope.row)"
v-hasPermi="['ai:aiSqlTable:remove']"></el-button>
</el-tooltip>
</template>
</el-table-column>
</el-table>
<pagination v-show="total > 0" :total="total" v-model:page="queryParams.pageNum"
v-model:limit="queryParams.pageSize" @pagination="getList"/>
</el-card>
<!-- 添加或修改AI生成sql信息对话框 -->
<import-table ref="importRef" @ok="handleQuery"/>
</div>
</template>
<script setup name="AiSqlTable" lang="ts">
import {
listAiSqlTable,
getAiSqlTable,
deleteAiSqlTable,
addAiSqlTable,
updateAiSqlTable,
refreshCache
} from '@/api/ai/base/aiSqlTable';
import {AiSqlTableVO, AiSqlTableQuery, AiSqlTableForm} from '@/api/ai/base/aiSqlTable/types';
import ImportTable from '../common/importTable.vue';
const {proxy} = getCurrentInstance() as ComponentInternalInstance;
const importRef = ref<InstanceType<typeof ImportTable>>();
const aiSqlTableList = ref<AiSqlTableVO[]>([]);
const buttonLoading = ref(false);
const loading = ref(true);
const showSearch = ref(true);
const ids = ref<Array<string | number>>([]);
const selectedRows = ref<AiSqlTableVO[]>([]);
const single = ref(true);
const multiple = ref(true);
const total = ref(0);
const queryFormRef = ref<ElFormInstance>();
const aiSqlTableFormRef = ref<ElFormInstance>();
const dialog = reactive<DialogOption>({
visible: false,
title: ''
});
//
const columns = ref<FieldOption[]>([
{key: 0, label: `编号`, visible: true},
{key: 1, label: `租户ID`, visible: true},
{key: 2, label: `表名称`, visible: true},
{key: 3, label: `表描述`, visible: true},
{key: 4, label: `关联子表的表名`, visible: true},
{key: 5, label: `子表关联的外键名`, visible: true},
{key: 6, label: `创建部门`, visible: true},
{key: 7, label: `创建者`, visible: true},
{key: 8, label: `创建时间`, visible: true},
{key: 9, label: `更新者`, visible: true},
{key: 10, label: `更新时间`, visible: true},
{key: 11, label: `删除标志`, visible: true},
]);
const initFormData: AiSqlTableForm = {
tableId: undefined,
tableName: undefined,
tableComment: undefined,
subTableName: undefined,
subTableFkName: undefined,
}
const data = reactive<PageData<AiSqlTableForm, AiSqlTableQuery>>({
form: {...initFormData},
queryParams: {
pageNum: 1,
pageSize: 10,
tableId: undefined,
tableName: undefined,
tableComment: undefined,
subTableName: undefined,
subTableFkName: undefined,
params: {}
},
rules: {
tableId: [
{required: true, message: "编号不能为空", trigger: "blur"}
],
tableName: [
{required: true, message: "表名称不能为空", trigger: "blur"}
],
tableComment: [
{required: true, message: "表描述不能为空", trigger: "blur"}
],
subTableName: [
{required: true, message: "关联子表的表名不能为空", trigger: "blur"}
],
subTableFkName: [
{required: true, message: "子表关联的外键名不能为空", trigger: "blur"}
],
}
});
const {queryParams, form, rules} = toRefs(data);
/** 查询AI生成sql信息列表 */
const getList = async () => {
loading.value = true;
const res = await listAiSqlTable(queryParams.value);
aiSqlTableList.value = res.rows;
total.value = res.total;
loading.value = false;
}
/** 取消按钮 */
const cancel = () => {
reset();
dialog.visible = false;
}
/** 表单重置 */
const reset = () => {
form.value = {...initFormData};
aiSqlTableFormRef.value?.resetFields();
}
/** 搜索按钮操作 */
const handleQuery = () => {
queryParams.value.pageNum = 1;
getList();
}
/** 重置按钮操作 */
const resetQuery = () => {
queryFormRef.value?.resetFields();
handleQuery();
}
/** 多选框选中数据 */
const handleSelectionChange = (selection: AiSqlTableVO[]) => {
ids.value = selection.map(item => item.tableId);
single.value = selection.length != 1;
multiple.value = !selection.length;
selectedRows.value = selection;
}
/** 新增按钮操作 */
const handleAdd = () => {
reset();
dialog.visible = true;
dialog.title = "添加AI生成sql信息";
importRef.value?.show(queryParams.value.dataName);
}
/** 修改按钮操作 */
const handleUpdate = async (row?: AiSqlTableVO) => {
reset();
const _tableId = row?.tableId || ids.value[0]
const res = await getAiSqlTable(_tableId);
Object.assign(form.value, res.data);
dialog.visible = true;
dialog.title = "修改AI生成sql信息";
}
/** 提交按钮 */
const submitForm = () => {
aiSqlTableFormRef.value?.validate(async (valid: boolean) => {
if (valid) {
buttonLoading.value = true;
if (form.value.tableId) {
await updateAiSqlTable(form.value).finally(() => buttonLoading.value = false);
} else {
await addAiSqlTable(form.value).finally(() => buttonLoading.value = false);
}
proxy?.$modal.msgSuccess("操作成功");
dialog.visible = false;
await getList();
}
});
}
/** 删除按钮操作 */
const handleDelete = async (row?: AiSqlTableVO) => {
const _tableIds = row?.tableId || ids.value;
let toDeletedRows = [];
if (row) {
toDeletedRows.push(row);
} else {
toDeletedRows = selectedRows.value;
}
await proxy?.$modal.confirm('是否确认删除AI生成sql信息编号为"' + _tableIds + '"的数据项?').finally(() => loading.value = false);
await deleteAiSqlTable(toDeletedRows);
proxy?.$modal.msgSuccess("删除成功");
await getList();
}
/** 导出按钮操作 */
const handleExport = () => {
proxy?.download('ai/aiSqlTable/export', {
...queryParams.value
}, `aiSqlTable_${new Date().getTime()}.xlsx`)
}
/** 刷新按钮操作 */
const handleRefresh = async () => {
await refreshCache();
proxy?.$modal.msgSuccess("刷新成功");
}
onMounted(() => {
getList();
});
</script>

@ -0,0 +1,141 @@
<template>
<!-- 导入表 -->
<el-dialog v-model="visible" title="导入表" width="1100px" top="5vh" append-to-body>
<el-form ref="queryFormRef" :model="queryParams" :inline="true">
<el-form-item label="数据源" prop="dataName">
<el-select v-model="queryParams.dataName" filterable placeholder="请选择/输入数据源名称">
<el-option v-for="item in dataNameList" :key="item" :label="item" :value="item"> </el-option>
</el-select>
</el-form-item>
<el-form-item label="表名称" prop="tableName">
<el-input v-model="queryParams.tableName" placeholder="请输入表名称" clearable @keyup.enter="handleQuery" />
</el-form-item>
<el-form-item label="表描述" prop="tableComment">
<el-input v-model="queryParams.tableComment" placeholder="请输入表描述" clearable @keyup.enter="handleQuery" />
</el-form-item>
<el-form-item>
<el-button type="primary" icon="Search" @click="handleQuery"></el-button>
<el-button icon="Refresh" @click="resetQuery"></el-button>
</el-form-item>
</el-form>
<el-row>
<el-table ref="tableRef" :data="dbTableList" height="260px" @row-click="clickRow" @selection-change="handleSelectionChange">
<el-table-column type="selection" width="55"></el-table-column>
<el-table-column prop="tableName" label="表名称" :show-overflow-tooltip="true"></el-table-column>
<el-table-column prop="tableComment" label="表描述" :show-overflow-tooltip="true"></el-table-column>
<el-table-column prop="createTime" label="创建时间"></el-table-column>
<el-table-column prop="updateTime" label="更新时间"></el-table-column>
</el-table>
<pagination v-show="total > 0" v-model:page="queryParams.pageNum" v-model:limit="queryParams.pageSize" :total="total" @pagination="getList" />
</el-row>
<template #footer>
<div class="dialog-footer">
<el-button type="primary" @click="handleImportTable"> </el-button>
<el-button @click="visible = false"> </el-button>
</div>
</template>
</el-dialog>
</template>
<script setup lang="ts">
import { listDbTable, importTable, getDataNames } from '@/api/ai/base/aiSqlTable';
import {DbTableQuery, DbTableVO} from '@/api/ai/base/aiSqlTable/types';
const total = ref(0);
const visible = ref(false);
const tables = ref<Array<string>>([]);
const selectedTables = ref([]);
const dbTableList = ref<Array<DbTableVO>>([]);
const { proxy } = getCurrentInstance() as ComponentInternalInstance;
const tableRef = ref<ElTableInstance>();
const queryFormRef = ref<ElFormInstance>();
const queryParams = ref<DbTableQuery>({
pageNum: 1,
pageSize: 10,
tableName: '',
tableComment: '',
dataName: ''
});
const dataNameList = ref<Array<string>>([]);
const emit = defineEmits(['ok']);
/** 查询参数列表 */
const show = (dataName: string) => {
getDataNameList();
if (dataName) {
queryParams.value.dataName = dataName;
} else {
queryParams.value.dataName = 'master';
}
getList();
visible.value = true;
};
/** 单击选择行 */
const clickRow = (row: DbTableVO) => {
// ele bug
tableRef.value?.toggleRowSelection(row, false);
};
/** 多选框选中数据 */
const handleSelectionChange = (selection: DbTableVO[]) => {
selectedTables.value = selection.map((item) => {
return {
tableName: item.tableName,
tableComment: item.tableComment,
dataName: queryParams.value.dataName
}
})
// const index = selectedTables.value.findIndex(t => t.tableName === selection.tableName);
// if (index >= 0) {
// selectedTables.value.splice(index, 1);
// } else {
// selectedTables.value.push({
// tableName: selection.tableName,
// tableComment: selection.tableComment
// });
// }
};
/** 查询表数据 */
const getList = async () => {
const res = await listDbTable(queryParams.value);
dbTableList.value = res.rows;
total.value = res.total;
};
/** 搜索按钮操作 */
const handleQuery = () => {
queryParams.value.pageNum = 1;
getList();
};
/** 重置按钮操作 */
const resetQuery = () => {
queryFormRef.value?.resetFields();
handleQuery();
};
/** 导入按钮操作 */
const handleImportTable = async () => {
if (selectedTables.value.length <= 0) {
proxy?.$modal.msgError('请选择要导入的表');
return;
}
const res = await importTable( selectedTables.value );
proxy?.$modal.msgSuccess(res.msg);
if (res.code === 200) {
visible.value = false;
emit('ok');
}
};
/** 查询多数据源名称 */
const getDataNameList = async () => {
const res = await getDataNames();
dataNameList.value = res.data;
};
defineExpose({
show
});
</script>
Loading…
Cancel
Save