lib/basic_serializer/swagger.rb in basic_serializer-0.1.7 vs lib/basic_serializer/swagger.rb in basic_serializer-0.1.8

@@ -1,22 +1,28 @@
 # frozen_string_literal: true
 
 class BasicSerializer
   module Swagger
     def swagger_ref
-      @schema_ref || "#/components/schemas/#{custom_model_name}"
+      @swagger_ref ||= @schema_ref || "#/components/schemas/#{custom_model_name}"
     end
 
     def swagger_schema
-      hash = {
+      @swagger_schema ||= build_swagger_schema
+    end
+
+    private
+
+    def build_swagger_schema
+      {
         type: "object",
-        properties: {}
+        properties: build_properties
       }
+    end
 
-      attributes.each_pair do |name, type|
-        hash[:properties][name] = { type: type }
+    def build_properties
+      attributes.transform_values do |type|
+        { type: type }
       end
-
-      hash
     end
   end
 end