bdb 0.1.0 → 0.2.0

data/VERSION

@@ -1 +1 @@
-0.1.0
+0.2.0

data/examples/replication.rb

@@ -0,0 +1,29 @@
+require 'rubygems'
+require 'bdb/database'
+require 'fileutils'
+
+role = ARGV.shift || 'client'
+
+MASTER = 'localhost:8888'
+CLIENT = 'localhost:9999'
+
+DIR = "/tmp/#{role}"
+FileUtils.rmtree DIR
+FileUtils.mkdir  DIR
+
+puts "Starting #{role}..."
+
+Bdb::Environment.replicate DIR, :from => MASTER, :to => CLIENT, :host => role == 'master' ? MASTER : CLIENT
+db = Bdb::Database.new('foo', :path => DIR)
+
+loop do
+  100.times do |i|
+    if role == 'master'
+      db.set(i, db[i].to_i + 1)
+      puts "#{i}: #{db[i]}"
+    else
+      puts "#{i}: #{db[i]}"
+    end
+    sleep 0.1
+  end
+end

data/ext/Makefile

@@ -0,0 +1,157 @@
+
+SHELL = /bin/sh
+
+#### Start of system configuration section. ####
+
+srcdir = .
+topdir = /opt/local/lib/ruby/1.8/i686-darwin10
+hdrdir = $(topdir)
+VPATH = $(srcdir):$(topdir):$(hdrdir)
+exec_prefix = $(prefix)
+prefix = $(DESTDIR)/opt/local
+sharedstatedir = $(prefix)/com
+mandir = $(DESTDIR)/opt/local/share/man
+psdir = $(docdir)
+oldincludedir = $(DESTDIR)/usr/include
+localedir = $(datarootdir)/locale
+bindir = $(exec_prefix)/bin
+libexecdir = $(exec_prefix)/libexec
+sitedir = $(libdir)/ruby/site_ruby
+htmldir = $(docdir)
+vendorarchdir = $(vendorlibdir)/$(sitearch)
+includedir = $(prefix)/include
+infodir = $(datarootdir)/info
+vendorlibdir = $(vendordir)/$(ruby_version)
+sysconfdir = $(prefix)/etc
+libdir = $(exec_prefix)/lib
+sbindir = $(exec_prefix)/sbin
+rubylibdir = $(libdir)/ruby/$(ruby_version)
+docdir = $(datarootdir)/doc/$(PACKAGE)
+dvidir = $(docdir)
+vendordir = $(DESTDIR)/opt/local/lib/ruby/vendor_ruby
+datarootdir = $(prefix)/share
+pdfdir = $(docdir)
+archdir = $(rubylibdir)/$(arch)
+sitearchdir = $(sitelibdir)/$(sitearch)
+datadir = $(datarootdir)
+localstatedir = $(prefix)/var
+sitelibdir = $(sitedir)/$(ruby_version)
+
+CC = /usr/bin/gcc-4.2
+LIBRUBY = $(LIBRUBY_SO)
+LIBRUBY_A = lib$(RUBY_SO_NAME)-static.a
+LIBRUBYARG_SHARED = -l$(RUBY_SO_NAME)
+LIBRUBYARG_STATIC = -l$(RUBY_SO_NAME)-static
+
+RUBY_EXTCONF_H = 
+CFLAGS   =  -fno-common -O2 -arch x86_64  -fno-common -pipe -fno-common $(cflags) -arch x86_64
+INCFLAGS = -I. -I. -I/opt/local/lib/ruby/1.8/i686-darwin10 -I.
+DEFS     = 
+CPPFLAGS =  -I/usr/local/BerkeleyDB.4.8/include -I/opt/local/include -D_XOPEN_SOURCE -D_DARWIN_C_SOURCE  -I/opt/local/include
+CXXFLAGS = $(CFLAGS) 
+ldflags  = -L. -L/opt/local/lib
+dldflags = 
+archflag = -arch x86_64
+DLDFLAGS = $(ldflags) $(dldflags) $(archflag)
+LDSHARED = $(CC) -dynamic -bundle -undefined suppress -flat_namespace
+AR = ar
+EXEEXT = 
+
+RUBY_INSTALL_NAME = ruby
+RUBY_SO_NAME = ruby
+arch = i686-darwin10
+sitearch = i686-darwin10
+ruby_version = 1.8
+ruby = /opt/local/bin/ruby
+RUBY = $(ruby)
+RM = rm -f
+MAKEDIRS = mkdir -p
+INSTALL = /usr/bin/install -c
+INSTALL_PROG = $(INSTALL) -m 0755
+INSTALL_DATA = $(INSTALL) -m 644
+COPY = cp
+
+#### End of system configuration section. ####
+
+preload = 
+
+libpath = . $(libdir) /usr/local/BerkeleyDB.4.8/lib
+LIBPATH =  -L. -L$(libdir) -L/usr/local/BerkeleyDB.4.8/lib
+DEFFILE = 
+
+CLEANFILES = mkmf.log
+DISTCLEANFILES = 
+
+extout = 
+extout_prefix = 
+target_prefix = 
+LOCAL_LIBS = 
+LIBS = $(LIBRUBYARG_SHARED) -ldb-4.8  -lpthread -ldl -lobjc  
+SRCS = bdb.c
+OBJS = bdb.o
+TARGET = bdb
+DLLIB = $(TARGET).bundle
+EXTSTATIC = 
+STATIC_LIB = 
+
+BINDIR        = $(bindir)
+RUBYCOMMONDIR = $(sitedir)$(target_prefix)
+RUBYLIBDIR    = $(sitelibdir)$(target_prefix)
+RUBYARCHDIR   = $(sitearchdir)$(target_prefix)
+
+TARGET_SO     = $(DLLIB)
+CLEANLIBS     = $(TARGET).bundle $(TARGET).il? $(TARGET).tds $(TARGET).map
+CLEANOBJS     = *.o *.a *.s[ol] *.pdb *.exp *.bak
+
+all:		$(DLLIB)
+static:		$(STATIC_LIB)
+
+clean:
+		@-$(RM) $(CLEANLIBS) $(CLEANOBJS) $(CLEANFILES)
+
+distclean:	clean
+		@-$(RM) Makefile $(RUBY_EXTCONF_H) conftest.* mkmf.log
+		@-$(RM) core ruby$(EXEEXT) *~ $(DISTCLEANFILES)
+
+realclean:	distclean
+install: install-so install-rb
+
+install-so: $(RUBYARCHDIR)
+install-so: $(RUBYARCHDIR)/$(DLLIB)
+$(RUBYARCHDIR)/$(DLLIB): $(DLLIB)
+	$(INSTALL_PROG) $(DLLIB) $(RUBYARCHDIR)
+install-rb: pre-install-rb install-rb-default
+install-rb-default: pre-install-rb-default
+pre-install-rb: Makefile
+pre-install-rb-default: Makefile
+$(RUBYARCHDIR):
+	$(MAKEDIRS) $@
+
+site-install: site-install-so site-install-rb
+site-install-so: install-so
+site-install-rb: install-rb
+
+.SUFFIXES: .c .m .cc .cxx .cpp .C .o
+
+.cc.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) -c $<
+
+.cxx.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) -c $<
+
+.cpp.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) -c $<
+
+.C.o:
+	$(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) -c $<
+
+.c.o:
+	$(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) -c $<
+
+$(DLLIB): $(OBJS) Makefile
+	@-$(RM) $@
+	$(LDSHARED) -o $@ $(OBJS) $(LIBPATH) $(DLDFLAGS) $(LOCAL_LIBS) $(LIBS)
+
+
+
+$(OBJS): ruby.h defines.h

