Added session_id to Query since it was not scalable to store the AR…

… ids in the cookie session.

app/controllers/application_controller.rb

@@ -1,13 +1,6 @@
 class ApplicationController < ActionController::Base
   # Only allow modern browsers supporting webp images, web push, badges, import maps, CSS nesting, and CSS :has.
-  # allow_browser versions: :modern
+  allow_browser versions: :modern
 
   default_form_builder ApplicationFormBuilder
-
-  helper_method :my_queries
-
-  def my_queries
-    Query.where(id: session[:queries]).order(created_at: :desc)
-  end
-
 end

app/controllers/queries_controller.rb

@@ -3,27 +3,26 @@ class QueriesController < ApplicationController
 
   def index
     @queries = if params[:filter] == 'all'
-      Query.all.order(created_at: :desc)
+      Query.all
     else
-      my_queries
-    end
+      Query.where(session_id: session.id.to_s)
+    end.order(created_at: :desc)
   end
 
   def show
   end
 
   def create
     @query = Query.new(query_params)
-    session[:last_query_type] = @query.type
+    @query.session_id = session.id.to_s
 
     if @query.valid?
       dns_lookup = DNSLookup.new(@query.domain, @query.type, @query.server_ip)
       @query.results = dns_lookup.run
       @query.duration = dns_lookup.duration
       @query.save
 
-      session[:queries] ||= []
-      session[:queries].push(@query.id)
+      session[:last_query_type] = @query.type
 
       redirect_to @query
     else

app/models/query.rb

@@ -11,6 +11,7 @@
 #  type       :string
 #  created_at :datetime         not null
 #  updated_at :datetime         not null
+#  session_id :string
 #
 class Query < ApplicationRecord
   self.inheritance_column = nil

app/views/queries/show.html.erb

@@ -33,7 +33,7 @@
   </div>
 </div>
 <div class="flex p-4 text-xs text-neutral-600 justify-between">
-  <% if my_queries.include?(@query) %>
+  <% if @query.session_id == session.id.to_s %>
     <%= button_to "Remove", query_path(@query), method: :delete,
       form: { data: { turbo_confirm: "Are you sure?" } },
       class: 'hover:underline text-neutral-600 transition hover:text-white' %>

db/migrate/20240212142204_add_session_id_to_query.rb

@@ -0,0 +1,5 @@
+class AddSessionIdToQuery < ActiveRecord::Migration[7.2]
+  def change
+    add_column :queries, :session_id, :string
+  end
+end

test/fixtures/queries.yml

@@ -11,6 +11,7 @@
 #  type       :string
 #  created_at :datetime         not null
 #  updated_at :datetime         not null
+#  session_id :string
 #
 
 one:

test/models/query_test.rb

@@ -11,6 +11,7 @@
 #  type       :string
 #  created_at :datetime         not null
 #  updated_at :datetime         not null
+#  session_id :string
 #
 require "test_helper"