Merge pull request nathantsoi#65 from weglov/feature/fix-reconnection

Fix reconnection ws instance

Author: weglov

CHANGELOG.md

@@ -2,6 +2,9 @@
 
 This package is [semantic versioned](http://semver.org/)
 
+## 2.0.9
+- [bugfix]: fixed reconnection
+
 ## 2.0.8
 - [feature]: custom pass to store logic. Defaults to original passToStore code if no custom logic provided
 

dist/build.js

@@ -1,6 +1,6 @@
 {
   "name": "vue-native-websocket",
-  "version": "2.0.8",
+  "version": "2.0.9",
   "description": "native websocket implemantation for vuejs and vuex",
   "main": "dist/build.js",
   "scripts": {

package.json

@@ -8,6 +8,10 @@ export default {
 
     let observer = null
 
+    opts.$setInstance = (wsInstance) => {
+      Vue.prototype.$socket = wsInstance
+    }
+
     if (opts.connectManually) {
       Vue.prototype.$connect = () => {
         observer = new Observer(connection, opts)

src/Main.js

@@ -55,7 +55,10 @@ export default class {
 
         if (this.store) { this.passToStore('SOCKET_' + eventType, event) }
 
-        if (this.reconnection && eventType === 'onopen') { this.reconnectionCount = 0 }
+        if (this.reconnection && eventType === 'onopen') {
+          this.opts.$setInstance(event.currentTarget)
+          this.reconnectionCount = 0
+        }
 
         if (this.reconnection && eventType === 'onclose') { this.reconnect() }
       }