rbnewt 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: d6bddaa45284da07b7cf3b51104397055ab5cf08
-  data.tar.gz: c10e6c4d5185b4f5bc576d92aeda038226afb12b
+  metadata.gz: 2fc9c96699a8b2aee42b4943b41863e7920a55f2
+  data.tar.gz: dafa4acf3efd038b7795ed6f12f303041a76aaa2
 SHA512:
-  metadata.gz: fdd158bf69d975ed10fcf9a07425e4cc809d17df80aaadb5f5af7bdc06942350fbfd044ed9f25fca3c64769d019a82e489c2617482f783318442c7c41bfb0236
-  data.tar.gz: 121196e8662281e56ff3e7cec45936eee922aa6e7b51deabda58175cd801f4fb4562d2d01068f4762996d5528bbdfce1b5a061948d0b6a83889be04668858c29
+  metadata.gz: c26d04528ac759b858a67f3f595fa94670616d5c871137acb19c457278619f07ff912eb244febf417b6b33ce6b55e340a3ea66b860ca002a66ae890df04cf350
+  data.tar.gz: 164bcf5a890544b1a1daf035ab084e2ec91486257027c0cbfe3f6409c21538786615bf8dc1d6392f3ee71fc7205f9bb3c8085357b310546c5310dc2c88280d71

data/extconf.rb

@@ -1,3 +1,26 @@
 require "mkmf"
 
-create_makefile("rbnewt")
+$libs = "-lpthread -lstomp"
+$CFLAGS = "-I./include"
+
+def is_header?
+  is_ok = true
+  is_ok &= have_header('stomp/stomp.h') ||
+    find_header('stomp/stomp.h', '/opt/local/include', '/usr/local/include', '/usr/include')
+  is_ok &= have_header('pthread.h') ||
+    find_header('pthread.h', '/opt/local/include', '/usr/local/include', '/usr/include')
+end
+
+def is_library?
+  is_ok = true
+  is_ok &= have_library('stomp', 'stomp_session_new') ||
+    find_library('stomp', 'stomp_session_new', '/opt/local/lib', '/usr/local/lib', '/usr/lib')
+  is_ok &= have_library('pthread', 'pthread_create') ||
+    find_library('pthread', 'pthread_create', '/opt/local/lib', '/usr/local/lib', '/usr/lib')
+end
+
+if is_header? and is_library?
+  create_makefile('rbnewt', 'src')
+else
+  raise "Dependent libraries have not been installed yet."
+end

data/include/newt/common.h

@@ -0,0 +1,7 @@
+#ifndef __NEWT_COMMON_H__
+#define __NEWT_COMMON_H__
+
+#define RET_SUCCESS 1
+#define RET_ERROR -1
+
+#endif

data/include/newt/stomp.h

@@ -0,0 +1,19 @@
+#ifndef __NEWT_STOMP_H__
+#define __NEWT_STOMP_H__
+
+#include <newt/common.h>
+#include <stomp/stomp.h>
+
+#define NEWT_STOMP_KEY_SERVER "server"
+#define NEWT_STOMP_KEY_USERID "userid"
+#define NEWT_STOMP_KEY_PASSWD "passwd"
+#define NEWT_STOMP_KEY_PORT "port"
+
+#define NEWT_STOMP_DEFAULT_SERVER "localhost"
+#define NEWT_STOMP_DEFAULT_USERID "guest"
+#define NEWT_STOMP_DEFAULT_PASSWD "guest"
+#define NEWT_STOMP_DEFAULT_PORT "61613"
+
+int cnewt_stomp_initialize(stomp_session_t *, char *, char *, char *, char *);
+
+#endif

data/src/rbnewt.c

