| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package org.apache.tuweni.devp2p.v5.dht |
| |
| import org.apache.tuweni.bytes.Bytes |
| import org.apache.tuweni.crypto.Hash |
| import org.apache.tuweni.kademlia.KademliaRoutingTable |
| import org.apache.tuweni.kademlia.xorDist |
| |
| class RoutingTable( |
| selfEnr: Bytes |
| ) { |
| |
| private val selfNodeId = key(selfEnr) |
| private val nodeIdCalculation: (Bytes) -> ByteArray = { enr -> key(enr) } |
| private val table = KademliaRoutingTable( |
| selfId = selfNodeId, |
| k = BUCKET_SIZE, |
| nodeId = nodeIdCalculation, |
| distanceToSelf = { key(it) xorDist selfNodeId }) |
| |
| fun add(enr: Bytes): Bytes? = table.add(enr) |
| |
| fun nearest(nodeId: Bytes, limit: Int): List<Bytes> = table.nearest(nodeId.toArray(), limit) |
| |
| fun evict(enr: Bytes): Boolean = table.evict(enr) |
| |
| fun nodesOfDistance(distance: Int): List<Bytes> = table.peersOfDistance(distance) |
| |
| private fun key(enr: Bytes): ByteArray = Hash.sha2_256(enr).toArray() |
| |
| companion object { |
| private const val BUCKET_SIZE: Int = 16 |
| } |
| } |