couchbase-jruby-client 0.1.0-java → 0.1.5-java

data/lib/couchbase/operations/arithmetic.rb

@@ -1,3 +1,20 @@
+# Author:: Mike Evans <mike@urlgonomics.com>
+# Copyright:: 2013 Urlgonomics LLC.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 module Couchbase::Operations
   module Arithmetic
 

data/lib/couchbase/operations/delete.rb

@@ -1,3 +1,20 @@
+# Author:: Mike Evans <mike@urlgonomics.com>
+# Copyright:: 2013 Urlgonomics LLC.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 module Couchbase::Operations
   module Delete
 

data/lib/couchbase/operations/design_docs.rb

@@ -0,0 +1,99 @@
+# Author:: Mike Evans <mike@urlgonomics.com>
+# Copyright:: 2013 Urlgonomics LLC.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the 'License');
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an 'AS IS' BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+module Couchbase::Operations
+  module DesignDocs
+
+    java_import com.couchbase.client.protocol.views.DesignDocument
+    java_import com.couchbase.client.protocol.views.ViewDesign
+
+    class DesignDocAccess
+      def initialize(bucket)
+        @bucket = bucket
+      end
+
+      def [](name)
+        doc = @bucket.client.getDesignDocument(name)
+        Couchbase::DesignDoc.new(@bucket, doc)
+      rescue Java::ComCouchbaseClientProtocolViews::InvalidViewException
+        nil
+      end
+    end
+
+    # Fetch design docs stored in current bucket
+    #
+    # @since 1.2.0
+    #
+    # @return [Hash]
+    def design_docs
+      DesignDocAccess.new(self)
+    end
+
+    # Update or create design doc with supplied views
+    #
+    # @since 1.2.0
+    #
+    # @param [Hash, IO, String] data The source object containing JSON
+    #   encoded design document. It must have +_id+ key set, this key
+    #   should start with +_design/+.
+    #
+    # @return [true, false]
+    def save_design_doc(data)
+      attrs = case data
+              when String
+                MultiJson.load(data)
+              when IO
+                MultiJson.load(data.read)
+              when Hash
+                data
+              else
+                raise ArgumentError, "Document should be Hash, String or IO instance"
+              end
+
+      id = attrs.delete('_id').to_s.split('/')[1]
+
+      design_doc = DesignDocument.new(id)
+
+      attrs['views'].each_pair do |view, functions|
+        view_design = if functions['reduce']
+                        ViewDesign.new(view.to_s.to_java_string, functions['map'].to_java_string, functions['reduce'].to_java_string)
+                      else
+                        ViewDesign.new(view.to_s.to_java_string, functions['map'].to_java_string)
+                      end
+        design_doc.getViews.add(view_design)
+      end
+
+      client.createDesignDoc(design_doc)
+    end
+
+    # Delete design doc with given id and revision.
+    #
+    # @since 1.2.0
+    #
+    # @param [String] id Design document id. It might have '_design/'
+    #   prefix.
+    #
+    # @param [String] rev Document revision. It uses latest revision if
+    #   +rev+ parameter is nil.
+    #
+    # @return [true, false]
+    def delete_design_doc(id, rev = nil)
+      client.deleteDesignDoc(id)
+    end
+
+  end
+end

data/lib/couchbase/operations/get.rb

@@ -1,3 +1,20 @@
+# Author:: Mike Evans <mike@urlgonomics.com>
+# Copyright:: 2013 Urlgonomics LLC.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 module Couchbase::Operations
   module Get
 
@@ -129,17 +146,18 @@ module Couchbase::Operations
     #     #=> {"foo" => val1, "bar" => val2}
     #
     def get(*args, &block)
-      sync_block_error if !async? && block_given?
       key, options = expand_get_args(args)
 
       if async?
-        if block_given?
-          async_get(key, options, &Proc.new)
-        else
-          async_get(key, options)
-        end
+        async_get(key, &block)
+        # if block_given?
+        #   async_get(key, &Proc.new)
+        # else
+        #   async_get(key)
+        # end
       else
-        sync_get(key, options)
+        sync_block_error if block_given?
+        get_key(key, options)
       end
     end
 
@@ -147,15 +165,14 @@ module Couchbase::Operations
       get(key, options)
     end
 
-    def get_multi(keys, options)
+    def get_bulk(keys, options)
       results = if options[:extended]
-                  get_multi_extended(keys)
+                  get_bulk_extended(keys)
                 else
-                  java_get_multi(keys)
+                  client_get_bulk(keys)
                 end
 
       not_found_error(results.size != keys.size, options)
-      results = transcode_multi_results(results) unless options[:extended]
 
       if options[:assemble_hash] || options[:extended]
         results
@@ -164,27 +181,30 @@ module Couchbase::Operations
       end
     end
 
-    private
-
-    def sync_get(key, options)
+    def async_get(key)
       case key
       when String, Symbol
-        get_single(key, options)
+        meta = { op: :get, key: key }
+        future = client.asyncGet(key)
       when Array
-        get_multi(key, options)
+        meta = { op: :get }
+        future = client.asyncGetBulk(key)
       when Hash
