Sha256: c8ec7b19f81b98c0fb550da208c4f167aec39d6b1ad9dcc40559e7544a34fd4d

Contents?: true

Size: 1.87 KB

Versions: 1

Compression:

Stored size: 1.87 KB

Contents

module MoSQL
  class SQLAdapter
    include MoSQL::Logging

    attr_reader :db

    def initialize(schema, uri, pgschema=nil)
      @schema = schema
      connect_db(uri, pgschema)
    end

    def connect_db(uri, pgschema)
      @db = Sequel.connect(uri, :after_connect => proc do |conn|
                             if pgschema
                               begin
                                 conn.execute("CREATE SCHEMA \"#{pgschema}\"")
                               rescue PG::Error
                               end
                               conn.execute("SET search_path TO \"#{pgschema}\"")
                             end
                           end)
    end

    def table_for_ns(ns)
      @db[@schema.table_for_ns(ns).intern]
    end


    def upsert_ns(ns, obj)
      h = {}
      cols = @schema.all_columns(@schema.find_ns(ns))
      row  = @schema.transform(ns, obj)
      cols.zip(row).each { |k,v| h[k] = v }
      upsert(table_for_ns(ns), h)
    end

    # obj must contain an _id field. All other fields will be ignored.
    def delete_ns(ns, obj)
      cols = @schema.all_columns(@schema.find_ns(ns))
      row  = @schema.transform(ns, obj)
      sqlid = row[cols.index("_id")]
      raise "No _id found in transform of #{obj}" if sqlid.nil?
      table_for_ns(ns).where(:_id => sqlid).delete
    end

    def upsert(table, item)
      begin
        upsert!(table, item)
      rescue Sequel::DatabaseError => e
        wrapped = e.wrapped_exception
        if wrapped.result
          log.warn("Ignoring row (_id=#{item['_id']}): #{e}")
        else
          raise e
        end
      end
    end

    def upsert!(table, item)
      begin
        table.insert(item)
      rescue Sequel::DatabaseError => e
        raise e unless e.message =~ /duplicate key value violates unique constraint/
        table.where(:_id => item['_id']).update(item)
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version Path
mosql-0.1.1 lib/mosql/sql.rb