whatap 0.4.91 → 0.4.92

package/lib/observers/socket.io-observer.js

@@ -4,137 +4,51 @@
  * can be found in the LICENSE file.
  */
 
-var MeterSocketio = require('../counter/meter/meter-socket.io'),
-    TraceContextManager = require('../trace/trace-context-manager'),
-    SocketStep = require('../step/socket-step'),
-    conf = require('../conf/configure'),
-    IPUtil = require('../util/iputil'),
-    Logger = require('../logger');
-const {Detector: URLPatternDetector} = require("whatap/lib/trace/serviceurl-pattern-detector");
-const HashUtil = require("whatap/lib/util/hashutil");
-const DataTextAgent = require("whatap/lib/data/datatext-agent");
-const ResourceProfile = require("whatap/lib/util/resourceprofile");
-const ProfilePack         = require('../pack/profile-pack');
-const TxRecord            = require('../service/tx-record');
-const DateUtil            = require('../util/dateutil');
-const SecurityMaster      = require('../net/security-master');
-const DataProfileAgent    = require('../data/dataprofile-agent');
-const MeterService        = require('../counter/meter/meter-service').MeterService;
+var MeterSocketio = require('../counter/meter/meter-socket.io');
 
 var SocketIOObserver = function(agent){
     this.agent = agent;
     this.packages = ['socket.io'];
 };
 
-SocketIOObserver.prototype.inject = function (mod, moduleName) {
-    if (mod.__whatap_observe__) {
-        return;
-    }
-    mod.__whatap_observe__ = true;
-    Logger.initPrint("SocketIOObserver");
-    if (conf.trace_background_socket_enabled === false) {
-        return;
-    }
-
+SocketIOObserver.prototype.inject = function( mod ){
     var self = this;
     var aop = self.agent.aop;
 
-    aop.before(mod.Server?.prototype, 'on', function (obj, args, ret, lctx) {
-        args[args.length-1] = self.__createTransactionObserver(args[args.length-1]);
-    })
-};
-
-SocketIOObserver.prototype.__createTransactionObserver = function (callback) {
-    var self = this;
-
-    return function (){
-        TraceContextManager._asyncLocalStorage.run(initCtx(), () => {
-            var ctx = TraceContextManager._asyncLocalStorage.getStore();
-            if(!ctx) {
-                return;
-            }
-
-            MeterSocketio.connected(1);
-
-            var host;
-            const socket = arguments[0];
-            const address = socket.handshake.address;
-            if(address && address.includes(':')){
-                host = address.substring(address.lastIndexOf(':')+1);
-            }
-
-            ctx.socket_connecting = true;
-
-            ctx.footprint('Socket Connecting: ' + host);
-
-            var step = new SocketStep();
-            step.start_time = ctx.getElapsedTime();
-            step.ipaddr = Buffer.from(IPUtil.stringToBytes(host));
-            // step.port = port;
-
-            ctx.socket_connecting = false;
-            step.elapsed = ctx.getElapsedTime() - step.start_time;
-            ctx.profile.push(step);
-
-            ctx.footprint('Socket Connecting Done');
-
-            self.__endTransaction(null, ctx);
-
-            return;
-        })
-    }
-}
-
-SocketIOObserver.prototype.__endTransaction = function(error, ctx) {
-    try {
-        var profile = new ProfilePack();
-        var wtx = new TxRecord();
-        wtx.endTime = DateUtil.currentTime();
-        profile.time  = wtx.endTime;
-        wtx.elapsed = ctx.getElapsedTime();
-
-        ctx.service_hash = HashUtil.hashFromString(ctx.service_name);
-        DataTextAgent.SERVICE.add(ctx.service_hash, ctx.service_name);
-
-        wtx.service = ctx.service_hash;
-        wtx.cpuTime = ResourceProfile.getCPUTime() - ctx.start_cpu;
-        wtx.malloc = ResourceProfile.getUsedHeapSize()-ctx.start_malloc;
-        if(wtx.malloc < 0) { wtx.malloc = 0; }
-        wtx.status = 2;
-
-        MeterService.add(wtx.service, wtx.elapsed,
-            wtx.errorLevel, ctx.mcaller_pcode, ctx.mcaller_okind, ctx.mcaller_oid);
+    if(mod.__whatap_observe__) { return; }
+    mod.__whatap_observe__ = true;
 
-        profile.oid = SecurityMaster.OID;
-        profile.service = wtx;
+    aop.after(mod.prototype, ['listen','attach'], function(obj, args, ret) {
+        if (!ret.sockets) return;
 
-        TraceContextManager.end(ctx._id);
-        setTimeout(function () {
-            DataProfileAgent.sendProfile(ctx, profile, false);
-            ctx = null;
-        }, 100);
+        aop.before(ret.sockets, ['on', 'addListener'], function(obj, args) {
+            if (args[0] !== 'connection') return;
 
-    } catch (e) {
-        Logger.printError('WHATAP-607', 'End transaction error..', e, false);
-        TraceContextManager.end(ctx._id);
-        ctx = null;
-    }
+            aop.functionHook(args, -1, function(obj, args) {
+                if (!args[0]) return; //no client socket
 
-};
+                var socket = args[0];
 
-function initCtx() {
-    const ctx = TraceContextManager.start();
-    if (!ctx) {
-        return;
-    }
+                MeterSocketio.connected(1);
 
-    ctx.start_malloc = ResourceProfile.getUsedHeapSize();
-    ctx.start_cpu = ResourceProfile.getCPUTime();
+                socket.on('disconnect', function() {
+                    MeterSocketio.disconnected();
+                });
 
-    var remote_addr = IPUtil.checkIp4("0.0.0.0");
-    ctx.remoteIp = IPUtil.stringToInt(remote_addr);
+                aop.before(socket, ['emit', 'send'], function(obj, args) {
+                    if (args[0] === 'newListener') return;
+                    MeterSocketio.send(null, args);
+                });
 
-    return ctx;
+                aop.before(socket, ['on', 'addListener'], function(obj, args) {
+                    if (args[0] === 'disconnect') return;
+                    aop.functionHook(args, -1, function (obj, args) {
+                        MeterSocketio.receive(null, args[0]);
+                    });
+                });
+            });
+        });
+    });
 }
 
 exports.SocketIOObserver = SocketIOObserver;

package/package.json

@@ -1,8 +1,8 @@
 {
   "name": "whatap",
   "homepage": "http://www.whatap.io",
-  "version": "0.4.91",
-  "releaseDate": "20240205",
+  "version": "0.4.92",
+  "releaseDate": "20240206",
   "description": "Monitoring and Profiling Service",
   "main": "index.js",
   "scripts": {},