A Keycloak示例 - 使用Quarkus构建我的第一个MCP服务器工具

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22

sequenceDiagram
autonumber
participant User
participant LLM
participant Client
participant MCPServer
User->>LLM: 您的问题（带有可用工具信息）
activate LLM
LLM->>LLM: 分析问题并决定使用哪些工具
   LLM->>Client: 指示使用选定的工具
deactivate LLM
activate Client
Client->>MCPServer: 执行选定的工具
activate MCPServer
MCPServer-->>Client: 工具结果
deactivate MCPServer
Client-->>LLM: 转发的工具结果
deactivate Client
activate LLM
LLM->>LLM: 使用工具结果制定答案
LLM-->>User: 最终响应！
deactivate LLM

1
2
3
4
5
6
7
8
9

graph LR
subgraph "逻辑分解"
direction LR
JA[Java应用]
MCPC[MCP客户端]
MCPS[MCP服务器]
end
JA --> MCPC -->|Stdio| MCPS
JA --> MCPC -->|SSE| MCPS

1
2
3
4
5

<dependency>
    <groupId>io.quarkiverse.mcp</groupId>
    <artifactId>quarkus-mcp-server-stdio</artifactId>
    <version>1.0.0.Alpha5</version>
</dependency>

1
2
3
4

<dependency>
    <groupId>io.quarkus</groupId>
    <artifactId>quarkus-keycloak-admin-rest-client</artifactId>
</dependency>

1
2
3
4
5
6
7

graph TD
subgraph Realm
U[用户]
G[组]
R[角色]
C[客户端]
end

1
2
3
4
5
6
7

@ApplicationScoped // [1]
public class UserService {
    
    @Inject 
    Keycloak keycloak; // [2]

}

1
2
3

public List<UserRepresentation> getUsers(String realm) {
    return keycloak.realm(realm).users().list();
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

public String addUser(String realm, String username, String firstName, String lastName, String email, String password) {
    UserRepresentation user = new UserRepresentation(); // 设置用户字段 [1]
    user.setFirstName(firstName);
    user.setLastName(lastName);
    user.setUsername(username);
    user.setEnabled(true);
    user.setEmail(email);

    CredentialRepresentation credential = new CredentialRepresentation(); // 添加密码 [2]
    credential.setType(CredentialRepresentation.PASSWORD);
    credential.setValue(password);
    credential.setTemporary(false);
    user.setCredentials(List.of(credential));
    
    Response response = keycloak.realm(realm).users().create(user); // 发送创建请求 [3]
    if (response.getStatus() == Response.Status.CREATED.getStatusCode()) {
        return "Successfully created user: " + username;
    } else {
        Log.error("Failed to create user. Status: " + response.getStatus());
        response.close();
        return "Error creating user: "+" "+username;
    }
}

1
2

public String deleteUser(String realm, String username)
public UserRepresentation getUserByUsername(String realm, String username)

1

quarkus.keycloak.admin-client.server-url=http://localhost:8081

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17

services:
  keycloak:
    image: quay.io/keycloak/keycloak:latest
    container_name: keycloak
    environment:
      - KEYCLOAK_ADMIN=admin
      - KEYCLOAK_ADMIN_PASSWORD=admin
    ports:
      - "8081:8080"
    command: >
      start-dev
    volumes:
      - keycloak_data:/opt/keycloak/data
    restart: unless-stopped

volumes:
  keycloak_data:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15

sequenceDiagram
autonumber
%% 定义参与者
participant UserTool
participant UserService
participant Keycloak
%% 交互序列
UserTool->>UserService: 调用UserService（例如，用户/领域/客户端）
activate UserService
UserService->>Keycloak: 调用Keycloak（获取用户/领域/客户端）
activate Keycloak
Keycloak-->>UserService: 返回结果（UserRepresentation, RealmRepresentation, ClientRepresentation）
deactivate Keycloak
UserService-->>UserTool: 从UserService返回转换结果为String
deactivate UserService

1
2
3
4
5
6
7
8

public class UserTool {

    @Inject
    UserService userService;

    @Inject
    ObjectMapper mapper;
}

1
2
3
4
5
6
7
8

@Tool(description = "从keycloak领域获取所有用户") // [1]
String getUsers(@ToolArg(description = "一个表示用户所在领域名称的字符串") String realm) { // [2]
    try {
        return mapper.writeValueAsString(userService.getUsers(realm)); // [3]
    } catch (Exception e) {
        throw new ToolCallException("从领域获取用户失败");
    }
}

1
2
3
4
5
6
7
8
9

@Tool(description = "在keycloak领域中创建新用户，具有以下必填字段：领域、用户名、firstName、lastName、email、password")  // [1]
String addUser(@ToolArg(description = "一个表示用户所在领域名称的字符串") String realm,  // [2]
               @ToolArg(description = "一个表示要创建用户的用户名的字符串") String username,
               @ToolArg(description = "一个表示要创建用户的名字的字符串") String firstName,
               @ToolArg(description = "一个表示要创建用户的姓氏的字符串") String lastName,
               @ToolArg(description = "一个表示要创建用户的电子邮件的字符串") String email,
               @ToolArg(description = "一个表示要创建用户密码的字符串") String password) {
    return userService.addUser(realm, username, firstName, lastName, email, password);  // [3]
}

1
2
3

quarkus.package.jar.type=uber-jar # [1]
quarkus.log.file.enable=true # [2]
quarkus.log.file.path=kcadmin-quarkus.log # [3]

1

mvn clean package

1

goose session --with-extension="java -jar target/keycloak-mcp-server-1.0.0-SNAPSHOT-runner.jar" 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

( O)> 我可以在keycloak中创建新用户吗？
是的，您可以在Keycloak中创建新用户。为此，您需要提供以下关于用户的信息：

- **领域**：用户将所在的领域名称。
- **用户名**：新用户的用户名。
- **名字**：用户的名字。
- **姓氏**：用户的姓氏。
- **电子邮件**：用户的电子邮件地址。
- **密码**：用户帐户的密码。

您可以提供这些详细信息，我可以协助您创建用户。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11

( O)> 列出所有领域中的所有用户
以下是"quarkus"领域中的用户：

1. **admin**
   - ID: `af134cab-f41c-4675-b141-205f975db679`

2. **alice**
   - ID: `eb4123a3-b722-4798-9af5-8957f823657a`

3. **jdoe**
   - ID: `1eed6a8e-a853-4597-b4c6-c4c2533546a0`

1

( O)> 你能从quarkus领域删除用户sshaaf吗