ro_daemon 0.0.4

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 664d505f80465c4138d376b6b975cdf3bc8ccd2d
+  data.tar.gz: 88bf4d5bffa691aa145e8c0cae2a0b58eb88cda2
+SHA512:
+  metadata.gz: d92895f89bdc6786e6901fb9b3345909326271750cde3a9e26b7e5021f829cb0dd5a12794824657936736f27f620319ea27977cb9afb2dc2791080ef902f7ebe
+  data.tar.gz: e3c9729487cbb41e728eceea100be2a11b03c9d862c59d34ae1c4f447ca09722c10055a9b0112953cd737c1264e55668152aa0c0c49cbc0d76bf59f0e1962c5b

data/bin/ro_daemonize

@@ -0,0 +1,3 @@
+require File.expand_path('../../lib/ro_daemon/daemon', __FILE__)
+
+RoDaemon::Daemon.st(*([:daemonize] + ARGV))

data/lib/ro_daemon/daemon.rb

@@ -0,0 +1,95 @@
+require File.expand_path('../../../conf/ro_daemon', __FILE__)
+
+require 'ro_thor'
+require 'sqlite3'
+require_relative 'data'
+require_relative 'helper'
+
+module RoDaemon
+  class Daemon < RoThor
+    include Helper
+
+    usg :daemonize
+
+    def daemonize(cmd, *args, &blk)
+      logs = Dir["#{logs_dir}/*.log"]
+      ms_cnn_db
+
+      d = Data::Daemon.new(cmd: cmd)
+
+      stdout, stderr = d.stdout, d.stderr
+      pid = spawn_daemon(cmd, stdout, stderr)
+      fix_db_err do
+        d.save
+      end
+
+      pid
+    end
+
+    def spawn_daemon(cmd, stdout, stderr)
+      pid = spawn(cmd, out: stdout, err: stderr)
+      puts
+      puts "spawn pid:#{pid} cmd:#{cmd}\nstdout: #{stdout}\nstderr: #{stderr}"
+      puts
+      pid
+    end
+
+    def fix_db_err(*args, &blk)
+      begin
+        yield
+      rescue ActiveRecord::StatementInvalid => e
+        if e.to_s.match(%r{Could not find table})
+          Data::Mg.st
+          yield
+        else
+          raise e
+        end
+      end
+    end
+
+    def get_stdout_stderr(logs)
+      logs.select do |log|
+        Time.now - File.mtime(log) > 60.days and Data::Daemon.first(stdout: log).closed?
+      end.each do |f|
+        FileUtils.rm_f(f)
+      end
+
+      if Data::Daemon.count > 0
+        (1.. Data::Daemon.all.last.id)
+      end
+
+      unless test(?d, logs_dir)
+        FileUtils.mkdir(logs_dir)
+      end
+
+      stdout = File.open logs_dir("#{stdout_fn}.log"), 'w+'
+      stderr = File.open logs_dir("#{stdout_fn}_err.log"), "w+"
+      return stderr, stdout
+    end
+
+    def ms_cnn_db(*args, &blk)
+      @cnn ||= ActiveRecord::Base.establish_connection(adapter: "sqlite3", database: db)
+    end
+
+    def db(*args, &blk)
+      _db = logs_dir('manager.sqlite')
+
+      if File.exist?(_db)
+        return _db
+      end
+
+      begin
+        sh "sqlite3 dbname.sqlite .dump"
+      rescue ShErr => e
+        if e.to_s.match(%r{command not found})
+          sh "sudo apt-get install sqlite3 -y"
+          sh "sqlite3 dbname.sqlite .dump"
+        else
+          raise e
+        end
+      end
+      _db
+    end
+
+  end
+end

data/lib/ro_daemon/data.rb

@@ -0,0 +1,48 @@
+require 'active_record'
+require_relative 'mg'
+require 'ro_thor'
+require 'ro_thor/sh'
+require_relative 'helper'
+
+module RoDaemon
+  module Data
+    class Daemon < ActiveRecord::Base
+
+      class << self
+        include RoDaemon::Helper
+        def new(*args, &blk)
+            r = super
+            r.stdout = logs_dir("#{r.id}.log")
+            r.stderr = logs_dir("#{r.id}_err.log")
+            r
+        end
+
+        def create(*args, &blk)
+          r = super
+          puts "create #{self}"
+          r
+        end
+      end
+      include RoThor::Sh
+
+      def closed?(*args, &blk)
+        begin
+          sh "ps --pid #{pid}"
+          true
+        rescue ShErr => e
+          if e.stdout.match(%r{PID TTY          TIME CMD\n})
+            false
+          else
+            raise e
+          end
+        end
+      end
+    end
+
+    class Mg < RoDaemon::Mg
+      def up(*args, &blk)
+        ms_tab(:daemons, pid: :integer, cmd: :string, stdout: :string, stderr: :string)
+      end
+    end
+  end
+end