data/ext/bdb.c

@@ -267,9 +267,9 @@ VALUE db_open(VALUE obj, VALUE vtxn, VALUE vdisk_file,
   
   dbh->db->app_private=dbh;
   rv = dbh->db->open(dbh->db,txn?txn->txn:NULL,
-		     StringValueCStr(vdisk_file),
-		     logical_db,
-		     dbtype,flags,mode);
+                     StringValueCStr(vdisk_file),
+                     logical_db,
+                     dbtype,flags,mode);
   if (rv != 0) {
     raise_error(rv,"db_open failure: %s(%d)",db_strerror(rv),rv);
   }
@@ -1637,6 +1637,8 @@ VALUE env_open(VALUE obj, VALUE vhome, VALUE vflags, VALUE vmode)
     raise_error(0,"env handle already used and closed");
 
   rv = eh->env->open(eh->env,StringValueCStr(vhome),flags,mode);
+  eh->env->app_private=eh;
+
   if (rv != 0) {
     raise_error(rv, "env_open failure: %s",db_strerror(rv));
   }
@@ -2749,6 +2751,205 @@ VALUE env_get_home(VALUE obj)
   return rb_str_new2(home);
 }
 
+/*
+ * call-seq:
+ * env.set_verbose(which, onoff)
+ *
+ * set verbose messages on or off
+ */
+VALUE env_set_verbose(VALUE obj, VALUE which, VALUE onoff)
+{
+  t_envh *eh;
+  int rv;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  rv = eh->env->set_verbose(eh->env, NUM2UINT(which), RTEST(onoff));
+
+  if ( rv != 0 )  raise_error(rv, "env_set_verbose: %s",db_strerror(rv));
+
+  return Qtrue;
+}
+
+/*
+ * call-seq:
+ * env.rep_priority = int
+ *
+ * specify how the replication manager will handle acknowledgement of replication messages
+ */
+VALUE env_rep_set_priority(VALUE obj, VALUE priority)
+{
+  t_envh *eh;
+  int rv;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  rv = eh->env->rep_set_priority(eh->env, NUM2UINT(priority));
+
+  if ( rv != 0 ) raise_error(rv, "env_rep_set_priority: %s", db_strerror(rv));
+  return priority;
+}
+
+/*
+ * call-seq:
+ * env.rep_priority -> int
+ *
+ * returns the replication manager's acknowledgement policy
+ */
+VALUE env_rep_get_priority(VALUE obj)
+{
+  t_envh *eh;
+  u_int32_t priority;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  eh->env->rep_get_priority(eh->env, &priority);
+
+  return INT2NUM(priority);
+}
+
+/*
+ * call-seq:
+ * env.rep_nsites = int
+ *
+ * specify how the replication manager will handle acknowledgement of replication messages
+ */
+VALUE env_rep_set_nsites(VALUE obj, VALUE nsites)
+{
+  t_envh *eh;
+  int rv;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  rv = eh->env->rep_set_nsites(eh->env, NUM2UINT(nsites));
+
+  if ( rv != 0 ) raise_error(rv, "env_rep_set_nsites: %s", db_strerror(rv));
+  return nsites;
+}
+
+/*
+ * call-seq:
+ * env.rep_nsites -> int
+ *
+ * returns the replication manager's acknowledgement policy
+ */
+VALUE env_rep_get_nsites(VALUE obj)
+{
+  t_envh *eh;
+  u_int32_t nsites;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  eh->env->rep_get_nsites(eh->env, &nsites);
+
+  return INT2NUM(nsites);
+}
+
+
+/*
+ * call-seq:
+ * env.repmgr_set_local_site(host, port)
+ *
+ * specify the local site for the replication manager
+ */
+VALUE env_repmgr_set_local_site(VALUE obj, VALUE host, VALUE port)
+{
+  t_envh *eh;
+  int rv;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  rv = eh->env->repmgr_set_local_site(eh->env, StringValuePtr(host), NUM2UINT(port), 0);
+
+  if ( rv != 0 ) raise_error(rv, "env_repmgr_set_local_site: %s", db_strerror(rv));
+  return Qtrue;
+}
+
+/*
+ * call-seq:
+ * env.repmgr_add_remote_site(host, port)
+ *
+ * add a remote for the replication manager
+ */
+VALUE env_repmgr_add_remote_site(VALUE obj, VALUE host, VALUE port)
+{
+  t_envh *eh;
+  int rv;
+  int eidp;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  rv = eh->env->repmgr_add_remote_site(eh->env, StringValuePtr(host), NUM2UINT(port), &eidp, 0);
+
+  if ( rv != 0 ) raise_error(rv, "env_repmgr_add_remote_site: %s", db_strerror(rv));
+  return INT2NUM(eidp);
+}
+
+/*
+ * call-seq:
+ * env.repmgr_ack_policy = int
+ *
+ * specify how the replication manager will handle acknowledgement of replication messages
+ */
+VALUE env_repmgr_set_ack_policy(VALUE obj, VALUE policy)
+{
+  t_envh *eh;
+  int rv;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  rv = eh->env->repmgr_set_ack_policy(eh->env, NUM2INT(policy));
+
+  if ( rv != 0 ) raise_error(rv, "env_repmgr_set_ack_policy: %s", db_strerror(rv));
+  return policy;
+}
+
+/*
+ * call-seq:
+ * env.repmgr_ack_policy -> int
+ *
+ * returns the replication manager's acknowledgement policy
+ */
+VALUE env_repmgr_get_ack_policy(VALUE obj)
+{
+  t_envh *eh;
+  int policy;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  eh->env->repmgr_get_ack_policy(eh->env, &policy);
+
+  return INT2NUM(policy);
+}
+
+/*
+ * call-seq:
+ * env.repmgr_start(num_threads, flags)
+ *
+ * start the replication manager
+ */
+VALUE env_repmgr_start(VALUE obj, VALUE num_threads, VALUE flags)
+{
+  t_envh *eh;
+  int rv;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  rv = eh->env->repmgr_start(eh->env, NUM2INT(num_threads), NUM2UINT(flags));
+
+  if ( rv != 0 ) raise_error(rv, "env_repmgr_start: %s", db_strerror(rv));
+  return Qtrue;
+}
+
+/*
+ * call-seq:
+ * env.repmgr_stat_print
+ *
+ * prints replication manager stats
+ */
+VALUE env_repmgr_stat_print(VALUE obj, VALUE flags)
+{
+  t_envh *eh;
+  int rv;
+
+  Data_Get_Struct(obj,t_envh,eh);
+  rv = eh->env->repmgr_stat_print(eh->env, NUM2UINT(flags));
+
+  if ( rv != 0 ) raise_error(rv, "env_repmgr_stat_print: %s", db_strerror(rv));
+  return Qtrue;
+}
+
+
 static void txn_finish(t_txnh *txn)
 {
   if ( RTEST(ruby_debug) )
@@ -2935,7 +3136,7 @@ void Init_bdb() {
   rb_define_method(cDb,"associate",db_associate,4);
   rb_define_method(cDb,"btree_compare=",db_btree_compare_set,1);
   rb_define_method(cDb,"flags=",db_flags_set,1);
-	rb_define_method(cDb,"flags",db_flags_get,0);
+  rb_define_method(cDb,"flags",db_flags_get,0);
   rb_define_method(cDb,"open",db_open,6);
   rb_define_method(cDb,"close",db_close,1);
   rb_define_method(cDb,"[]",db_aget,1);
@@ -2977,8 +3178,8 @@ void Init_bdb() {
   rb_define_method(cEnv,"close",env_close,0);
   rb_define_method(cEnv,"db",env_db,0);
   rb_define_method(cEnv,"cachesize=",env_set_cachesize,1);
- 	rb_define_method(cEnv,"cachesize",env_get_cachesize,0);
-	rb_define_method(cEnv,"flags",env_get_flags,0);
+  rb_define_method(cEnv,"cachesize",env_get_cachesize,0);
+  rb_define_method(cEnv,"flags",env_get_flags,0);
   rb_define_method(cEnv,"flags_on=",env_set_flags_on,1);
   rb_define_method(cEnv,"flags_off=",env_set_flags_off,1);
   rb_define_method(cEnv,"list_dbs",env_list_dbs,0);
@@ -2995,19 +3196,31 @@ void Init_bdb() {
   rb_define_method(cEnv,"set_lk_detect",env_set_lk_detect,1);
   rb_define_method(cEnv,"get_lk_detect",env_get_lk_detect,0);
   rb_define_method(cEnv,"set_lk_max_locks",env_set_lk_max_locks,1);
-	rb_define_method(cEnv,"get_lk_max_locks",env_get_lk_max_locks,0);
+  rb_define_method(cEnv,"get_lk_max_locks",env_get_lk_max_locks,0);
   rb_define_method(cEnv,"set_lk_max_objects",env_set_lk_max_objects,1);
-	rb_define_method(cEnv,"get_lk_max_objects",env_get_lk_max_objects,0);
+  rb_define_method(cEnv,"get_lk_max_objects",env_get_lk_max_objects,0);
   rb_define_method(cEnv,"set_shm_key",env_set_shm_key,1);
   rb_define_method(cEnv,"get_shm_key",env_get_shm_key,0);
 
   rb_define_method(cEnv,"set_data_dir",env_set_data_dir,1);
-	rb_define_method(cEnv,"get_data_dirs",env_get_data_dirs,0);
+  rb_define_method(cEnv,"get_data_dirs",env_get_data_dirs,0);
   rb_define_method(cEnv,"set_lg_dir",env_set_lg_dir,1);
-	rb_define_method(cEnv,"get_lg_dir",env_get_lg_dir,0);
+  rb_define_method(cEnv,"get_lg_dir",env_get_lg_dir,0);
   rb_define_method(cEnv,"set_tmp_dir",env_set_tmp_dir,1);
-	rb_define_method(cEnv,"get_tmp_dir",env_get_tmp_dir,0);
-	rb_define_method(cEnv,"get_home",env_get_home,0);
+  rb_define_method(cEnv,"get_tmp_dir",env_get_tmp_dir,0);
+  rb_define_method(cEnv,"get_home",env_get_home,0);
+  rb_define_method(cEnv,"set_verbose",env_set_verbose,2);
+
+  rb_define_method(cEnv,"rep_priority=", env_rep_set_priority, 1);
+  rb_define_method(cEnv,"rep_priority", env_rep_get_priority, 0);
+  rb_define_method(cEnv,"rep_nsites=", env_rep_set_nsites, 1);
+  rb_define_method(cEnv,"rep_nsites", env_rep_get_nsites, 0);
+  rb_define_method(cEnv,"repmgr_set_local_site", env_repmgr_set_local_site, 2);
+  rb_define_method(cEnv,"repmgr_add_remote_site", env_repmgr_add_remote_site, 2);
+  rb_define_method(cEnv,"repmgr_ack_policy=", env_repmgr_set_ack_policy, 1);
+  rb_define_method(cEnv,"repmgr_ack_policy", env_repmgr_get_ack_policy, 0);
+  rb_define_method(cEnv,"repmgr_start", env_repmgr_start, 2);
+  rb_define_method(cEnv,"repmgr_stat_print", env_repmgr_stat_print, 1);
 
   cTxnStat = rb_define_class_under(mBdb,"TxnStat",rb_cObject);
   rb_define_method(cTxnStat,"[]",stat_aref,1);