lib/active_finite.rb in active_finite-0.1.0 vs lib/active_finite.rb in active_finite-0.1.1

@@ -4,12 +4,11 @@
 require 'active_record'
 require 'json'
 
 def active_finite sym
   anon = Class.new ActiveRecord::Base
-  klass = Object.const_set(sym.to_s.singularize.capitalize, anon)
-  klass
+  Object.const_set(sym.to_s.singularize.capitalize, anon)
 end
 
 def create_finite args
   modify_finite args do |vs, klass, column_name|
     vs.each do |v|
@@ -51,10 +50,12 @@
   if not file_name.nil?
     to_be_modified = to_be_modified.concat JSON.load open file_name
   end
 
   klass = active_finite table_name
-  yield to_be_modified, klass, column_name
+  ActiveRecord::Base.transaction do
+    yield to_be_modified, klass, column_name
+  end
 end
 
 def default_column_name
   :value
 end