@@ -0,0 +1,99 @@
+#include <ruby.h>
+#include <assert.h>
+#include <stdlib.h>
+#include <pthread.h>
+#include <stomp/stomp.h>
+#include <newt/stomp.h>
+
+static char *get_str_from_hash(VALUE hash, char *key) {
+  assert(TYPE(hash) == T_HASH);
+  
+  VALUE v = rb_hash_aref(hash, ID2SYM(rb_intern(key)));
+  if(v != Qnil && TYPE(v) == T_STRING) {
+    return StringValueCStr(v);
+  }
+
+  return NULL;
+}
+
+static int get_int_from_hash(VALUE hash, char *key) {
+  assert(TYPE(hash) == T_HASH);
+
+  VALUE v = rb_hash_aref(hash, ID2SYM(rb_intern(key)));
+  if(v != Qnil && TYPE(v) == T_FIXNUM) {
+    return NUM2INT(v);
+  }
+
+  return -1;
+}
+
+VALUE newt_stomp_close(VALUE self) {
+  stomp_session_t *session;
+
+  Data_Get_Struct(self, stomp_session_t, session);
+
+  if(cnewt_stomp_close(session) == RET_SUCCESS) {
+    return self;
+  }
+
+  return Qnil;
+}
+
+VALUE newt_stomp_initialize(int argc, VALUE *argv, VALUE self) {
+  stomp_session_t *session;
+  VALUE opts;
+  char *server, *userid, *passwd, *port;
+
+  rb_scan_args(argc, argv, "01", &opts);
+
+  Data_Get_Struct(self, stomp_session_t, session);
+
+  // zero-set for initialization
+  server = userid = passwd = port = NULL;
+
+  if(! NIL_P(opts)) {
+    server = get_str_from_hash(opts, NEWT_STOMP_KEY_SERVER);
+    userid = get_str_from_hash(opts, NEWT_STOMP_KEY_USERID);
+    passwd = get_str_from_hash(opts, NEWT_STOMP_KEY_PASSWD);
+    port   = get_str_from_hash(opts, NEWT_STOMP_KEY_PORT);
+  }
+
+  //set default values
+  if(server == NULL) server = NEWT_STOMP_DEFAULT_SERVER;
+  if(userid == NULL) userid = NEWT_STOMP_DEFAULT_USERID;
+  if(passwd == NULL) passwd = NEWT_STOMP_DEFAULT_PASSWD;
+  if(port == NULL)   port   = NEWT_STOMP_DEFAULT_PORT;
+
+  if(cnewt_stomp_initialize(session, server, port, userid, passwd) != RET_SUCCESS) {
+    printf("[warning] failed to connect server\n");
+  }
+
+  return self;
+}
+
+static VALUE newt_stomp_alloc(VALUE klass) {
+  stomp_session_t *session;
+  pthread_t *thread_id;
+  VALUE ret = Qnil;
+
+  thread_id = (pthread_t *)malloc(sizeof(pthread_t));
+  assert(thread_id != NULL);
+
+  session = stomp_session_new(thread_id);
+  if(session != NULL) {
+    ret = rb_data_object_alloc(klass, session, 0, NULL);
+  }
+
+  return ret;
+}
+
+void Init_rbnewt(void) {
+  VALUE module, stomp_kls;
+
+  module = rb_define_module("Newt");
+  stomp_kls = rb_define_class_under(module, "STOMP", rb_cObject);
+
+  rb_define_alloc_func(stomp_kls, newt_stomp_alloc);
+  rb_define_method(stomp_kls, "initialize", newt_stomp_initialize, -1);
+  rb_define_method(stomp_kls, "close", newt_stomp_close, 0);
+}

data/src/stomp.c

@@ -0,0 +1,49 @@
+#include <stomp/stomp.h>
+#include <newt/common.h>
+#include <newt/stomp.h>
+#include <pthread.h>
+
+static void *thread_stomp(void *data) {
+  stomp_session_t *session = (stomp_session_t *)data;
+
+  //here is main-loop
+  stomp_run(session);
+
+  stomp_session_free(session);
+}
+
+int cnewt_stomp_initialize(stomp_session_t *session, char *server, char *port, char *userid, char *passwd) {
+  const struct stomp_hdr connection_headers[] = {
+    {"login",    userid},
+    {"passcode", passwd},
+  };
+  int ret = RET_ERROR;
+  pthread_t *thread_id = (pthread_t *)session->ctx;
+
+  if(thread_id != NULL) {
+    stomp_connect(session, server, port, 2, connection_headers);
+
+    pthread_create(thread_id, NULL, thread_stomp, session);
+
+    ret = RET_SUCCESS;
+  }
+
+  return ret;
+}
+
+int cnewt_stomp_close(stomp_session_t *session) {
+  pthread_t *thread_id = (pthread_t *)session->ctx;
+  const struct stomp_hdr disconnect_header[] = {{0}};
+  int ret = RET_ERROR;
+
+  if(thread_id != NULL) {
+    // set flag to escape stomp mainloop
+    session->run = 0;
+
+    pthread_join(*thread_id, NULL);
+
+    ret = RET_SUCCESS;
+  }
+
+  return ret;
+}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rbnewt
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Hiroyasu OHYAMA
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-05-30 00:00:00.000000000 Z
+date: 2016-06-01 00:00:00.000000000 Z
 dependencies: []
 description: This aims to to be the fastest Ruby client for NewtMQ which is an implementation
   of STOMP server.
@@ -20,6 +20,10 @@ extensions:
 extra_rdoc_files: []
 files:
 - extconf.rb
+- include/newt/common.h
+- include/newt/stomp.h
+- src/rbnewt.c
+- src/stomp.c
 homepage: https://github.com/userlocalhost2000/rbnewt
 licenses:
 - MIT