From 5457c94f0d1e3537b3cbe398d05257e9f3485782 Mon Sep 17 00:00:00 2001
From: yuua <mumin.nonnon0216@gmail.com>
Date: Fri, 13 Nov 2015 17:07:16 +0900
Subject: [PATCH] add database use

---
 lib/rbhive/t_c_l_i_connection.rb | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lib/rbhive/t_c_l_i_connection.rb b/lib/rbhive/t_c_l_i_connection.rb
index ab795c5..5f899ea 100644
--- a/lib/rbhive/t_c_l_i_connection.rb
+++ b/lib/rbhive/t_c_l_i_connection.rb
@@ -54,6 +54,8 @@ def tcli_connect(server, port = 10_000, options)
     begin
       connection.open
       connection.open_session
+      options[:database] ||= 'default'
+      connection.execute("USE #{options[:database]}")
       ret = yield(connection)
 
     ensure