-        get_and_touch(key, options)
+        # async_get_and_touch(key, options, &block)
       end
+      register_future(future, meta, &Proc.new) if block_given?
     end
 
-    def async_get(key, options, &block)
+    private
+
+    def get_key(key, options)
       case key
       when String, Symbol
-        async_get_single(key, options, &block)
+        get_single(key, options)
       when Array
-        async_get_multi(key, options, &block)
+        get_bulk(key, options)
       when Hash
-        async_get_and_touch(key, options, &block)
+        get_and_touch(key, options)
       end
     end
 
@@ -196,60 +216,53 @@ module Couchbase::Operations
     end
 
     def get_single(key, options)
-      if options[:extended]
-        get_single_extended(key, options)
+      if options[:lock]
+        client_get_and_lock(key, options)
+      elsif options[:extended]
+        get_extended(key, options)
       else
-        value = if options.key?(:lock)
-                  java_get_and_lock(key, options[:lock])
-                elsif options.key?(:ttl)
-                  java_get_and_touch(key, options[:ttl])
+        value = if options.key?(:ttl)
+                  client_get_and_touch(key, options[:ttl])
                 else
-                  java_get(key)
+                  client.get(key)
                 end
 
         not_found_error(value.nil?, options)
-        value.nil? ? nil : load(value)
+        value.nil? ? nil : value
       end
     end
 
-    def get_single_extended(key, options = {})
-      extended = java_get_extended(key)
+    def get_extended(key, options = {})
+      if options.key?(:lock)
+        client_get_and_lock(key, options[:lock])
+      end
+      extended = client_get_extended(key)
       not_found_error(extended.nil?, options)
       extended
     end
 
-    def async_get_single(key, options, &block)
-      future = client.asyncGet(key)
-      register_future(future, { op: :get, key: key }, &block)
-    end
-
-    def async_get_multi(keys, options, &block)
-      future = client.asyncGetBulk(keys)
-      register_future(future, { op: :get }, &block)
-    end
-
     def get_and_touch(key, options = {})
       if key.size > 1
-        get_multi_and_touch(key, options)
+        get_bulk_and_touch(key, options)
       else
         key, ttl = key.first
-        value = java_get_and_touch(key, ttl)
+        value = client_get_and_touch(key, ttl)
         not_found_error(value.nil?)
-        { key => load(value) }
+        { key => value }
       end
     end
 
-    def get_multi_and_touch(keys, options = {})
+    def get_bulk_and_touch(keys, options = {})
       options.merge!(assemble_hash: true)
-      results = get_multi(keys.keys, options)
+      results = get_bulk(keys.keys, options)
       touch(keys)
       results.to_hash
     end
 
-    def get_multi_extended(keys, options = {})
+    def get_bulk_extended(keys, options = {})
       {}.tap do |results|
         keys.each do |key|
-          results[key] = get_single_extended(key, options)
+          results[key] = get_extended(key, options)
         end
       end
     end
@@ -258,41 +271,34 @@ module Couchbase::Operations
       keys.map { |key| results[key] }
     end
 
-    def java_get(key)
-      client.get(key)
-    end
-
-    def java_get_and_touch(key, ttl)
+    def client_get_and_touch(key, ttl)
       client.getAndTouch(key, ttl).getValue
     end
 
-    def java_get_and_lock(key, lock)
-      client.getAndLock(key, lock).value
+    def client_get_and_lock(key, options)
+      lock = options[:lock] == true ? 30 : options[:lock]
+      cas = client.getAndLock(key, lock)
+      if options[:extended]
+        [cas.getValue, nil, cas.getCas]
+      else
+        cas.getValue
+      end
     end
 
-    def java_get_extended(key)
+    def client_get_extended(key)
       cas_value = client.gets(key)
 
       if cas_value.nil?
         nil
       else
-        [load(cas_value.getValue), nil, cas_value.getCas]
+        [cas_value.getValue, nil, cas_value.getCas]
       end
     end
 
-    def java_get_multi(keys)
+    def client_get_bulk(keys)
       client.getBulk(keys)
     rescue java.lang.ClassCastException
       raise TypeError.new
     end
-
-    def transcode_multi_results(results)
-      {}.tap do |new_results|
-        results.each do |k, v|
-          new_results[k] = load(v)
-        end
-      end
-    end
-
   end
 end

data/lib/couchbase/operations/stats.rb

@@ -1,3 +1,20 @@
+# Author:: Mike Evans <mike@urlgonomics.com>
+# Copyright:: 2013 Urlgonomics LLC.
+# License:: Apache License, Version 2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
 module Couchbase::Operations
   module Stats
 
@@ -9,7 +26,17 @@ module Couchbase::Operations
                 client.getStats(statname)
               end
 
-      stats.first.last.to_hash
+      stats = stats.to_hash
+
+      {}.tap do |hash|
+        stats.each_pair do |node, values|
+          node_value = node.to_s
+          values.each_pair do |stat, value|
+            hash[stat] ||= {}
+            hash[stat][node_value] = value
+          end
+        end
+      end
     end
 
   end