Fix key names in statsd

diff --git a/config/initializers/statsd.rb b/config/initializers/statsd.rb
index 4e772a5edc714fc5d74a9f5895744853c534b375..f00b1d4015ce63fcd6a5375d8477ecca11693857 100644 (file)
--- a/config/initializers/statsd.rb
+++ b/config/initializers/statsd.rb
@@ -1,13 +1,18 @@
 # frozen_string_literal: true
+RESERVED_CHARACTERS_REGEX = /[\:\|\@]/
 
 StatsD.prefix              = 'mastodon'
 StatsD.default_sample_rate = 1
 
+def clean_name(str)
+  str.gsub('::', '.').gsub(RESERVED_CHARACTERS_REGEX, '_')
+end
+
 ActiveSupport::Notifications.subscribe(/performance/) do |name, _start, _finish, _id, payload|
   action      = payload[:action] || :increment
   measurement = payload[:measurement]
   value       = payload[:value]
-  key_name    = "#{name}.#{measurement}"
+  key_name    = clean_name("#{name}.#{measurement}")
 
   StatsD.send(action.to_s, key_name, (value || 1))
 end