Use format option

ruby-grape · Nov 22, 2024 · e7a8ace · e7a8ace
1 parent c54ed8f
commit e7a8ace
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/lib/grape-swagger/endpoint.rb b/lib/grape-swagger/endpoint.rb
@@ -311,12 +311,14 @@ def build_memo_schema(memo, route, value, response_model, options)
     end
 
     def build_primitive_response(memo, _route, value, _options)
-      type = GrapeSwagger::DocMethods::DataType.call(value[:type])
+      data_type = GrapeSwagger::DocMethods::DataType.call(value[:type])
+      type, format = GrapeSwagger::DocMethods::DataType.mapping(data_type)
 
       if memo[value[:code]].include?(:schema) && value.include?(:as)
-        memo[value[:code]][:schema][:properties].merge!(value[:as] => { type: type })
+        memo[value[:code]][:schema][:properties].merge!(value[:as] => { type: type, format: format }.compact)
       elsif value.include?(:as)
-        memo[value[:code]][:schema] = { type: :object, properties: { value[:as] => { type: type } } }
+        memo[value[:code]][:schema] =
+          { type: :object, properties: { value[:as] => { type: type, format: format }.compact } }
       else
         memo[value[:code]][:schema] = { type: type }
       end

diff --git a/spec/swagger_v2/api_swagger_v2_response_with_models_and_primitive_types_spec.rb b/spec/swagger_v2/api_swagger_v2_response_with_models_and_primitive_types_spec.rb
@@ -14,7 +14,8 @@ class ResponseApiModelsAndPrimitiveTypes < Grape::API
              success: [
                { type: 'Integer', as: :integer_response },
                { model: Entities::UseResponse, as: :user_response },
-               { type: 'String', as: :string_response }
+               { type: 'String', as: :string_response },
+               { type: 'Float', as: :float_response }
              ],
              failure: [
                { code: 400, message: 'NotFound', model: '' },
@@ -51,8 +52,9 @@ def app
               'type' => 'object',
               'properties' => {
                 'user_response' => { '$ref' => '#/definitions/UseResponse' },
-                'integer_response' => { 'type' => 'integer' },
-                'string_response' => { 'type' => 'string' }
+                'integer_response' => { 'type' => 'integer', 'format' => 'int32' },
+                'string_response' => { 'type' => 'string' },
+                'float_response' => { 'type' => 'number', 'format' => 'float' }
               }
             }
           },