Skip to content

Commit

Permalink
feat: implemented setting cache for can-i-merge badges based on default
Browse files Browse the repository at this point in the history
configuration
  • Loading branch information
Phanindra Srungavarapu authored and Phanindra Srungavarapu committed Sep 9, 2024
1 parent 7588de6 commit e2bce55
Show file tree
Hide file tree
Showing 6 changed files with 19 additions and 1 deletion.
4 changes: 4 additions & 0 deletions docs/configuration.yml
Original file line number Diff line number Diff line change
Expand Up @@ -355,6 +355,10 @@ groups:
description: The URL of the shields.io server used to generate the README badges.
default_value: https://img.shields.io
more_info: https://shields.io
badge_default_cache_setting:
description: Cache Control header value for the badge, this sets the Cache-Control header for the badge image, for more information on header formats see https://developer.mozilla.org/en-US/docs/Web/HTTP/Headers/Cache-Control
default_value: "max-age=30"

badge_provider_mode:
description: |-
The method by which the badges are generated. When set to `redirect`, a request to the Pact Broker for a badge will be sent a redirect response
Expand Down
1 change: 1 addition & 0 deletions example/config/pact_broker.yml
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,4 @@ basic_auth_password: "pass"
basic_auth_read_only_username: "rouser"
basic_auth_read_only_password: "ropass"
database_url: sqlite:////tmp/pact_broker.sqlite3
badge_default_cache_setting: "max-age=30"
6 changes: 5 additions & 1 deletion lib/pact_broker/api/resources/badge_methods.rb
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,7 @@ def policy_name
end

def moved_temporarily?
response.headers["Cache-Control"] = "no-cache"
set_cache_control("no-cache")
begin
badge_url
rescue StandardError => e
Expand All @@ -46,6 +46,10 @@ def badge_url
raise NotImplementedError
end

def set_cache_control(cache_control_str)
response.headers["Cache-Control"] = cache_control_str
end

private

def label
Expand Down
6 changes: 6 additions & 0 deletions lib/pact_broker/api/resources/can_i_merge_badge.rb
Original file line number Diff line number Diff line change
Expand Up @@ -15,13 +15,19 @@ def badge_url
# when there is no main branch version, we return an error badge url
badge_service.error_badge_url("main branch version", "not found")
else
# when badge is available, set cache based on configuration
set_cache_control(default_cache_for_succesful_badge)
# we call badge_service to build the badge url
badge_service.can_i_merge_badge_url(deployable: results)
end
end

private

def default_cache_for_succesful_badge
PactBroker.configuration.badge_default_cache_setting
end

def results
# can_i_merge returns true or false if the main branch version is compatible with all the integrations
@results ||= matrix_service.can_i_merge(pacticipant: pacticipant, latest_main_branch_version: version)
Expand Down
1 change: 1 addition & 0 deletions lib/pact_broker/config/runtime_configuration.rb
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,7 @@ class RuntimeConfiguration < Anyway::Config
badge_provider_mode: :redirect,
enable_public_badge_access: false,
shields_io_base_url: "https://img.shields.io",
badge_default_cache_setting: "max-age=30",
use_case_sensitive_resource_names: true,
pact_content_diff_timeout: 15
)
Expand Down
2 changes: 2 additions & 0 deletions spec/lib/pact_broker/api/resources/can_i_merge_badge_spec.rb
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@ module Resources
it "return the badge URL" do
expect(badge_service). to receive(:can_i_merge_badge_url).with(deployable: true)
expect(subject.headers["Location"]).to eq "http://badge_url"
expect(subject.headers["Cache-Control"]).to eq "max-age=30"
end
end

Expand All @@ -44,6 +45,7 @@ module Resources
it "returns an error badge URL" do
expect(badge_service).to receive(:error_badge_url).with("pacticipant", "not found")
expect(subject.headers["Location"]).to eq "http://error_badge_url"
expect(subject.headers["Cache-Control"]).to eq "no-cache"
end
end

Expand Down

0 comments on commit e2bce55

Please sign in to comment.