TRF-STB

Author: root

imports.js

@@ -8,6 +8,12 @@ const grpcAddress = require ('./grpc_address')
 const { v4: uuid } = require('uuid');
 const redis = require('redis');
 
+const { MongoClient, ObjectId } = require("mongodb");
+const GridFSBucket = require("mongodb").GridFSBucket;
+
+const MongoDB_URL = "mongodb+srv://andresilmor:[email protected]/?retryWrites=true&w=majority";
+const Redis_URL = 'redis://default:EGGjURloNvz8K6fpudILQdYQWbEV8zhm@redis-19874.c233.eu-west-1-1.ec2.cloud.redislabs.com:19874';
+
 module.exports = {
 EXPRESS: express,
 PARSE_ARGS: parseArgs,
@@ -17,6 +23,11 @@ module.exports = {
 GRPC_CLIENT: grpcClient,
 GRPC_ADDRESS: grpcAddress,
 UUID: uuid,
-REDIS: redis
+REDIS: redis,
+GRID_FS_BUCKET: GridFSBucket,
+MONGO_CLIENT: MongoClient,
+OBJECT_ID: ObjectId,
+MONGO_URL: MongoDB_URL,
+REDIS_URL: Redis_URL
 
 }

package-lock.json

@@ -14,8 +14,10 @@
 "@grpc/proto-loader": "^0.7.4",
 "dotenv": "^16.0.3",
 "express": "^4.18.2",
+"fs": "^0.0.1-security",
 "google-protobuf": "^3.21.2",
 "minimalist": "^1.0.0",
+"mongodb": "^4.16.0",
 "protobufjs": "^7.1.2",
 "redis": "^3.1.2",
 "shortid": "^2.2.16",

package.json

@@ -0,0 +1,6 @@
+package protoImage;
+syntax = "proto3";
+
+message ProtoImage {
+bytes image = 1;
+}

protobufs/messages/ImageInferenceRequest.proto

@@ -70,8 +70,8 @@ const connection = function(connection) {
 
 var inferenceResult
 var request
-protobuf.load("protobufs/messages/ImageInferenceRequest.proto", function(err, root) {
-request = root.lookupType("imageInferenceRequest.ImageInferenceRequest");
+protobuf.load("protobufs/messages/ProtoImage.proto", function(err, root) {
+request = root.lookupType("protoImage.ProtoImage");
 
 //console.log("So it begins...")
 

protobufs/messages/ProtoImage.proto

@@ -33,8 +33,8 @@ const connection = function(connection) {
 var inferenceResult
 var request
 
-protobuf.load("protobufs/messages/ImageInferenceRequest.proto", function(err, root) {
-request = root.lookupType("imageInferenceRequest.ImageInferenceRequest");
+protobuf.load("protobufs/messages/ProtoImage.proto", function(err, root) {
+request = root.lookupType("protoImage.ProtoImage");
 
 var decoded = request.decode(new Uint8Array(data));