data/lib/ro_daemon/helper.rb

@@ -0,0 +1,7 @@
+module RoDaemon
+	module Helper
+		def logs_dir(*args, &blk)
+			File.join '/tmp/ro_daemonize/', *args, &blk
+		end
+	end
+end

data/lib/ro_daemon/mg.rb

@@ -0,0 +1,67 @@
+require 'active_record'
+require 'ro_thor'
+module RoDaemon
+  class Mg < ::ActiveRecord::Migration
+    class << self
+      def st(*args, &blk)
+        if block_given?
+          kls = Class.new(self)
+          kls.class_eval &blk
+          kls.st
+        else
+          migrate(:up)
+        end
+      end
+    end
+
+    def mst(*args, &blk)
+      ms_tab(*args, &blk)
+    end
+
+    def ms_tab(tab, attr_name_types=nil, *args, &blk)
+      attr_name_types ||= {}
+      unless exist?(tab)
+        create :"#{tab}" do |t|
+          t.timestamps null: false
+        end
+      end
+
+      attr_name_types.each do |name, type|
+        unless exist_col?(tab, name)
+          add_col(tab, name, type)
+        end
+      end
+    end
+
+    def add_col(*args, &blk)
+      add_column(*args, &blk)
+    end
+
+    def change_col(*args, &blk)
+      change_column(*args, &blk)
+    end
+
+    def exist_col?(*args, &blk)
+      column_exists?(*args, &blk)
+    end
+
+    def rm_col(*args, &blk)
+      remove_column(*args, &blk)
+    end
+
+    def create(*args, &blk)
+      create_table(*args, &blk)
+    end
+
+    def exist?(*args, &blk)
+      table_exists?(*args, &blk)
+    end
+
+    def del(name, *args, &blk)
+      if exist?(name)
+        drop_table(name, *args, &blk)
+      end
+    end
+
+  end
+end

data/spec/lib/ro_daemon/daemon_spec.rb

@@ -0,0 +1,12 @@
+require 'rails_helper'
+
+describe "RoDaemon::Daemon.new" do
+  let(:o) do
+    RoDaemon::Daemon.new
+  end
+
+  it :daemonize do
+    pid = o.daemonize RoDaemon.rt('spec/try.rb')
+    Sh.wait(pid)
+  end
+end

data/spec/rails_helper.rb

@@ -0,0 +1 @@
+require_relative 'spec_helper'

data/spec/spec_helper.rb

@@ -0,0 +1,21 @@
+require 'rubygems'
+require 'spork'
+
+
+Spork.prefork do
+
+
+  require 'pj'
+  Pj.set_test
+
+  require 'ro_rspec_helpers_plus'
+
+  RoRspec.spec = __dir__
+  
+  require File.expand_path("../../conf/ro_daemon", __FILE__)
+
+  module RoDaemon
+    extend PjHelper
+    autoload_all "ro_daemon"
+  end
+end

data/spec/try.rb

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+
+puts "puts at #{Time.now}"
+raise "raise at #{Time.now}"

metadata

@@ -0,0 +1,98 @@
+--- !ruby/object:Gem::Specification
+name: ro_daemon
+version: !ruby/object:Gem::Version
+  version: 0.0.4
+platform: ruby
+authors:
+- ''
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-04-03 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: rails
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: ro_thor
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: sqlite3
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: ''
+email:
+- ''
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/ro_daemonize
+- lib/ro_daemon/daemon.rb
+- lib/ro_daemon/data.rb
+- lib/ro_daemon/helper.rb
+- lib/ro_daemon/mg.rb
+- spec/lib/ro_daemon/daemon_spec.rb
+- spec/rails_helper.rb
+- spec/spec_helper.rb
+- spec/try.rb
+homepage: ''
+licenses: []
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: ''
+test_files:
+- spec/lib/ro_daemon/daemon_spec.rb
+- spec/spec_helper.rb
+- spec/rails_helper.rb
+- spec/try.rb