From a9f6839a04415a36d096f2f66b172ec0aa1fae8b Mon Sep 17 00:00:00 2001
From: zzz <zzz@mail.i2p>
Date: Sun, 15 Mar 2015 13:37:22 +0000
Subject: [PATCH] NetDb: Track flood success

---
 .../FloodfillNetworkDatabaseFacade.java       | 23 +++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/router/java/src/net/i2p/router/networkdb/kademlia/FloodfillNetworkDatabaseFacade.java b/router/java/src/net/i2p/router/networkdb/kademlia/FloodfillNetworkDatabaseFacade.java
index 8b603776f..ccbe092ca 100644
--- a/router/java/src/net/i2p/router/networkdb/kademlia/FloodfillNetworkDatabaseFacade.java
+++ b/router/java/src/net/i2p/router/networkdb/kademlia/FloodfillNetworkDatabaseFacade.java
@@ -221,10 +221,12 @@ public class FloodfillNetworkDatabaseFacade extends KademliaNetworkDatabaseFacad
             DatabaseStoreMessage msg = new DatabaseStoreMessage(_context);
             msg.setEntry(ds);
             OutNetMessage m = new OutNetMessage(_context, msg, _context.clock().now()+FLOOD_TIMEOUT, FLOOD_PRIORITY, target);
-            // note send failure but don't give credit on success
-            // might need to change this
             Job floodFail = new FloodFailedJob(_context, peer);
             m.setOnFailedSendJob(floodFail);
+            // we want to give credit on success, even if we aren't sure,
+            // because otherwise no use noting failure
+            Job floodGood = new FloodSuccessJob(_context, peer);
+            m.setOnSendJob(floodGood);
             _context.commSystem().processMessage(m);
             flooded++;
             if (_log.shouldLog(Log.INFO))
@@ -249,6 +251,23 @@ public class FloodfillNetworkDatabaseFacade extends KademliaNetworkDatabaseFacad
         }
     }
 
+    /**
+     *  Note in the profile that the store succeeded
+     *  @since 0.9.19
+     */
+    private static class FloodSuccessJob extends JobImpl {
+        private final Hash _peer;
+    
+        public FloodSuccessJob(RouterContext ctx, Hash peer) {
+            super(ctx);
+            _peer = peer;
+        }
+        public String getName() { return "Flood succeeded"; }
+        public void runJob() {
+            getContext().profileManager().dbStoreSuccessful(_peer);
+        }
+    }
+
     @Override
     protected PeerSelector createPeerSelector() { return new FloodfillPeerSelector(_context); }