Fix keep-alive

Rule activity wasn't updated on data packets incoming

pkg/router/router.go

@@ -403,15 +403,16 @@ func (r *router) handleKeepAlivePacket(_ context.Context, packet routing.Packet)
 
 	r.logger.Infof("Received keepalive packet for route ID %v", routeID)
 
-	rule, err := r.GetRule(routeID)
-
+	_, err := r.GetRule(routeID)
 	if err != nil {
 		return err
 	}
 
-	r.logger.Infof("Route ID %v found, updating activity", routeID)
+	// TODO: propagate keep-alive packet down the chain
+
+	r.logger.Infof("Route ID %v found, updated activity", routeID)
 
-	return r.SaveRoutingRules(rule)
+	return nil
 }
 
 // GetRule gets routing rule.

pkg/routing/table.go

@@ -124,10 +124,10 @@ func (mt *memTable) SaveRule(rule Rule) error {
 }
 
 func (mt *memTable) Rule(key RouteID) (Rule, error) {
-	mt.RLock()
-	rule, ok := mt.rules[key]
-	mt.RUnlock()
+	mt.Lock()
+	defer mt.Unlock()
 
+	rule, ok := mt.rules[key]
 	if !ok {
 		return nil, fmt.Errorf("rule of id %v not found", key)
 	}
@@ -136,6 +136,8 @@ func (mt *memTable) Rule(key RouteID) (Rule, error) {
 		return nil, ErrRuleTimedOut
 	}
 
+	mt.activity[key] = time.Now()
+
 	return rule, nil
 }