Browse Source

[ruby/roda-sequel] Use default_headers plugin (#10409)

+-------------------------------+---------------------------+-------+
|                           name|                branch_name|fortune|
+-------------------------------+---------------------------+-------+
|roda-sequel-postgres-iodine-mri|                     master|  58339|
|roda-sequel-postgres-iodine-mri|roda-sequel/default-headers|  62333|
+-------------------------------+---------------------------+-------+
Petrik de Heus 3 days ago
parent
commit
9d03045cb0
1 changed files with 2 additions and 12 deletions
  1. 2 12
      frameworks/Ruby/roda-sequel/hello_world.rb

+ 2 - 12
frameworks/Ruby/roda-sequel/hello_world.rb

@@ -4,6 +4,7 @@
 class HelloWorld < Roda
   plugin :hooks
   plugin :render, escape: true, layout_opts: { cache_key: "default_layout" }
+  plugin :default_headers, SERVER_HEADER => SERVER_STRING
 
   def bounded_queries
     queries = request.params["queries"].to_i
@@ -15,19 +16,8 @@ class HelloWorld < Roda
     rand(MAX_PK) + 1
   end
 
-  if defined?(Puma)
-    def set_default_headers(response)
-      response[DATE_HEADER] = Time.now.httpdate
-      response[SERVER_HEADER] = SERVER_STRING
-    end
-  else
-    def set_default_headers(response)
-      response[SERVER_HEADER] = SERVER_STRING
-    end
-  end
-
   route do |r|
-    set_default_headers(response)
+    response[DATE_HEADER] = Time.now.httpdate if defined?(Puma)
 
     # Test type 1: JSON serialization
     r.is "json" do