Skip to content

Commit a791ffc

Browse files
committed
feat: add header_name parameter to Kong rate limiting plugin
Add support for header-based rate limiting by adding the header_name parameter to both create_rate_limiting_plugin and update_rate_limiting_plugin functions.
1 parent 9cb1505 commit a791ffc

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

src/kong_mcp_server/tools/kong_rate_limiting.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ async def create_rate_limiting_plugin(
1818
policy: str = "local",
1919
fault_tolerant: bool = True,
2020
hide_client_headers: bool = False,
21+
header_name: Optional[str] = None,
2122
redis_host: Optional[str] = None,
2223
redis_port: int = 6379,
2324
redis_password: Optional[str] = None,
@@ -64,6 +65,10 @@ async def create_rate_limiting_plugin(
6465
"fault_tolerant": fault_tolerant,
6566
"hide_client_headers": hide_client_headers,
6667
}
68+
69+
# Add header configuration for header-based limiting
70+
if header_name is not None:
71+
config["header_name"] = header_name
6772

6873
# Add time-based limits
6974
if second is not None:
@@ -171,6 +176,7 @@ async def update_rate_limiting_plugin(
171176
policy: Optional[str] = None,
172177
fault_tolerant: Optional[bool] = None,
173178
hide_client_headers: Optional[bool] = None,
179+
header_name: Optional[str] = None,
174180
redis_host: Optional[str] = None,
175181
redis_port: Optional[int] = None,
176182
redis_password: Optional[str] = None,
@@ -231,6 +237,8 @@ async def update_rate_limiting_plugin(
231237
config["fault_tolerant"] = fault_tolerant
232238
if hide_client_headers is not None:
233239
config["hide_client_headers"] = hide_client_headers
240+
if header_name is not None:
241+
config["header_name"] = header_name
234242

235243
# Update Redis configuration
236244
if redis_host is not None:

0 commit comments

Comments
 (0)