@agentscope/studio
Version:
AgentScope Studio is a powerful local monitoring and visualization tool designed to provide real-time insights into your system's performance and behavior.
155 lines (154 loc) • 6.81 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.OtelGrpcServer = void 0;
const grpc = __importStar(require("@grpc/grpc-js"));
const Trace_1 = require("../dao/Trace");
const socket_1 = require("../trpc/socket");
const trace_service_1 = require("./opentelemetry/proto/collector/trace/v1/trace_service");
const processor_1 = require("./processor");
/**
* gRPC server implementation for OpenTelemetry TraceService
*/
class OtelGrpcServer {
constructor(defaultPort = 4317) {
this.defaultPort = defaultPort;
this.server = new grpc.Server();
this.setupService();
}
setupService() {
// Create TraceService implementation
const traceServiceImpl = {
Export: this.handleExport.bind(this),
};
// Add the service to the server
this.server.addService(trace_service_1.opentelemetry.proto.collector.trace.v1
.UnimplementedTraceServiceService.definition, traceServiceImpl);
}
/**
* Handle Export RPC call
*/
handleExport(call, callback) {
return __awaiter(this, void 0, void 0, function* () {
try {
console.debug('[OTEL gRPC] Received ExportTraceServiceRequest');
const request = call.request;
const resourceSpans = request.resource_spans || [];
if (resourceSpans.length === 0) {
console.warn('[OTEL gRPC] Empty resource_spans in request');
const response = new trace_service_1.opentelemetry.proto.collector.trace.v1.ExportTraceServiceResponse();
callback(null, response);
return;
}
// Convert ResourceSpans to plain objects for processing
const resourceSpansArray = resourceSpans.map((rs) => rs.toObject());
// Process spans using the existing processor
const spans = processor_1.SpanProcessor.batchProcessOTLPTraces(resourceSpansArray);
// Save spans to the database
yield Trace_1.SpanDao.saveSpans(spans);
// Broadcast spans to the run room
socket_1.SocketManager.broadcastSpanDataToRunRoom(spans);
console.debug(`[OTEL gRPC] Successfully processed ${spans.length} spans`);
// Create success response
const response = new trace_service_1.opentelemetry.proto.collector.trace.v1.ExportTraceServiceResponse();
callback(null, response);
}
catch (error) {
console.error('[OTEL gRPC] Error processing traces:', error);
const errorMessage = error instanceof Error ? error.message : 'Unknown error';
const response = new trace_service_1.opentelemetry.proto.collector.trace.v1.ExportTraceServiceResponse();
const partialSuccess = new trace_service_1.opentelemetry.proto.collector.trace.v1.ExportTracePartialSuccess();
partialSuccess.error_message = errorMessage;
response.partial_success = partialSuccess;
callback({
code: grpc.status.INTERNAL,
details: errorMessage,
metadata: new grpc.Metadata(),
}, response);
}
});
}
/**
* Start the gRPC server
* @param port - Port to bind the server to. If not provided, uses the default port from constructor.
*/
start(port) {
const bindPort = port !== null && port !== void 0 ? port : this.defaultPort;
return new Promise((resolve, reject) => {
this.server.bindAsync(`0.0.0.0:${bindPort}`, grpc.ServerCredentials.createInsecure(), (error) => {
if (error) {
console.error(`[OTEL gRPC] Failed to start server on port ${bindPort}:`, error);
reject(error);
return;
}
resolve();
});
});
}
/**
* Stop the gRPC server
*/
stop() {
return new Promise((resolve, reject) => {
this.server.tryShutdown((error) => {
if (error) {
console.error('[OTEL gRPC] Error shutting down server:', error);
reject(error);
}
else {
console.debug('[OTEL gRPC] Server stopped');
resolve();
}
});
});
}
/**
* Force shutdown the gRPC server
*/
forceShutdown() {
this.server.forceShutdown();
console.debug('[OTEL gRPC] Server force shutdown');
}
}
exports.OtelGrpcServer = OtelGrpcServer;