From 50febe38b6c0f75076cb19bbd8b4d3fe3aa903b8 Mon Sep 17 00:00:00 2001
From: slioch <slioch@eng-140.vyatta.com>
Date: Tue, 7 Jul 2009 13:02:31 -0700
Subject: fix for bug 4587. Remove output file on orderly shutdown.

---
 src/lboutput.cc    | 7 +++++++
 src/lboutput.hh    | 3 +++
 src/loadbalance.cc | 1 +
 3 files changed, 11 insertions(+)

(limited to 'src')

diff --git a/src/lboutput.cc b/src/lboutput.cc
index ff915ec..e673ee0 100644
--- a/src/lboutput.cc
+++ b/src/lboutput.cc
@@ -14,6 +14,13 @@
 #include "lbdata.hh"
 #include "lboutput.hh"
 
+void
+LBOutput::shutdown()
+{
+  string wlb_out = _output_path + "/wlb.out";
+  unlink(wlb_out.c_str());
+}
+
 void
 LBOutput::write(const LBData &lbdata)
 {
diff --git a/src/lboutput.hh b/src/lboutput.hh
index 29b9a00..f79c024 100644
--- a/src/lboutput.hh
+++ b/src/lboutput.hh
@@ -18,6 +18,9 @@ public:
   LBOutput(bool debug, string &output_path) : _debug(debug), _output_path(output_path) {}
   ~LBOutput() {}
 
+  void
+  shutdown();
+
   void
   write(const LBData &lbdata);
 
diff --git a/src/loadbalance.cc b/src/loadbalance.cc
index 0324c08..9b2c8a5 100644
--- a/src/loadbalance.cc
+++ b/src/loadbalance.cc
@@ -30,6 +30,7 @@ LoadBalance::LoadBalance(bool debug, string &output_path) :
 LoadBalance::~LoadBalance()
 {
   _decision.shutdown(_lbdata);
+  _output.shutdown();
 }
 
 /**
-- 
cgit v1.2.3