write_riemann plugin: Unify reference counting code with other plugins.

diff --git a/src/write_riemann.c b/src/write_riemann.c
index 1196e911324936a2ce220d3c53d15d9fac7a1900..b4ed2c19949345471d706d418a5da245a3ac22fe 100644 (file)
--- a/src/write_riemann.c
+++ b/src/write_riemann.c
@@ -617,15 +617,14 @@ static void wrr_free(void *p) /* {{{ */
 
   pthread_mutex_lock(&host->lock);
   host->reference_count--;
-  int reference_count = host->reference_count;
-  pthread_mutex_unlock(&host->lock);
-
-  if (reference_count > 0) {
+  if (host->reference_count > 0) {
+    pthread_mutex_unlock(&host->lock);
     return;
   }
 
   wrr_disconnect(host);
 
+  pthread_mutex_unlock(&host->lock);
   pthread_mutex_destroy(&host->lock);
   sfree(host);
 } /* }}} void wrr_free */