Java DHTConfiguration类代码示例

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
void initKey(DHTConfiguration config)
{
	if(config != null && config.isPersistingID()) {
		Path keyPath = config.getStoragePath().resolve("baseID.config");
		File keyFile = keyPath.toFile();
		
		if (keyFile.exists() && keyFile.isFile()) {
			try {
				List<String> raw = Files.readAllLines(keyPath);
				baseKey = raw.stream().map(String::trim).filter(Key.STRING_PATTERN.asPredicate()).findAny().map(Key::new).orElseThrow(() -> new IllegalArgumentException(keyPath.toString()+" did not contain valid node ID"));
				return;
			} catch (Exception e) {
				e.printStackTrace();
			}
		}
	}
	
	baseKey = Key.createRandomKey();
	
	persistKey();
}
 

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
private DHTConfiguration toMldhtConfig(DHTConfig config) {
    return new DHTConfiguration() {
        @Override
        public boolean isPersistingID() {
            return false;
        }

        @Override
        public Path getStoragePath() {
            return Files.createTempDir().toPath();
        }

        @Override
        public int getListeningPort() {
            return config.getListeningPort();
        }

        @Override
        public boolean noRouterBootstrap() {
            return true;
        }

        @Override
        public boolean allowMultiHoming() {
            return false;
        }

        @Override
        public Predicate<InetAddress> filterBindAddress() {
            return address -> {
                boolean bothAnyLocal = address.isAnyLocalAddress() && localAddress.isAnyLocalAddress();
                boolean couldUse = bothAnyLocal || localAddress.equals(address);
                if (LOGGER.isDebugEnabled()) {
                    LOGGER.debug("Filtering addresses to bind DHT server to.. Checking " + address + ".. Could use: " + couldUse);
                }
                return couldUse;
            };
        }
    };
}
 

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
static DHT buildDHT(DHT.DHTtype type) {
	DHT dht = new DHT(type);
	dht.config = new DHTConfiguration() {
		
		@Override
		public boolean noRouterBootstrap() {
			// TODO Auto-generated method stub
			return false;
		}
		
		@Override
		public boolean isPersistingID() {
			// TODO Auto-generated method stub
			return false;
		}
		
		@Override
		public Path getStoragePath() {
			// TODO Auto-generated method stub
			return Paths.get(".", "does", "not", "exist");
		}
		
		@Override
		public int getListeningPort() {
			// TODO Auto-generated method stub
			return 0;
		}
		
		@Override
		public boolean allowMultiHoming() {
			// TODO Auto-generated method stub
			return false;
		}
	};
	dht.populate();
	
	return dht;
	
}
 

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
private static DHTConfiguration getConfiguration() {
    return new DHTConfiguration() {

        @Override
        public boolean noRouterBootstrap() {
            return false;
        }

        @Override
        public boolean isPersistingID() {
            return true;
        }

        @Override
        public File getNodeCachePath() {
            return new File("/tmp/dht");
        }

        @Override
        public int getListeningPort() {
            return 5556;
        }

        @Override
        public boolean allowMultiHoming() {
            return false;
        }
    };
}
 

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
public DHTConfiguration getConfig() {
	return config;
}
 

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
public void start (DHTConfiguration config)
		throws SocketException {
	if (running) {
		return;
	}
	
	if(this.scheduler == null)
		this.scheduler = getDefaultScheduler();
	this.config = config;
	useRouterBootstrapping = !config.noRouterBootstrap();
	
	if(!Files.isDirectory(config.getStoragePath()))
		DHT.log("Warning: storage path " + config.getStoragePath() +" is not a directory. DHT will not be able to persist state" , LogLevel.Info);
	
	table_file = config.getStoragePath().resolve(type.shortName+"-table.cache");

	setStatus(DHTStatus.Stopped, DHTStatus.Initializing);
	stats.resetStartedTimestamp();

	logInfo("Starting DHT on port " + getPort());

	// we need the IPs to filter bootstrap nodes out from the routing table. but don't block startup on DNS resolution
	scheduler.execute(this::resolveBootstrapAddresses);
	
	connectionManager = new NIOConnectionManager("mlDHT "+type.shortName+" NIO Selector");
	
	populate();
	
	node.initKey(config);
	node.loadTable(table_file);
	

	// these checks are fairly expensive on large servers (network interface enumeration)
	// schedule them separately
	scheduledActions.add(scheduler.scheduleWithFixedDelay(serverManager::doBindChecks, 10, 10, TimeUnit.SECONDS));
	
	scheduledActions.add(scheduler.scheduleWithFixedDelay(() -> {
		// maintenance that should run all the time, before the first queries
		tman.dequeue();

		if (running)
			onStatsUpdate();
	}, 5000, DHTConstants.DHT_UPDATE_INTERVAL, TimeUnit.MILLISECONDS));

	// initialize as many RPC servers as we need
	serverManager.refresh(System.currentTimeMillis());
	
	if(serverManager.getServerCount() == 0) {
		logError("No network interfaces eligible for DHT sockets found during startup."
				+ "\nAddress family: " + this.getType()
				+ "\nmultihoming [requires public IP addresses if enabled]: " + config.allowMultiHoming()
				+ "\nPublic IP addresses: " + AddressUtils.getAvailableGloballyRoutableAddrs(getType().PREFERRED_ADDRESS_TYPE)
				+ "\nDefault route: " + AddressUtils.getDefaultRoute(getType().PREFERRED_ADDRESS_TYPE));
	}
	
	started();

}
 

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
void persistKey() {
	DHTConfiguration config = dht.getConfig();
	
	if(config == null)
		return;
	
	Path keyPath = config.getStoragePath().resolve("baseID.config");
	
	try {
		if(!Files.isDirectory(config.getStoragePath()))
			return;
		Path tmpFile = Files.createTempFile(config.getStoragePath(), "baseID", ".tmp");
		
		Files.write(tmpFile, Collections.singleton(baseKey.toString(false)), StandardCharsets.ISO_8859_1);
		Files.move(tmpFile, keyPath, StandardCopyOption.ATOMIC_MOVE, StandardCopyOption.REPLACE_EXISTING);
		
	} catch (IOException e) {
		// TODO Auto-generated catch block
		e.printStackTrace();
	}
	
	
	
	
	
}
 

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
@Test
public void testBindFilter() {
	DHT dht = NodeFactory.buildDHT(DHTtype.IPV4_DHT);
	dht.getNode().initKey(null);
	dht.setScheduler(Executors.newSingleThreadScheduledExecutor());
	AtomicReference<Predicate<InetAddress>> predicate = new AtomicReference<>(null);
	
	dht.config = new DHTConfiguration() {
		
		@Override
		public boolean noRouterBootstrap() {
			return true;
		}
		
		@Override
		public boolean isPersistingID() {
			// TODO Auto-generated method stub
			return false;
		}
		
		@Override
		public Path getStoragePath() {
			// TODO Auto-generated method stub
			return null;
		}
		
		@Override
		public int getListeningPort() {
			// TODO Auto-generated method stub
			return 0;
		}
		
		@Override
		public boolean allowMultiHoming() {
			return false;
		}
		
		@Override
		public Predicate<InetAddress> filterBindAddress() {
			return predicate.get();
		}
	};
	
	predicate.set((unused) -> true);
	RPCServerManager srvman = dht.getServerManager();
	srvman.refresh(System.currentTimeMillis());
	assertNotEquals(0, srvman.getServerCount());
	
	predicate.set((unused) -> false);
	srvman.doBindChecks();
	assertEquals(0, srvman.getServerCount());
	
	// wildcard addr same as allowing all addresses of that family
	predicate.set((addr) -> addr instanceof Inet4Address && addr.isAnyLocalAddress());
	srvman.refresh(System.currentTimeMillis());
	assertNotEquals(0, srvman.getServerCount());
	
	// select a specific address
	InetAddress localAddr = AddressUtils.nonlocalAddresses().filter(Inet4Address.class::isInstance).findAny().get();
	predicate.set((addr) -> addr.equals(localAddr));
	srvman.doBindChecks();
	srvman.refresh(System.currentTimeMillis());
	assertEquals(localAddr, srvman.getAllServers().get(0).getBindAddress());
	
	
}
 

import lbms.plugins.mldht.DHTConfiguration; //导入依赖的package包/类
/**
 * Start the DHT
 */
void start (DHTConfiguration config) throws SocketException;