Add files via upload

Author: walkor

composer.json

@@ -1,7 +1,8 @@
 {
   "name": "webman/stomp",
-  "description": "Stomp client written in PHP for webman.",
+  "description": "Stomp queue plugin for webman.",
   "require": {
+    "workerman/webman-framework": "^1.2.1",
     "workerman/stomp": "^1.0"
   },
   "autoload": {

src/Client.php

@@ -87,7 +87,7 @@ public function __call($name, $arguments)
      */
     public static function connection($name = 'default') {
         if (!isset(static::$_connections[$name])) {
-            $config = config('stomp', []);
+            $config = config('stomp', config('plugin.webman.stomp.stomp', []));
             if (!isset($config[$name])) {
                 throw new \RuntimeException("RedisQueue connection $name not found");
             }

src/Install.php

@@ -0,0 +1,70 @@
+<?php
+namespace Webman\Stomp;
+
+class Install
+{
+    const WEBMAN_PLUGIN = true;
+
+    /**
+     * @var array
+     */
+    protected static $pathRelation = array (
+  'config/plugin/webman/stomp' => 'config/plugin/webman/stomp',
+);
+
+    /**
+     * Install
+     * @return void
+     */
+    public static function install()
+    {
+        static::installByRelation();
+        mkdir(app_path() . '/queue/stomp', 0777, true);
+    }
+
+    /**
+     * Uninstall
+     * @return void
+     */
+    public static function uninstall()
+    {
+        self::uninstallByRelation();
+    }
+
+    /**
+     * installByRelation
+     * @return void
+     */
+    public static function installByRelation()
+    {
+        foreach (static::$pathRelation as $source => $dest) {
+            if ($pos = strrpos($dest, '/')) {
+                $parent_dir = base_path().'/'.substr($dest, 0, $pos);
+                if (!is_dir($parent_dir)) {
+                    mkdir($parent_dir, 0777, true);
+                }
+            }
+            //symlink(__DIR__ . "/$source", base_path()."/$dest");
+            copy_dir(__DIR__ . "/$source", base_path()."/$dest");
+        }
+    }
+
+    /**
+     * uninstallByRelation
+     * @return void
+     */
+    public static function uninstallByRelation()
+    {
+        foreach (static::$pathRelation as $source => $dest) {
+            $path = base_path()."/$dest";
+            if (!is_dir($path) && !is_file($path)) {
+                continue;
+            }
+            /*if (is_link($path) {
+                unlink($path);
+            }*/
+            remove_dir($path);
+        }
+    }
+    
+}

src/Process/Consumer.php

@@ -43,6 +43,10 @@ public function __construct($consumer_dir = '')
      */
     public function onWorkerStart()
     {
+        if (!is_dir($this->_consumerDir)) {
+            echo "Consumer directory {$this->_consumerDir} not exists\r\n";
+            return;
+        }
         $dir_iterator = new \RecursiveDirectoryIterator($this->_consumerDir);
         $iterator = new \RecursiveIteratorIterator($dir_iterator);
         foreach ($iterator as $file) {

src/config/plugin/webman/stomp/app.php

@@ -0,0 +1,4 @@
+<?php
+return [
+    'enable' => true,
+];

src/config/plugin/webman/stomp/process.php

@@ -0,0 +1,11 @@
+<?php
+return [
+    'consumer'  => [
+        'handler'     => Webman\Stomp\Process\Consumer::class,
+        'count'       => 8, // 可多进程消费
+        'constructor' => [
+            // 消费者类目录
+            'consumer_dir' => app_path() . '/queue/stomp'
+        ]
+    ]
+];

src/config/plugin/webman/stomp/stomp.php

@@ -0,0 +1,12 @@
+<?php
+return [
+    'default' => [
+        'host'    => 'stomp://127.0.0.1:61613',
+        'options' => [
+            'vhost'    => '/',
+            'login'    => 'guest',
+            'passcode' => 'guest',
+            'debug'    => true,
+        ]
+    ]
+];