jdbc-helper 0.4.1 → 0.4.2

data/lib/jdbc-helper/connection.rb

@@ -12,6 +12,7 @@ require 'jdbc-helper/connection/row'
 require 'jdbc-helper/wrapper/object_wrapper'
 require 'jdbc-helper/wrapper/table_wrapper'
 require 'jdbc-helper/wrapper/table_wrapper'
+require 'jdbc-helper/wrapper/sequence_wrapper'
 require 'jdbc-helper/wrapper/function_wrapper'
 require 'jdbc-helper/wrapper/procedure_wrapper'
 
@@ -353,6 +354,14 @@ class Connection
 		JDBCHelper::TableWrapper.new self, table_name
 	end
 
+	# Returns a sequence wrapper for the given name
+	# @since 0.4.2
+	# @param [String/Symbol] sequence_name Name of the sequence to be wrapped
+	# @return [JDBCHelper::SequenceWrapper]
+	def sequence sequence_name
+		JDBCHelper::SequenceWrapper.new self, sequence_name
+	end
+
 	# Returns a function wrapper for the given function name
 	# @since 0.2.2
 	# @param [String/Symbol] func_name Name of the function to be wrapped

data/lib/jdbc-helper/wrapper/sequence_wrapper.rb

@@ -0,0 +1,60 @@
+# encoding: UTF-8
+# Junegunn Choi (junegunn.c@gmail.com)
+
+module JDBCHelper
+# A wrapper object representing a database sequence.
+# @since 0.4.2
+# @example Usage
+#  conn.sequence(:my_seq).create!(1)
+#  conn.sequence(:my_seq).nextval
+#  conn.sequence(:my_seq).currval
+#  conn.sequence(:my_seq).reset!(1)
+#  conn.sequence(:my_seq).drop!
+class SequenceWrapper < ObjectWrapper
+	# Returns the name of the sequence
+	# @return [String]
+	alias to_s name
+
+	# Increments the sequence and returns the value
+	# @return [Fixnum]
+	def nextval
+		@connection.query("select #{name}.nextval from dual")[0][0].to_i
+	end
+
+	# Returns the incremented value of the sequence
+	# @return [Fixnum]
+	def currval
+		@connection.query("select #{name}.currval from dual")[0][0].to_i
+	end
+
+	# Recreates the sequence. Cannot be undone.
+	# @param [Fixnum] value Initial value
+	# @param [Fixnum] increment_by Increment size for a single nextval call
+	# @return [JDBCHelper::SequenceWrapper] Self.
+	def reset! value = 1, increment_by = 1
+		drop! rescue nil
+		create! value, increment_by
+		self
+	end
+
+	# Drops the sequence. Cannot be undone.
+	# @return [JDBCHelper::SequenceWrapper] Self.
+	def drop!
+		@connection.update("drop sequence #{name}")
+		self
+	end
+
+	# Creates the sequence. Cannot be undone.
+	# @param [Fixnum] value Initial value
+	# @param [Fixnum] increment_by Increment size for a single nextval call
+	# @return [JDBCHelper::SequenceWrapper] Self.
+	def create! value = 1, increment_by = 1
+		create = JDBCHelper::SQL.check(
+				"create sequence #{name} start with #{value} increment by #{increment_by}")
+		@connection.update(create)
+		self
+	end
+end#SequenceWrapper
+end#JDBCHelper
+
+

data/lib/jdbc-helper/wrapper/table_wrapper.rb

@@ -103,17 +103,21 @@ class TableWrapper < ObjectWrapper
 
 	# Empties the table.
 	# @note This operation cannot be undone
-	# @return [Fixnum] executeUpdate return value
-	def truncate_table!
+	# @return [JDBCHelper::TableWrapper] Self.
+	def truncate!
 		@connection.update(JDBCHelper::SQL.check "truncate table #{name}")
+		self
 	end
+	alias truncate_table! truncate!
 	
 	# Drops the table.
 	# @note This operation cannot be undone
-	# @return [Fixnum] executeUpdate return value
-	def drop_table!
+	# @return [JDBCHelper::TableWrapper] Self.
+	def drop!
 		@connection.update(JDBCHelper::SQL.check "drop table #{name}")
+		self
 	end
+	alias drop_table! truncate!
 
 	# Select SQL wrapper
 	include Enumerable

data/test/database.yml

@@ -5,9 +5,9 @@ mysql:
     user: root
     password:
 
-#oracle:
-    #driver: oracle.jdbc.driver.OracleDriver
-    #url: jdbc:oracle:thin:@localhost/test
-    #user: testuser
-    #password: testpassword
+oracle:
+    driver: oracle.jdbc.driver.OracleDriver
+    url: jdbc:oracle:thin:@10.20.253.105/didev
+    user: wisefn
+    password: wisefndb
 

data/test/test_object_wrapper.rb

@@ -339,7 +339,7 @@ class TestObjectWrapper < Test::Unit::TestCase
 			table = conn.table(@table_name)
 			insert table
 
-			table.truncate_table!
+			table.truncate!
 			assert table.empty?
 		end
 	end
@@ -348,10 +348,33 @@ class TestObjectWrapper < Test::Unit::TestCase
 		each_connection do |conn|
 			create_table conn
 			table = conn.table(@table_name)
-			table.drop_table!
+			table.drop!
 
 			assert_equal false, drop_table(conn)
 		end
 	end
+
+	def test_sequence
+		each_connection do |conn|
+			# MySQL doesn't support sequences
+			next if @type == :mysql
+
+			seq = conn.sequence(@table_name + '_seq')
+			seq.reset!(100)
+			assert (prev = seq.nextval) >= 100
+			assert_equal prev, seq.currval
+			assert_equal 1, seq.nextval - prev
+
+			seq.reset! 1, 2
+			assert seq.nextval >= 1
+			assert seq.nextval <= 4
+			assert seq.nextval >= 5
+
+			seq.drop!
+			seq.create!(10)
+			assert seq.nextval >= 10
+			seq.drop!
+		end
+	end
 end
 

metadata

@@ -2,7 +2,7 @@
 name: jdbc-helper
 version: !ruby/object:Gem::Version 
   prerelease: 
-  version: 0.4.1
+  version: 0.4.2
 platform: ruby
 authors: 
   - Junegunn Choi
@@ -10,7 +10,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2011-06-01 00:00:00 +09:00
+date: 2011-06-04 00:00:00 +09:00
 default_executable: 
 dependencies: 
   - !ruby/object:Gem::Dependency 
@@ -84,6 +84,7 @@ files:
   - lib/jdbc-helper/wrapper/function_wrapper.rb
   - lib/jdbc-helper/wrapper/object_wrapper.rb
   - lib/jdbc-helper/wrapper/procedure_wrapper.rb
+  - lib/jdbc-helper/wrapper/sequence_wrapper.rb
   - lib/jdbc-helper/wrapper/table_wrapper.rb
   - LICENSE.txt
   - README.rdoc