sonic_ax_impl supports redis connection options (sonic-net#17)

src/sonic_ax_impl/__main__.py

@@ -7,6 +7,7 @@
 
 import ax_interface
 import sonic_ax_impl
+from . import mibs
 
 LOG_FORMAT = "snmp-subagent [%(name)s] %(levelname)s: %(message)s"
 
@@ -63,6 +64,9 @@ def install_fragments():
     ax_interface.logger.handlers = sonic_ax_impl.logger.handlers
     sswsdk.logger.handlers = sonic_ax_impl.logger.handlers
 
+    # init mibs module with command line arguments
+    mibs.config(**args)
+
     if warn_syslog:
         # syslog was unavailable when it should've been.
         sonic_ax_impl.logger.warning("Syslog is unavailable. Logging to STDERR.")

src/sonic_ax_impl/mibs/__init__.py

@@ -11,6 +11,7 @@
 ASIC_DB = 'ASIC_DB'
 COUNTERS_DB = 'COUNTERS_DB'
 
+redis_kwargs = {'unix_socket_path': '/var/run/redis/redis.sock'}
 
 def counter_table(sai_id):
     """
@@ -39,14 +40,18 @@ def get_index(if_name):
         return int(n) + 1
 
 
+def config(**kwargs):
+    global redis_kwargs
+    redis_kwargs = {k:v for (k,v) in kwargs.items() if k in ['unix_socket_path', 'host', 'port']}
+
 def init_sync_d_interface_tables():
     """
     DRY helper method. Connects to and initializes interface maps for SyncD-connected MIB(s).
     :return: tuple(db_conn, if_name_map, if_id_map, oid_map, if_alias_map)
     """
     # SyncD database connector. THIS MUST BE INITIALIZED ON A PER-THREAD BASIS.
     # Redis PubSub objects (such as those within sswsdk) are NOT thread-safe.
-    db_conn = SonicV2Connector()
+    db_conn = SonicV2Connector(**redis_kwargs)
     db_conn.connect(COUNTERS_DB)
 
     # { if_name (SONiC) -> sai_id }