summaryrefslogtreecommitdiffstats
path: root/hicn-light/src/hicn/strategies/loadBalancerWithPD.h
diff options
context:
space:
mode:
Diffstat (limited to 'hicn-light/src/hicn/strategies/loadBalancerWithPD.h')
-rw-r--r--hicn-light/src/hicn/strategies/loadBalancerWithPD.h7
1 files changed, 6 insertions, 1 deletions
diff --git a/hicn-light/src/hicn/strategies/loadBalancerWithPD.h b/hicn-light/src/hicn/strategies/loadBalancerWithPD.h
index d8a215913..e6c9f8271 100644
--- a/hicn-light/src/hicn/strategies/loadBalancerWithPD.h
+++ b/hicn-light/src/hicn/strategies/loadBalancerWithPD.h
@@ -21,10 +21,15 @@
#ifndef loadBalancerWithPD_h
#define loadBalancerWithPD_h
-#include <hicn/core/connectionTable.h>
#include <hicn/strategies/strategyImpl.h>
+#include <hicn/core/connectionTable.h>
+#ifdef WITH_POLICY
+StrategyImpl *strategyLoadBalancerWithPD_Create(const ConnectionTable * table);
+#else
StrategyImpl *strategyLoadBalancerWithPD_Create();
+#endif /* WITH_POLICY */
+
void strategyLoadBalancerWithPD_SetConnectionTable(StrategyImpl *strategy,
ConnectionTable *connTable);
#endif // loadBalancerWithPD_h