feat: externalizing config for Mongo (#7)

* feat: externalizing config for Mongo

* feat: externalizing config

Author: weliasz

config.yaml

@@ -2,24 +2,15 @@ snmp:
   mibs:
     url:
       - "file:///usr/share/snmp/mibs"
-      # *TODO*: Replace below URL with MIB Server's URL returned during service discovery. 
-      # - "http://mibs.thola.io/asn1/@mib@"
-      - "http://localhost:5000/files/asn1/@mib@"
     dir: "/work/mibs/pysnmp"
     # dir: "/Users/yling/snmp-project/splunk-connect-for-snmp-mib-server/mibs/pysnmp"
     load_list: "lookups/mibs_list.csv"
     mibs_path: "/work/mibs"
     # mibs_path: "/Users/yling/snmp-project/splunk-connect-for-snmp-mib-server/mibs"
 mongo:
   oid:
-    # host: "localhost"
-    host: "mongo"
-    port: 27017
     database: "snmp_traps"
     collection: "oids"
   mib:
-    # host: "localhost"
-    host: "mongo"
-    port: 27017
     database: "files"
     collection: "mib_files"

splunk_connect_for_snmp_mib_server/mib_server.py

@@ -18,7 +18,7 @@ def __init__(self, args, server_config):
     def build_flask_app(self):    
         app = Flask(__name__)
 
-        ppath = self._server_config["snmp"]["mibs"]["mibs_path"]
+        ppath = "http://" + os.environ['MONGO_SERVICE_SERVICE_HOST'] + ":" + os.environ['MONGO_SERVICE_PORT'] + "/files/asn1/@mib@"
 
         files_index = AutoIndex(app, ppath, add_url_rules=False)
 

splunk_connect_for_snmp_mib_server/mongo.py

@@ -9,7 +9,7 @@ def __init__(self, mongo_config):
         """
         Create a collection in mongodb to store mib files
         """
-        self._client = MongoClient(host=mongo_config['host'], port=mongo_config['port'])
+        self._client = MongoClient(os.environ['MONGO_SERVICE_SERVICE_HOST'], os.environ['MONGO_SERVICE_PORT'])
         self._mibs = self._client[mongo_config['database']][mongo_config['collection']]
 
 
@@ -62,7 +62,7 @@ def clear(self):
 
 class OidsRepository:
     def __init__(self, mongo_config):
-        self._client = MongoClient(host=mongo_config['host'], port=mongo_config['port'])
+        self._client = MongoClient(os.environ['MONGO_SERVICE_SERVICE_HOST'], os.environ['MONGO_SERVICE_PORT'])
         self._oids = self._client[mongo_config['database']][mongo_config['collection']]
 
     def contains_oid(self, oid):