diff options
Diffstat (limited to 'lib/gitlab_keys.rb')
| -rw-r--r-- | lib/gitlab_keys.rb | 31 |
1 files changed, 20 insertions, 11 deletions
diff --git a/lib/gitlab_keys.rb b/lib/gitlab_keys.rb index 0eb04eb..7a93b09 100644 --- a/lib/gitlab_keys.rb +++ b/lib/gitlab_keys.rb @@ -2,6 +2,7 @@ require 'timeout' require_relative 'gitlab_config' require_relative 'gitlab_logger' +require_relative 'gitlab_metrics' class GitlabKeys class KeyError < StandardError ; end @@ -28,17 +29,25 @@ class GitlabKeys end def exec - case @command - when 'add-key'; add_key - when 'batch-add-keys'; batch_add_keys - when 'rm-key'; rm_key - when 'list-keys'; puts list_keys - when 'clear'; clear - when 'check-permissions'; check_permissions - else - $logger.warn "Attempt to execute invalid gitlab-keys command #{@command.inspect}." - puts 'not allowed' - false + GitlabMetrics.measure("command-#{@command.inspect}") do + case @command + when 'add-key'; + add_key + when 'batch-add-keys'; + batch_add_keys + when 'rm-key'; + rm_key + when 'list-keys'; + list_keys + when 'clear'; + clear + when 'check-permissions'; + check_permissions + else + $logger.warn "Attempt to execute invalid gitlab-keys command #{@command.inspect}." + puts 'not allowed' + false + end end end |