Skip to content

Commit 8559b5e

Browse files
authored
Merge pull request #1539 from rocket-admin/backend_ws_server
Add new command types and extend test agent token support for Redis, Cassandra, and ClickHouse
2 parents 674667e + a755b32 commit 8559b5e

2 files changed

Lines changed: 7 additions & 0 deletions

File tree

autoadmin-ws-server/src/constants/command-type.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ export const COMMAND_TYPE = {
1515
dataFromAgent: 'dataFromAgent',
1616
isView: 'isView',
1717
getRowsAsStream: 'getRowsAsStream',
18+
executeRawQuery: 'executeRawQuery',
1819
} as const;
1920

2021
export type CommandType = (typeof COMMAND_TYPE)[keyof typeof COMMAND_TYPE];

backend/src/entities/agent/repository/custom-agent-repository-extension.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,12 @@ export const customAgentRepositoryExtension = {
6363
return 'IBMDB2-TEST-AGENT-TOKEN';
6464
case ConnectionTypeTestEnum.agent_mongodb:
6565
return 'MONGODB-TEST-AGENT-TOKEN';
66+
case ConnectionTypeTestEnum.agent_redis:
67+
return 'REDIS-TEST-AGENT-TOKEN';
68+
case ConnectionTypeTestEnum.agent_cassandra:
69+
return 'CASSANDRA-TEST-AGENT-TOKEN';
70+
case ConnectionTypeTestEnum.agent_clickhouse:
71+
return 'CLICKHOUSE-TEST-AGENT-TOKEN';
6672
default:
6773
throw new Error(`Unsupported connection type for test agent token: ${connectionType}`);
6874
}

0 commit comments

Comments
 (0)