show "GW: START"
show system "pwd"

show "Command Line Arguments..."

params:.Q.opt .z.X
show params

/ read in params
dbname:first params`dbname
codebase:first params`codebase
rdb_name:first params`rdb_name
hdb_name:first params`hdb_name

/ assign paths
app_path: "/opt/kx/app"

dbpath: app_path, "/db/", dbname
codepath: app_path, "/code/", codebase

/ If database exists, mount it
$[count key hsym `$dbpath;[ show "loading database: ", dbpath; system "l ", dbpath;];
    [show "no database at: ", dbpath;]]

/ if code directory exists, cd to it
$[count key hsym `$codepath;[ show "cd to code directory: ", codepath; system "cd ", codepath;];
    [show "no code at: ", codepath;]];

/ BEGIN load libraries relative to the codepath

/ example gateway process

/ load the connect library
\l connectmkdb.q

/This lib should not be compiled as has func to send down the handle
\l funcDownHandle.q

/func to query gw with to dispact to random rdb and hdb
queryData:{[table;syms]

    targets:.gw.getTargets[];

    results:targets@\:(`.query.data;table;syms);
    (uj/) results
    }

/get handles for a hdb and rdb
.gw.getTargets:{[]

   rdbs:0Ni^exec handle from .conn.procs where connected,process=`rdb;
   hdbs:0Ni^exec handle from .conn.procs where connected,process=`hdb;

   targets:raze 1?/:(rdbs;hdbs);
   targets:targets where not null targets;
   if[not count targets;
        '"No available data nodes";
        ];
    targets
    };

/ Every 10 seconds connect to data nodes unless all are connected.
.gw.connectToDataNodes:{[zx]
    .conn.connectToProcs[`rdb`hdb;zx];
    };

.gw.getDataNodes:{[hdb_name; rdb_name]
    hdb_nodes: .aws.list_kx_cluster_nodes[hdb_name];
    rdb_nodes: .aws.list_kx_cluster_nodes[rdb_name];

    rdb_conn_strs: {.aws.get_kx_node_connection_string[rdb_name;x]} each rdb_nodes`node_id;
    hdb_conn_strs: {.aws.get_kx_node_connection_string[hdb_name;x]} each hdb_nodes`node_id;

    raze (enlist "-hdb"; hdb_conn_strs; enlist "-rdb"; rdb_conn_strs)
    };
    

init:{[hdb_name; rdb_name]
    zx: .gw.getDataNodes[hdb_name; rdb_name];

    .gw.connectToDataNodes[zx];

    .z.ts:{[]
        /If not all nodes connected - attempt to reconnect
        if[not all .conn.procs`connected;
            show"Attempting to connect to disconnected data nodes...";
            .conn.connectDisconnected[];
            if[all .conn.procs`connected;
                show"All connected!";
                ];
            ];
        };

    .z.pc:.conn.handleDrop;

    /Attempt recon every 10s
    system"t 10000";
    }

reinit:{[hdb_name; rdb_name]
    delete from `.conn.procs;
    init[hdb_name; rdb_name];
    }

init[hdb_name; rdb_name]

/ must be in this path for db reads to work
system "cd /opt/kx"

/ count partitioned tables
count each value each tables[]

show "HDB: DONE"
show system "pwd"