Method testBench

ZezeJava/ZezeJavaTest/src/Dbh2/Dbh2FullTest.java:41–97

()

	}

	@Test
	public void testBench() throws Exception {
		System.setProperty("Dbh2MasterDefaultBucketPortId", "18000");
		Task.tryInitThreadPool();
		Zeze.Net.Selectors.getInstance().add(7);

		var master = new Zeze.Dbh2.Master.Main("zeze.xml");
		var managers = new ArrayList<Dbh2Manager>();
		var serviceManager = Application.createServiceManager(Config.load(), "Dbh2ServiceManager");
		assert serviceManager != null;
		serviceManager.start();
		serviceManager.waitReady();

		var value = ByteBuffer.Wrap(new byte[]{1, 2, 3, 4});
		Database database = null;
		Application.renameAndDeleteDirectory(new File("CommitRocks"));
		var dbh2AgentManager = new Dbh2AgentManager(serviceManager, null, 100);
		try {
			master.start();
			for (int i = 0; i < 3; ++i)
				managers.add(new Zeze.Dbh2.Dbh2Manager("manager" + i, "zeze" + i + ".xml"));
			for (var manager : managers)
				manager.start();
			dbh2AgentManager.start();

			database = newDatabase(dbh2AgentManager, "dbh2TestDb");
			var tables = new ArrayList<AbstractKVTable>();
			for (int i = 0; i < 4; ++i) {
				var tableName = "table" + i;
				tables.add((Database.AbstractKVTable)database.openTable(tableName, Bean.hash32(tableName)));
			}
			for (var table : tables)
				table.waitReady();

			var count = 1000;
			var threads = 2;
			var futures = new ArrayList<Future<?>>();
			var b = new Zeze.Util.Benchmark();
			for (var t = 0; t < threads; ++t) {
				var keyStart = t * count;
				var keyEnd = keyStart + count;
				futures.add(startBench(keyStart, keyEnd, database, tables.get(t % tables.size()), value));
			}
			Thread.sleep(1000); // 等待agent都连上，然后dump出来。此时任务在并发执行。
			dbh2AgentManager.dumpAgents();
			for (var future : futures)
				future.get();
			b.report("Bench Dbh2 Full Transaction", count * threads);
		} finally {
			master.stop();
			for (var manager : managers)
				manager.stop();
			if (database != null)
				database.close();
			dbh2AgentManager.stop();
		}
	}