Skip to content

feat(platform-api): support upstreamDefinitions pool and per-operation upstream overrides#2155

Draft
mehara-rothila wants to merge 1 commit into
wso2:mainfrom
mehara-rothila:feat/platform-api-upstream-definitions
Draft

feat(platform-api): support upstreamDefinitions pool and per-operation upstream overrides#2155
mehara-rothila wants to merge 1 commit into
wso2:mainfrom
mehara-rothila:feat/platform-api-upstream-definitions

Conversation

@mehara-rothila

@mehara-rothila mehara-rothila commented Jun 11, 2026

Copy link
Copy Markdown
Contributor

Summary

This PR implements the upstreamDefinitions pool and per-operation upstream overrides in the platform-api control plane, so an API can declare a set of reusable named backends and route the API-level upstream or any individual operation to one of them by reference. This single commit adds both the OpenAPI schema for these shapes and the Go behavior behind it: model types, request and response mapping, persistence, validation, and deployment-descriptor emission.

Why

The gateway side of this feature (#2012) understands upstreamDefinitions and an operation-level upstream.ref, but the control plane never carried the pool through. As a result a ref (API-level or operation-level) was accepted by the API and then had nothing to resolve against, producing a configuration the gateway rejects when it is deployed. This PR closes that gap on the control plane: the pool is stored, returned on reads, and emitted into the deployment descriptor, and every reference is resolved before anything is persisted or deployed, so an unresolved reference is reported to the caller as a validation error instead of failing later at deploy time.

Design decisions

Operation-level overrides are reference-only. An operation's upstream.main and upstream.sandbox each carry only a ref into the pool, never an inline URL, and a side that is omitted falls back to the API-level upstream. This keeps every backend definition in one place (the pool) and matches the gateway contract.

Validation mirrors the gateway's rules and runs on every write path. On create, and on the merged result of an update, the control plane rejects: a duplicate or malformed pool name (must match ^[a-zA-Z0-9\-_]+$, at most 100 characters), a pool entry with no backends, a backend URL that is not host-only http or https (a path, query, or fragment is rejected because the base path belongs in basePath), a weight outside 0 to 100, a connect timeout that is not a single unsigned ms, s, m, or h unit or is not positive, an API-level endpoint that sets both url and ref or neither, and any API-level or operation-level ref that does not resolve to a declared pool entry. Validating the merged result of an update means an update cannot remove a pool entry that a still-present operation references. These surface as HTTP 400 through the standard error catalog.

No database migration. The pool and the per-operation references are stored inside the existing rest_apis.configuration JSON blob, so rows written before this change simply decode with the new fields empty and remain fully compatible.

The pool entry type is named ReusableUpstream. The platform-api generated code already has an UpstreamDefinition type for the API-level endpoint (the url-or-ref union), so the pool entry uses a distinct name to avoid a collision. This is open to the maintainers' preference if a different name is preferred for consistency with the gateway.

Scope is limited to the upstream feature. This PR only touches the upstreamDefinitions pool and per-operation upstream behavior; no unrelated validation or formatting changes are included.

What changed

Added the ReusableUpstream, UpstreamReference, OperationUpstream, and UpstreamTimeout schemas to the OpenAPI spec and regenerated api/generated.go (the new types plus the new fields on the REST API and operation request shapes). Added persistence types and fields in internal/model, lossless converters and deployment-YAML emission in internal/utils, the pool and reference validation plus create and update wiring in internal/service, the deployment-YAML DTOs in internal/dto, the validation error sentinel in internal/constants, and the 400 mapping in internal/handler. The pool and per-operation references are carried through create, update, get, list, and deployment emission.

Test coverage

Table-driven validation tests cover the full rule set above (pool integrity, backend URL shape, weight and timeout bounds, the API-level union, and API-level and operation-level reference resolution). Real-SQLite integration tests, wired through the concrete service and repositories, cover create, persist, and read of a pool with a per-operation override, rejection of an unresolved operation reference on create, and rejection of an update that removes a pool entry still referenced by an unchanged operation. Emission tests assert the exact spec.upstreamDefinitions and spec.operations[].upstream content and that a WebSub API omits the pool, a conversion test proves the pool and per-operation references survive the request-to-model-to-response round trip, and a repository test proves they survive the configuration blob's database round trip.

Example

An API can declare a pool and route one operation to an alternate backend while the rest use the API-level upstream:

upstream:
  main:
    url: http://default-backend:8080
upstreamDefinitions:
  - name: alt-backend
    basePath: /api/v2
    timeout:
      connect: 5s
    upstreams:
      - url: http://alt-backend:9090
        weight: 100
operations:
  - method: GET
    path: /whoami
    upstream:
      main:
        ref: alt-backend
  - method: GET
    path: /ping

Here GET /whoami is routed to alt-backend from the pool, while GET /ping uses the API-level main upstream.

Related PRs

Depends on #2012, the gateway (data plane) side of this feature, which should merge first: the deployment YAML this PR emits for per-operation upstreams is only understood by the gateway once #2012 lands. This PR supersedes #2108, which carried the same schema and was closed in favor of this branch.

@mehara-rothila mehara-rothila changed the title feat(platform-api): add upstreamDefinitions pool to the REST API Open… feat(platform-api): support API-level upstreamDefinitions pool Jun 11, 2026
@coderabbitai

coderabbitai Bot commented Jun 11, 2026

Copy link
Copy Markdown
Contributor

Review Change Stack

📝 Walkthrough

Walkthrough

This change adds reusable upstream definitions and per-operation upstream overrides across the OpenAPI contract, generated API models, internal models, YAML DTOs, service validation, API handlers, conversion utilities, repository persistence, and deployment YAML generation. Upstream definitions support named backend pools, optional base paths, weighted targets, and connect timeouts. Operations can reference named definitions for main and sandbox upstreams. Create and update flows validate definitions and references, while tests cover conversion, merging, persistence, and deployment output.

Sequence Diagram(s)

sequenceDiagram
  participant Client
  participant APIHandler
  participant APIService
  participant Repository
  participant DeploymentYAML
  Client->>APIHandler: submit API with upstream definitions
  APIHandler->>APIService: create or update request
  APIService->>APIService: validate definitions and operation references
  APIService->>Repository: persist validated configuration
  Repository-->>APIService: stored API configuration
  APIService->>DeploymentYAML: build deployment specification
  DeploymentYAML-->>Client: upstream pools and operation references
Loading

Suggested reviewers: RakhithaRR, Tharsanan1, VirajSalaka, renuka-fernando, AnuGayan, chamilaadhi, Arshardh, dushaniw, tgtshanika, senthuran16, CrowleyRajapakse, hisanhunais, HiranyaKavishani, ashera96, pubudu538

🚥 Pre-merge checks | ✅ 5
✅ Passed checks (5 passed)
Check name Status Explanation
Docstring Coverage ✅ Passed Docstring coverage is 81.25% which is sufficient. The required threshold is 80.00%.
Linked Issues check ✅ Passed Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check ✅ Passed Check skipped because no linked issues were found for this pull request.
Title check ✅ Passed The title clearly summarizes the main feature: reusable upstream definitions and per-operation overrides.
Description check ✅ Passed The description covers the problem, goals, implementation approach, tests, and related PRs, though some template sections are not explicit.
✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands.

@mehara-rothila

Copy link
Copy Markdown
Contributor Author

@coderabbitai review

@coderabbitai

coderabbitai Bot commented Jun 11, 2026

Copy link
Copy Markdown
Contributor
✅ Action performed

Review finished.

Note: CodeRabbit is an incremental review system and does not re-review already reviewed commits. This command is applicable only when automatic reviews are paused.

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 2

🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@platform-api/src/resources/openapi.yaml`:
- Around line 8455-8464: UpstreamTimeout.connect's OpenAPI regex only allows a
single unit but the runtime uses time.ParseDuration (see
spec.upstreamDefinitions[%d].timeout.connect), so update the OpenAPI schema for
UpstreamTimeout.connect to either (A) broaden the pattern to match Go
time.ParseDuration format (allow multiple sequential value+unit segments,
decimals, and units like ns|us|µs|ms|s|m|h, optional leading minus) or (B)
explicitly document/validate that only single-unit durations are accepted;
locate the UpstreamTimeout.connect property in the OpenAPI file and change the
pattern or description accordingly.
- Around line 8427-8447: Update the ReusableUpstream schema so its basePath
property is constrained to a path-only form and each upstream.url is constrained
to host[:port]-only (no path/query/fragment) to match gateway validation:
tighten basePath (property name basePath) to require a leading "/" and a path
pattern (e.g., path segment characters only) and change
upstreams.items.properties.url to validate host[:port] (e.g., restrict
format/pattern to host with optional port) instead of generic uri; keep timeout
($ref: '`#/components/schemas/UpstreamTimeout`') unchanged.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro

Run ID: c978ac83-b305-4917-85d3-8826d792c30e

📥 Commits

Reviewing files that changed from the base of the PR and between 66e4674 and e6e8d18.

📒 Files selected for processing (1)
  • platform-api/src/resources/openapi.yaml

Comment thread platform-api/resources/openapi.yaml
Comment thread platform-api/resources/openapi.yaml
@renuka-fernando renuka-fernando changed the base branch from main to feature/operation-level-ep June 12, 2026 11:52
@renuka-fernando renuka-fernando self-assigned this Jun 12, 2026
Comment thread platform-api/resources/openapi.yaml
Comment thread platform-api/resources/openapi.yaml
Comment thread platform-api/src/resources/openapi.yaml Outdated
@mehara-rothila mehara-rothila changed the title feat(platform-api): support API-level upstreamDefinitions pool feat(platform-api): support upstreamDefinitions pool and per-operation upstream overrides Jul 7, 2026
@mehara-rothila mehara-rothila force-pushed the feat/platform-api-upstream-definitions branch 2 times, most recently from e27ea92 to 3cb7216 Compare July 7, 2026 15:57
@mehara-rothila

Copy link
Copy Markdown
Contributor Author

@coderabbitai a naming sanity-check on the upstream schemas in this PR, please.

Context: this PR (#2155) is the platform-api / control-plane side of the per-operation upstream feature. Its gateway / data-plane companion is #2012 (#2012), which adds the matching schemas to gateway/gateway-controller/api/management-openapi.yaml. The two sides must use matching schema shapes and, ideally, matching names. Please review the naming with both in mind.

Current state in platform-api/src/resources/openapi.yaml (this PR):

  • Upstream , the {main, sandbox} wrapper, shared by RESTAPI, LLMProvider, MCPServer and WebSub, so it cannot be renamed.
  • UpstreamDefinition , a single endpoint (url or ref + auth), used only by Upstream.main/sandbox.
  • ReusableUpstream , a named pool entry (name, basePath, timeout, upstreams[]), held by the new upstreamDefinitions array.

How the gateway side names the same things:

The clash: our field is upstreamDefinitions, but its items are ReusableUpstream, while a different schema here is named UpstreamDefinition. The gateway's pool name (UpstreamDefinition) is already taken on our side by the endpoint schema.

Proposed rename, keeping Upstream as-is:

  • ReusableUpstream -> UpstreamDefinition (so the upstreamDefinitions items match both the field name and the gateway pool name)
  • endpoint UpstreamDefinition -> UpstreamEndpoint

Please advise:

  1. Is this the clearest naming, or do you have a better suggestion?
  2. Any $ref, collision, or backward-compatibility risk we are missing?
  3. Does this align the control-plane (feat(platform-api): support upstreamDefinitions pool and per-operation upstream overrides #2155) and data-plane (Add per-operation upstream override for REST API operations #2012) naming as closely as the shared Upstream wrapper allows?

@coderabbitai

coderabbitai Bot commented Jul 8, 2026

Copy link
Copy Markdown
Contributor

Tip

For best results, initiate chat on the files or code changes.

🧩 Analysis chain

🏁 Script executed:

#!/bin/bash
set -euo pipefail

echo "== Find the OpenAPI files =="
fd -p 'openapi.yaml|management-openapi.yaml'

echo
echo "== Outline current platform-api upstream-related schema names =="
rg -n -C2 '^\s{4}(Upstream|UpstreamDefinition|ReusableUpstream|UpstreamTimeout|RESTAPI):|upstreamDefinitions:|`#/components/schemas/`(Upstream|UpstreamDefinition|ReusableUpstream|UpstreamTimeout)\b' \
  platform-api/src/resources/openapi.yaml || true

echo
echo "== Read the relevant platform-api slice =="
start=$(rg -n '^\s{4}RESTAPI:' platform-api/src/resources/openapi.yaml | cut -d: -f1 | head -n1)
end=$(rg -n '^\s{4}UpstreamTimeout:' platform-api/src/resources/openapi.yaml | cut -d: -f1 | head -n1)
if [[ -n "${start:-}" && -n "${end:-}" ]]; then
  sed -n "$((start-10)),$((end+80))p" platform-api/src/resources/openapi.yaml | nl -ba
fi

echo
echo "== Current branch: all refs to UpstreamDefinition / ReusableUpstream in platform-api =="
rg -n -C2 '`#/components/schemas/`(UpstreamDefinition|ReusableUpstream)\b|^\s{4}(UpstreamDefinition|ReusableUpstream):' platform-api/src/resources/openapi.yaml || true

echo
echo "== Companion PR `#2012`: inspect gateway management-openapi upstream naming =="
gh pr diff 2012 --repo wso2/api-platform -- gateway/gateway-controller/api/management-openapi.yaml | \
  rg -n -C3 'UpstreamDefinition|UpstreamReference|OperationUpstream|upstreamDefinitions|timeout|basePath|main:|sandbox:' || true

echo
echo "== Companion PR `#2012` metadata =="
gh pr view 2012 --repo wso2/api-platform --json number,title,headRefName,baseRefName,state,url

Length of output: 4416


🏁 Script executed:

#!/bin/bash
set -euo pipefail

echo "== platform-api: RESTAPI and upstream schema slice =="
sed -n '6318,8498p' platform-api/src/resources/openapi.yaml

echo
echo "== platform-api: precise refs/usages of UpstreamDefinition and ReusableUpstream =="
rg -n -C2 '`#/components/schemas/`(UpstreamDefinition|ReusableUpstream)\b|^\s{4}(UpstreamDefinition|ReusableUpstream):' \
  platform-api/src/resources/openapi.yaml

echo
echo "== gateway companion PR `#2012`: files changed =="
gh pr view 2012 --repo wso2/api-platform --json files \
  --jq '.files[].path'

echo
echo "== gateway companion PR `#2012`: upstream-related diff excerpt from management-openapi.yaml =="
gh pr diff 2012 --repo wso2/api-platform | \
  sed -n '/gateway\/gateway-controller\/api\/management-openapi.yaml/,$p' | \
  rg -n -C4 'UpstreamDefinition|UpstreamReference|OperationUpstream|upstreamDefinitions|`#/components/schemas/`' || true

echo
echo "== gateway main branch: current upstream naming in management-openapi.yaml =="
rg -n -C3 '^\s{4}(UpstreamDefinition|UpstreamReference|OperationUpstream|API|RESTAPI):|upstreamDefinitions:|`#/components/schemas/`(UpstreamDefinition|UpstreamReference|OperationUpstream)\b' \
  gateway/gateway-controller/api/management-openapi.yaml || true

Length of output: 131926


@mehara-rothila I think your proposed rename is the clearest option.

  1. Recommended naming
    I would go with exactly this:

    • ReusableUpstream -> UpstreamDefinition
    • current endpoint-level UpstreamDefinition -> UpstreamEndpoint

    That gives a cleaner model:

    • Upstream = API-level {main, sandbox} wrapper
    • OperationUpstream = per-operation {main, sandbox} wrapper
    • UpstreamEndpoint = a single direct target or ref (url / ref / auth)
    • UpstreamDefinition = a named reusable pool entry in upstreamDefinitions
    • UpstreamReference = the reusable name token
    • UpstreamTimeout = timeout block used by the reusable definition

    Why this is better:

    • In this PR, upstreamDefinitions currently contains ReusableUpstream (Line 6329-Line 6333), while UpstreamDefinition means something else entirely (Line 8376 onward). That is the main source of confusion.
    • The endpoint-level schema is only used under Upstream.main / Upstream.sandbox (Line 8372, Line 8374), so UpstreamEndpoint is a more accurate name.
    • On the gateway side, upstreamDefinitions.items already points to UpstreamDefinition (Line 4155-Line 4159), so this change aligns the pool-entry concept across control plane and data plane.
  2. $ref, collision, and compatibility risk
    I do not see a structural $ref risk as long as all local refs are updated consistently.

    The concrete updates would be:

    • RESTAPI.upstreamDefinitions.items:
      • #/components/schemas/ReusableUpstream -> #/components/schemas/UpstreamDefinition
    • Upstream.main and Upstream.sandbox:
      • #/components/schemas/UpstreamDefinition -> #/components/schemas/UpstreamEndpoint
    • rename the component keys themselves accordingly
    • update descriptions that still say “ReusableUpstream” to say “UpstreamDefinition”
    • keep UpstreamReference pointing at the reusable-definition name contract

    Compatibility caveat:

    • Wire format: no payload-field change, so API JSON/YAML shape stays the same.
    • Generated code/docs: yes, schema symbol names will change in generated types, DTO names, docs, and any downstream code that refers to component names rather than payload fields. Since this PR is still draft and explicitly schema-only, this is the best time to do it, before regeneration and follow-up wiring land.
  3. Alignment with #2012
    Yes — this is probably the closest practical alignment possible given the shared Upstream wrapper on the platform-api side.

    After the rename, the mapping would be coherent across both sides:

    • reusable pool entry: UpstreamDefinition on both sides
    • reference token: UpstreamReference on both sides
    • per-operation wrapper: OperationUpstream on both sides
    • API-level wrapper: only platform-api has Upstream, which is fine and reflects the existing shared model constraint

    So the only unavoidable naming divergence would be the presence of Upstream in platform-api, not the reusable-definition terminology.

Bottom line: I recommend making the rename now. It reduces local ambiguity and brings the control-plane terminology into line with the gateway model.

One small follow-up suggestion: if you do this rename, also update the UpstreamReference description from “Name of a ReusableUpstream entry...” to “Name of an UpstreamDefinition entry...”, so the docs stay consistent.


✏️ Learnings added
Learnt from: mehara-rothila
URL: https://github.com/wso2/api-platform/pull/2155

Timestamp: 2026-07-08T01:21:26.025Z
Learning: In `platform-api/src/resources/openapi.yaml`, the shared `Upstream` `{main, sandbox}` wrapper is reused by multiple artifact types and should remain unchanged, while naming for the reusable upstream pool should align with the gateway model from pull request `#2012`: use `UpstreamDefinition` for entries in `upstreamDefinitions`, `UpstreamReference` for the ref-name token, `OperationUpstream` for per-operation overrides, and rename the current endpoint-level `UpstreamDefinition` leaf schema to `UpstreamEndpoint`.

Note: Learnings are effective only in the context of similar code segments. To apply general coding guidelines, please configure review instructions. You can manage existing learnings in the app.

@mehara-rothila

Copy link
Copy Markdown
Contributor Author

@coderabbitai we're renaming the ReusableUpstream schema in platform-api/src/resources/openapi.yaml (the pool entry held by upstreamDefinitions: name, basePath, timeout, weighted upstreams[]). The gateway-controller names this same concept UpstreamDefinition. Note platform-api already uses UpstreamDefinition for a different schema (the url/ref endpoint under Upstream.main/sandbox), so adopting UpstreamDefinition for the pool would rename that endpoint schema to UpstreamEndpoint.

Our candidates:

  1. UpstreamDefinition (matches the gateway pool name; requires endpoint -> UpstreamEndpoint)
  2. ReusableUpstream (current)
  3. NamedUpstream
  4. UpstreamPoolEntry
  5. SharedUpstream

Please (a) add any stronger options, then (b) rank ALL of them best-to-worst with a one-line rationale each. Judge by: cross-plane consistency with the gateway (which calls this UpstreamDefinition), clarity, and fit with the sibling schemas (Upstream, UpstreamReference, OperationUpstream, UpstreamTimeout).

@coderabbitai

coderabbitai Bot commented Jul 8, 2026

Copy link
Copy Markdown
Contributor

Tip

For best results, initiate chat on the files or code changes.

@mehara-rothila Here's the ranking, including two additional candidates for completeness.

Additional candidates to consider:
6. PooledUpstream — conveys "member of a reusable pool" while keeping the two-word pattern used by every sibling schema.
7. UpstreamPool — rejected as a schema name for the entry itself; this name reads as the collection/array type, not a single item, so it would be confusing next to upstreamDefinitions: [UpstreamPool].

Ranking (best to worst):

Rank Name Rationale
1 UpstreamDefinition Exact match with the gateway's pool-entry name; zero cross-plane ambiguity; only cost is the one-time endpoint rename to UpstreamEndpoint, which you've already accepted.
2 NamedUpstream Clear standalone meaning (an upstream identified by a name) and fits the two-word sibling pattern, but has no cross-plane equivalent and slightly overlaps conceptually with UpstreamReference (the name token itself).
3 PooledUpstream Clearly conveys "reusable pool member," consistent two-word style, but still a local-only term with no gateway counterpart.
4 UpstreamPoolEntry Unambiguous and descriptive, but three words breaks the naming rhythm of Upstream / UpstreamReference / OperationUpstream / UpstreamTimeout.
5 ReusableUpstream (current) Accurately describes intent, already implemented, but diverges from gateway terminology, the concern that started this thread.
6 SharedUpstream Weakest option — "shared" is ambiguous (could imply shared across APIs/tenants rather than "one of several named upstreams"), and it doesn't align with the gateway naming either.

Recommendation unchanged: UpstreamDefinition remains the strongest choice given the explicit goal of cross-plane consistency with #2012, and it composes cleanly with UpstreamReference, OperationUpstream, UpstreamTimeout, and the new UpstreamEndpoint.


✏️ Learnings added
Learnt from: mehara-rothila
URL: https://github.com/wso2/api-platform/pull/2155

Timestamp: 2026-07-08T05:19:54.263Z
Learning: In `platform-api/src/resources/openapi.yaml`, when naming the reusable upstream pool-entry schema (currently `ReusableUpstream`, held by `RESTAPI.upstreamDefinitions`), the preferred name is `UpstreamDefinition` to match the gateway-controller's naming in `gateway/gateway-controller/api/management-openapi.yaml` (from PR `#2012`), with the existing endpoint-level `UpstreamDefinition` (used under `Upstream.main`/`Upstream.sandbox`) renamed to `UpstreamEndpoint` to avoid collision. Alternatives considered and ranked below `UpstreamDefinition`: `NamedUpstream`, `PooledUpstream`, `UpstreamPoolEntry`, `ReusableUpstream`, `SharedUpstream`.

Note: Learnings are effective only in the context of similar code segments. To apply general coding guidelines, please configure review instructions. You can manage existing learnings in the app.

@mehara-rothila

Copy link
Copy Markdown
Contributor Author

@coderabbitai for the agreed rename, please show a focused before/after of ONLY the affected schema blocks, not the full file: the Upstream wrapper's main/sandbox refs, the endpoint schema (UpstreamDefinition -> UpstreamEndpoint), the pool schema (ReusableUpstream -> UpstreamDefinition), and the RESTAPI.upstreamDefinitions.items ref. Keep it to those blocks so the concrete diff is easy to review.

@mehara-rothila

Copy link
Copy Markdown
Contributor Author

@coderabbitai full review

This is the control-plane (platform-api) side of per-operation upstream routing. The data-plane contract it must match is in PR #2012 (gateway-controller/pkg/config/api_validator.go and management-openapi.yaml). Please verify each of these and flag any divergence:

  • Validation parity with the gateway validator: upstreamDefinition name pattern/length, duplicate names, backend URL host-only http/https (path, query, or fragment rejected), weight 0-100, connect timeout format + positive + blank-treated-as-unset, API-level url-xor-ref, and that every API-level and per-operation ref resolves to a declared pool entry.
  • Validation runs on create and on the merged result of an update (an update must not be able to remove a pool entry a surviving operation still references), and surfaces as HTTP 400 on create, update, project import, and OpenAPI import.
  • Persistence round-trip through the rest_apis.configuration JSON blob: no field dropped, and rows written before this change still decode.
  • Deployment-YAML emission matches the gateway shape (spec.upstreamDefinitions and operations[].upstream.main/sandbox.ref) and is emitted for REST APIs only, not WebSub.
  • Nil-safety on the generated OperationUpstream main/sandbox pointers, and that api/generated.go is regeneration-only (no hand edits).
  • Whether the added tests actually exercise these paths.

@coderabbitai

coderabbitai Bot commented Jul 11, 2026

Copy link
Copy Markdown
Contributor
🧩 Analysis chain

🏁 Script executed:

#!/bin/bash
set -euo pipefail
echo "== PR 2155 files changed =="
gh pr view 2155 --repo wso2/api-platform --json files --jq '.files[].path'
echo
echo "== PR 2155 diff stat =="
gh pr diff 2155 --repo wso2/api-platform --stat
echo
echo "== Confirm no generated.go / validator / test files touched =="
gh pr diff 2155 --repo wso2/api-platform | rg -n '^diff --git' || true

Length of output: 1675


🏁 Script executed:

#!/bin/bash
set -euo pipefail

echo "== Full diff PR 2155 (patch) =="
gh pr diff 2155 --repo wso2/api-platform --patch > /tmp/pr2155.diff
wc -l /tmp/pr2155.diff

echo
echo "== Show diff per file with line counts =="
rg -n '^diff --git' /tmp/pr2155.diff

echo
echo "== upstream.go full diff =="
awk '/^diff --git a\/platform-api\/src\/internal\/model\/upstream.go/,/^diff --git a\/platform-api\/src\/internal\/(model\/api|repository)/' /tmp/pr2155.diff | head -300

echo
echo "== utils/api.go diff =="
awk '/^diff --git a\/platform-api\/src\/internal\/utils\/api\.go/,/^diff --git a\/platform-api\/src\/internal\/utils\/api_test\.go/' /tmp/pr2155.diff | head -400

Length of output: 18754


🏁 Script executed:

#!/bin/bash
set -euo pipefail

echo "== constants/error.go diff =="
sed -n '353,365p' /tmp/pr2155.diff
awk '/^diff --git a\/platform-api\/src\/internal\/constants\/error\.go/,/^diff --git a\/platform-api\/src\/internal\/dto\/api\.go/' /tmp/pr2155.diff

echo
echo "== dto/api.go diff =="
awk '/^diff --git a\/platform-api\/src\/internal\/dto\/api\.go/,/^diff --git a\/platform-api\/src\/internal\/handler\/api\.go/' /tmp/pr2155.diff

echo
echo "== handler/api.go diff =="
awk '/^diff --git a\/platform-api\/src\/internal\/handler\/api\.go/,/^diff --git a\/platform-api\/src\/internal\/model\/api\.go/' /tmp/pr2155.diff

echo
echo "== model/api.go diff =="
awk '/^diff --git a\/platform-api\/src\/internal\/model\/api\.go/,/^diff --git a\/platform-api\/src\/internal\/model\/upstream\.go/' /tmp/pr2155.diff

echo
echo "== service/api.go diff (validation logic) =="
awk '/^diff --git a\/platform-api\/src\/internal\/service\/api\.go/,/^diff --git a\/platform-api\/src\/internal\/service\/api_test\.go/' /tmp/pr2155.diff

Length of output: 43390


🏁 Script executed:

#!/bin/bash
set -euo pipefail

echo "== PR 2012 files changed =="
gh pr view 2012 --repo wso2/api-platform --json files --jq '.files[].path' || true

echo
echo "== PR 2012 diff for api_validator.go =="
gh pr diff 2012 --repo wso2/api-platform --patch -- gateway/gateway-controller/pkg/config/api_validator.go > /tmp/pr2012_validator.diff || true
wc -l /tmp/pr2012_validator.diff
cat /tmp/pr2012_validator.diff | head -400

echo
echo "== generated.go diff (nil safety on OperationUpstream) =="
awk '/^diff --git a\/platform-api\/src\/api\/generated\.go/,/^diff --git a\/platform-api\/src\/internal\/constants\/error\.go/' /tmp/pr2155.diff | head -200

echo
echo "== repository/api_test.go and service/api_test.go and utils/api_test.go: search upstream-related test names =="
awk '/^diff --git a\/platform-api\/src\/internal\/repository\/api_test\.go/,/^diff --git a\/platform-api\/src\/internal\/service\/api\.go/' /tmp/pr2155.diff | rg -n '^\+func Test|^\+\s*t\.Run' 
awk '/^diff --git a\/platform-api\/src\/internal\/service\/api_test\.go/,/^diff --git a\/platform-api\/src\/internal\/utils\/api\.go/' /tmp/pr2155.diff | rg -n '^\+func Test|^\+\s*\{name:|^\+\s*name:'
awk '/^diff --git a\/platform-api\/src\/internal\/utils\/api_test\.go/,0' /tmp/pr2155.diff | rg -n '^\+func Test|^\+\s*\{name:|^\+\s*name:'

Length of output: 12110


🏁 Script executed:

#!/bin/bash
set -euo pipefail

echo "== PR 2012 head branch =="
head_ref=$(gh pr view 2012 --repo wso2/api-platform --json headRefName --jq .headRefName)
echo "$head_ref"

echo
echo "== Fetch api_validator.go content from PR2012 head =="
gh api repos/wso2/api-platform/contents/gateway/gateway-controller/pkg/config/api_validator.go?ref="$head_ref" --jq .content | base64 -d > /tmp/api_validator.go
wc -l /tmp/api_validator.go
rg -n 'func Validate|upstream|Upstream|weight|Weight|timeout|Timeout|regexp|MustCompile' /tmp/api_validator.go | head -100

echo
echo "== Fetch upstreamref.go content =="
gh api repos/wso2/api-platform/contents/gateway/gateway-controller/pkg/utils/upstreamref/upstreamref.go?ref="$head_ref" --jq .content | base64 -d > /tmp/upstreamref.go
cat /tmp/upstreamref.go

Length of output: 1301


🏁 Script executed:

#!/bin/bash
set -euo pipefail

echo "== Try git fetch branch directly =="
git fetch origin feat/per-op-upstream-gateway:refs/remotes/origin/feat/per-op-upstream-gateway 2>&1 | tail -20 || true
git show origin/feat/per-op-upstream-gateway:gateway/gateway-controller/pkg/config/api_validator.go > /tmp/api_validator.go 2>&1 || echo "FAILED git show validator"
git show origin/feat/per-op-upstream-gateway:gateway/gateway-controller/pkg/utils/upstreamref/upstreamref.go > /tmp/upstreamref.go 2>&1 || echo "FAILED git show upstreamref"
wc -l /tmp/api_validator.go /tmp/upstreamref.go 2>&1 || true

echo
echo "== grep upstream-related validation in api_validator.go =="
rg -n 'upstream|Upstream|weight|Weight|timeout|Timeout|regexp|MustCompile|func ' /tmp/api_validator.go 2>/dev/null | head -150

echo
echo "== upstreamref.go content =="
cat /tmp/upstreamref.go 2>/dev/null

Length of output: 1443


🏁 Script executed:

#!/bin/bash
set -euo pipefail

git fetch origin pull/2012/head:refs/remotes/origin/pr-2012 2>&1 | tail -20
git show origin/pr-2012:gateway/gateway-controller/pkg/config/api_validator.go > /tmp/api_validator.go 2>&1 || echo "FAILED validator"
git show origin/pr-2012:gateway/gateway-controller/pkg/utils/upstreamref/upstreamref.go > /tmp/upstreamref.go 2>&1 || echo "FAILED upstreamref"
wc -l /tmp/api_validator.go /tmp/upstreamref.go

echo
echo "== grep validation logic =="
rg -n 'upstream|Upstream|weight|Weight|timeout|Timeout|regexp|MustCompile|func ' /tmp/api_validator.go 2>/dev/null | head -200

echo
echo "== upstreamref.go =="
cat /tmp/upstreamref.go 2>/dev/null

Length of output: 13864


🏁 Script executed:

#!/bin/bash
set -euo pipefail
sed -n '158,260p' /tmp/api_validator.go

Length of output: 3206


@mehara-rothila Here's the full review against the gateway contract (PR #2012, gateway/gateway-controller/pkg/config/api_validator.go and pkg/utils/upstreamref/upstreamref.go). Note that this PR has grown well past the "schema-only" scope described in its description — it now also touches platform-api/src/api/generated.go, internal/dto/api.go, internal/handler/api.go, internal/model/api.go, internal/model/upstream.go, internal/service/api.go, and three test files, so I reviewed the full implementation rather than just the OpenAPI diff.

1. Validation parity with the gateway validator — Matches closely.

  • Name pattern/length: platform-api/src/internal/service/api.go's upstreamRefNameRe (^[a-zA-Z0-9\-_]+$) and 100-char cap in validateUpstreamReferenceName are identical to the gateway's upstreamRefRegex and validateUpstreamRefName.
  • Duplicate names: both reject via a defined/seen-name map.
  • Backend URL: both require http/https scheme, a non-empty host, and reject a non-root path, query string, or fragment (comments in validateUpstreamRefs explicitly call out matching the gateway validator).
  • Weight: both enforce 0–100.
  • Connect timeout: identical regex ^\d+(\.\d+)?(ms|s|m|h)$, time.ParseDuration for the error message, positive-duration check, and strings.TrimSpace before checking emptiness (blank treated as unset) on both sides.
  • API-level url-xor-ref: validateAPIUpstreamEndpoint mirrors the gateway's validateUpstream (reject both set, require one, then validate).
  • Ref resolution: resolveUpstreamRef against the defined map is functionally equivalent to the gateway's upstreamref.FindByName.

2. Validation on create and merged update, surfaced as 400 — Confirmed.

  • validateCreateAPIRequest calls validateUpstreamRefs on create.
  • applyAPIUpdates re-validates the merged existingAPI (after applying req.Upstream/req.UpstreamDefinitions), so a surviving per-operation ref pointing at a removed pool entry is caught — covered by TestAPIService_UpdateRejectsPoolRemovalWithDanglingRef.
  • handler/api.go maps constants.ErrInvalidUpstreamDefinition to HTTP 400 in CreateAPI, UpdateAPI, ImportAPIProject, and ImportOpenAPI — all four paths are covered.

3. Persistence round-tripmodel.RestAPIConfig.UpstreamDefinitions (in internal/model/api.go) rides through the same JSON configuration blob as the other nested config fields (db:"-" tag), and TestAPIRepo_CreateAndRead_PreservesUpstreamDefinitionsAndPerOp in internal/repository/api_test.go exercises the round trip. I didn't find a decode test specifically for rows written before this change; worth double-checking that an existing JSON blob without upstreamDefinitions/operation upstream still unmarshals cleanly (both are optional/omitempty, so it should, but an explicit regression test would be good defensive coverage).

4. Deployment YAML emission — Matches the gateway shape. BuildAPIDeploymentYAML only sets apiYAMLData.UpstreamDefinitions inside case constants.RestApi:, and TestBuildAPIDeploymentYAML_WebSubOmitsUpstreamDefinitions confirms WebSub is excluded. spec.upstreamDefinitions (dto.UpstreamDefinitionYAML: name/basePath/timeout/upstreams[url,weight]) and operations[].upstream.main/sandbox.ref are both emitted and covered by TestBuildAPIDeploymentYAML_EmitsUpstreamDefinitions, TestBuildAPIDeploymentYAML_EmitsPerOpUpstreamRef, and TestBuildAPIDeploymentYAML_FullPerOpRefContract.

5. Nil-safety and generated.gooperationUpstreamAPIToModel/operationUpstreamModelToAPI/overlayPerOpUpstream all guard Main/Sandbox with nil checks before dereferencing. api/generated.go's diff is limited to schema-driven additions (ReusableUpstream, UpstreamTimeout, UpstreamReference type alias, OperationUpstream, RESTAPI.UpstreamDefinitions, retyping UpstreamDefinition.Ref to *UpstreamReference) — no hand-written logic; the anonymous-struct constructor helpers (newAPIOperationUpstreamTarget, newAPIUpstreamBackend) correctly live in internal/utils/api.go, not in generated.go.

6. Tests — Yes, they exercise the intended paths: repository round-trip, validateUpstreamRefs unit cases, create-rejects-unresolved-ref, update-rejects-pool-removal-with-dangling-ref, YAML emission (including the WebSub-omission case and the full per-op ref contract), and merge/overlay behavior for per-operation upstream overrides during OpenAPI/project import.

One open item from earlier in this thread: the agreed rename (ReusableUpstreamUpstreamDefinition, endpoint-level UpstreamDefinitionUpstreamEndpoint) has not been applied yet — openapi.yaml, generated.go, and model/upstream.go still use the old names. Now that PR #2012 has landed UpstreamDefinition (pool) and UpstreamReference (type alias) on the gateway side, this is the natural point to apply that rename before this branch grows further, since every file above (dto, model, service, utils, and their tests) would need the same symbol renames.

✅ Action performed

Review finished.

Note: CodeRabbit is an incremental review system and does not re-review already reviewed commits. This command is applicable only when automatic reviews are paused.

@coderabbitai coderabbitai Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 4

🧹 Nitpick comments (3)
platform-api/src/internal/utils/api_test.go (1)

672-679: 📐 Maintainability & Code Quality | 🔵 Trivial | 🏗️ Heavy lift

Apply the agreed upstream type rename.

These tests continue to encode ReusableUpstream; rename the reusable pool entry to UpstreamDefinition and reserve UpstreamEndpoint for direct endpoint shapes across the schema, generated code, models, and tests. As noted in the PR objectives, this is the agreed naming model.

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@platform-api/src/internal/utils/api_test.go` around lines 672 - 679, Rename
the reusable pool type represented by UpstreamDefinitions and its entries from
ReusableUpstream to UpstreamDefinition across the schema, generated code,
models, and tests. Reserve UpstreamEndpoint for direct endpoint shapes, updating
all references and generated artifacts consistently while preserving the
existing behavior and structure.
platform-api/src/internal/handler/api.go (1)

162-163: 📐 Maintainability & Code Quality | 🔵 Trivial | ⚡ Quick win

Use Warn for expected 400-level validation failures.

Invalid upstream definitions are client-input errors, not unexpected server failures. Reserve Error for 5xx conditions and use Warn here.

Based on learnings, expected 4xx handler errors should use slog.Warn, while slog.Error is reserved for unexpected server-side failures.

Also applies to: 319-320

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@platform-api/src/internal/handler/api.go` around lines 162 - 163, Change the
logging call for constants.ErrInvalidUpstreamDefinition in the handler to
h.slogger.Warn, preserving the existing message and structured fields. Apply the
same severity change to the corresponding handler branch around lines 319-320,
while leaving unexpected server-side failures logged at Error.

Source: Learnings

platform-api/src/internal/utils/api.go (1)

1539-1545: 📐 Maintainability & Code Quality | 🔵 Trivial | ⚡ Quick win

Canonicalize methods before constructing lookup keys.

Normalize the method once at the conversion boundary and use that canonical value throughout merging. The current inline normalization only affects the temporary key; the underlying operation can retain user-supplied casing.

As per coding guidelines, user-supplied HTTP methods must be uppercased before map-key creation and downstream route processing.

Also applies to: 1552-1554

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@platform-api/src/internal/utils/api.go` around lines 1539 - 1545, In the
override-operation merge logic around byKey, uppercase each user-supplied
op.Request.Method at the conversion boundary before constructing lookup keys,
and use that canonical method for downstream route processing as well. Update
the corresponding logic at the additional referenced range so the underlying
operation no longer retains mixed-case methods while preserving the existing
upstream lookup behavior.

Source: Coding guidelines

🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@platform-api/src/internal/handler/api.go`:
- Around line 162-166: Sanitize client responses in the
invalid-upstream-definition branches, including the corresponding occurrences
near lines 319, 702, and 925: keep logging the detailed err server-side, but
replace err.Error() passed to utils.NewErrorResponse with the fixed message
“Invalid upstream definition”.

In `@platform-api/src/internal/service/api_test.go`:
- Around line 844-846: Update the schema fixture setup around runtime.Caller and
schemaPath to resolve the path to an absolute, cleaned value, then verify it
remains within the intended test root before calling os.ReadFile. Reject or fail
the test for paths outside that root while preserving the existing
schema-loading behavior for valid paths.

In `@platform-api/src/internal/service/api.go`:
- Around line 694-701: Update resolveUpstreamRef to reject whitespace-padded
references or consistently normalize the reference before it is persisted and
emitted; ensure validation, resolution, and deployment use the same canonical
value rather than only validating strings.TrimSpace(ref).
- Around line 723-731: Update both URL validation paths around url.Parse to
validate parsed.Hostname() rather than only parsed.Host, rejecting empty
hostnames such as http://:8080. Preserve the existing http/https scheme
requirement, and only accept a port when a non-empty hostname is present.

---

Nitpick comments:
In `@platform-api/src/internal/handler/api.go`:
- Around line 162-163: Change the logging call for
constants.ErrInvalidUpstreamDefinition in the handler to h.slogger.Warn,
preserving the existing message and structured fields. Apply the same severity
change to the corresponding handler branch around lines 319-320, while leaving
unexpected server-side failures logged at Error.

In `@platform-api/src/internal/utils/api_test.go`:
- Around line 672-679: Rename the reusable pool type represented by
UpstreamDefinitions and its entries from ReusableUpstream to UpstreamDefinition
across the schema, generated code, models, and tests. Reserve UpstreamEndpoint
for direct endpoint shapes, updating all references and generated artifacts
consistently while preserving the existing behavior and structure.

In `@platform-api/src/internal/utils/api.go`:
- Around line 1539-1545: In the override-operation merge logic around byKey,
uppercase each user-supplied op.Request.Method at the conversion boundary before
constructing lookup keys, and use that canonical method for downstream route
processing as well. Update the corresponding logic at the additional referenced
range so the underlying operation no longer retains mixed-case methods while
preserving the existing upstream lookup behavior.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: CHILL

Plan: Pro

Run ID: 9aa52fdf-c292-4869-ad6a-295d40587ab3

📥 Commits

Reviewing files that changed from the base of the PR and between e6e8d18 and 13d47a5.

📒 Files selected for processing (12)
  • platform-api/src/api/generated.go
  • platform-api/src/internal/constants/error.go
  • platform-api/src/internal/dto/api.go
  • platform-api/src/internal/handler/api.go
  • platform-api/src/internal/model/api.go
  • platform-api/src/internal/model/upstream.go
  • platform-api/src/internal/repository/api_test.go
  • platform-api/src/internal/service/api.go
  • platform-api/src/internal/service/api_test.go
  • platform-api/src/internal/utils/api.go
  • platform-api/src/internal/utils/api_test.go
  • platform-api/src/resources/openapi.yaml
✅ Files skipped from review due to trivial changes (1)
  • platform-api/src/api/generated.go

Comment thread platform-api/src/internal/handler/api.go Outdated
Comment thread platform-api/src/internal/service/api_test.go Outdated
Comment thread platform-api/src/internal/service/api.go Outdated
Comment thread platform-api/src/internal/service/api.go Outdated
@mehara-rothila

Copy link
Copy Markdown
Contributor Author

@coderabbitai we've decided to keep the pool entry schema named ReusableUpstream and are not renaming it to UpstreamDefinition. My earlier comment that called the rename "agreed" was premature; on reflection we are not doing it in this PR.

Reasoning: adopting UpstreamDefinition for the pool would also force renaming the pre-existing UpstreamDefinition endpoint schema (the url/ref endpoint under Upstream.main/sandbox) to UpstreamEndpoint, which churns a large amount of pre-existing code (dto, model, service, utils, and their tests) that is unrelated to this feature. This PR is scoped to adding the upstreamDefinitions pool and per-operation upstream overrides, and we want to keep that scope tight. The schema shapes already match the gateway contract in #2012; only the pool entry name differs, and ReusableUpstream reads clearly for a reusable named entry in the pool.

Please note this decision and do not re-flag the ReusableUpstream -> UpstreamDefinition rename in future reviews of this PR. If the maintainers prefer the rename, we will do it as a separate follow-up.

@coderabbitai

coderabbitai Bot commented Jul 11, 2026

Copy link
Copy Markdown
Contributor

Tip

For best results, initiate chat on the files or code changes.

@mehara-rothila Understood — noted. I will not re-flag the ReusableUpstreamUpstreamDefinition rename in future reviews of this PR. Keeping the pool-entry schema as ReusableUpstream and deferring the rename (along with the associated UpstreamDefinitionUpstreamEndpoint change) to a separate follow-up is a reasonable way to keep this PR's scope focused on adding the upstreamDefinitions pool and per-operation upstream overrides, especially since the wire-level schema shapes already match the gateway contract in #2012 and only the pool-entry symbol name differs.


✏️ Learnings added
Learnt from: mehara-rothila
URL: https://github.com/wso2/api-platform/pull/2155

Timestamp: 2026-07-11T22:44:52.702Z
Learning: In `platform-api/src/resources/openapi.yaml` (PR `#2155`), the team decided NOT to rename the reusable upstream pool-entry schema `ReusableUpstream` to `UpstreamDefinition` within this PR, reversing an earlier tentative agreement. Rationale: adopting `UpstreamDefinition` for the pool would force renaming the pre-existing endpoint-level `UpstreamDefinition` schema (used under `Upstream.main`/`Upstream.sandbox`) to `UpstreamEndpoint`, causing unrelated churn across `internal/dto`, `internal/model`, `internal/service`, `internal/utils`, and their tests. The schema shapes already match the gateway contract in PR `#2012`; only the pool-entry name differs. Do not re-flag this rename suggestion in future reviews of this PR; it may be done as a separate follow-up PR if maintainers prefer.

Note: Learnings are effective only in the context of similar code segments. To apply general coding guidelines, please configure review instructions. You can manage existing learnings in the app.

@mehara-rothila mehara-rothila changed the base branch from feature/operation-level-ep to main July 12, 2026 04:38
@mehara-rothila mehara-rothila force-pushed the feat/platform-api-upstream-definitions branch from 13d47a5 to 47ed679 Compare July 12, 2026 04:38
@mehara-rothila

Copy link
Copy Markdown
Contributor Author

@coderabbitai review

@coderabbitai

coderabbitai Bot commented Jul 12, 2026

Copy link
Copy Markdown
Contributor
✅ Action performed

Review finished.

Note: CodeRabbit is an incremental review system and does not re-review already reviewed commits. This command is applicable only when automatic reviews are paused.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants