fix routing table key list

This commit is contained in:
Xi Yan 2024-10-02 18:23:02 -07:00
parent 8d049000e3
commit 703ab9385f
2 changed files with 19 additions and 10 deletions

View file

@ -52,8 +52,8 @@ repos:
# - id: pydoclint # - id: pydoclint
# args: [--config=pyproject.toml] # args: [--config=pyproject.toml]
- repo: https://github.com/tcort/markdown-link-check # - repo: https://github.com/tcort/markdown-link-check
rev: v3.11.2 # rev: v3.11.2
hooks: # hooks:
- id: markdown-link-check # - id: markdown-link-check
args: ['--quiet'] # args: ['--quiet']

View file

@ -94,6 +94,15 @@ class ShieldsRoutingTable(CommonRoutingTableImpl, Shields):
async def list_shields(self) -> List[ShieldSpec]: async def list_shields(self) -> List[ShieldSpec]:
specs = [] specs = []
for entry in self.routing_table_config: for entry in self.routing_table_config:
if isinstance(entry.routing_key, list):
for k in entry.routing_key:
specs.append(
ShieldSpec(
shield_type=k,
provider_config=entry,
)
)
else:
specs.append( specs.append(
ShieldSpec( ShieldSpec(
shield_type=entry.routing_key, shield_type=entry.routing_key,