diff --git a/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/MongoDBPlugin.java b/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/MongoDBPlugin.java new file mode 100644 index 00000000..aee15254 --- /dev/null +++ b/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/MongoDBPlugin.java @@ -0,0 +1,86 @@ +package org.bench4q.agent.plugin.basic.MongoDB; + +import java.net.UnknownHostException; + +import org.bench4q.agent.plugin.Constructor; +import org.bench4q.agent.plugin.Parameter; +import org.bench4q.agent.plugin.Plugin; +import org.bench4q.agent.plugin.behavior.Behavior; +import org.bench4q.agent.plugin.behavior.BehaviorType; +import org.bench4q.agent.utils.Type.SupportTypes; + +import com.mongodb.BasicDBObject; +import com.mongodb.DB; +import com.mongodb.DBCollection; +import com.mongodb.DBCursor; +import com.mongodb.DBObject; +import com.mongodb.Mongo; +import com.mongodb.util.JSON; + +@Plugin("MongoDBPlugin") +public class MongoDBPlugin { + + @Constructor + public MongoDBPlugin() { + } + + @Behavior(value = "Insert", type = BehaviorType.USER_BEHAVIOR) + public MongoDBReturn insert( + @Parameter(value = "key", type = SupportTypes.Field) String key, + @Parameter(value = "value", type = SupportTypes.Field) String value, + @Parameter(value = "hostName", type = SupportTypes.Field) String hostName, + @Parameter(value = "port", type = SupportTypes.Field) int port, + @Parameter(value = "dbName", type = SupportTypes.Field) String dbName) { + // TODO：测试MongoDB的插入性能 + + int recordsum = 0; + try { + @SuppressWarnings("deprecation") + Mongo mongo = new Mongo(hostName, port); + DB db = mongo.getDB(dbName); + DBCollection users; + // 获取users DBCollection；如果默认没有创建，mongodb会自动创建 + users = db.getCollection("users"); + + DBObject user = new BasicDBObject(); + user.put(key, value); + users.save(user); + + DBCollection test = db.getCollection("users"); + // 查询所有的数据 + DBCursor cur = test.find(); + while (cur.hasNext()) { + System.out.println(cur.next()); + } + + recordsum = test.find().count(); + System.out.println("总数目： " + cur.count()); + System.out.println("当前游标的id号: " + cur.getCursorId()); + System.out.println(JSON.serialize(cur)); + + if (mongo != null) { + mongo.close(); + } + mongo = null; + db = null; + users = null; + + } catch (UnknownHostException e) { + e.printStackTrace(); + } + return new MongoDBReturn(recordsum > 0); + } + /* + * @Behavior(value = "Query", type = BehaviorType.USER_BEHAVIOR) public + * MongoDBReturn query( + * + * @Parameter(value = "queryString", type = SupportTypes.Field) String + * queryString, + * + * @Parameter(value = "attributes", type = SupportTypes.Field) List + * attributes){ + * + * + * return null; } + */ +} diff --git a/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/MongoDBReturn.java b/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/MongoDBReturn.java new file mode 100644 index 00000000..76606307 --- /dev/null +++ b/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/MongoDBReturn.java @@ -0,0 +1,9 @@ +package org.bench4q.agent.plugin.basic.MongoDB; + +import org.bench4q.agent.plugin.basic.PluginReturn; + +public class MongoDBReturn extends PluginReturn { + public MongoDBReturn(boolean success) { + super(success); + } +} diff --git a/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/ui.xml b/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/ui.xml new file mode 100644 index 00000000..d2546492 --- /dev/null +++ b/Bench4Q-Agent/src/main/java/org/bench4q/agent/plugin/basic/MongoDB/ui.xml @@ -0,0 +1,27 @@ + + + + + + + + + + + + + + + + + + + + + + + + + + + \ No newline at end of file diff --git a/Bench4Q-Agent/src/test/java/org/bench4q/agent/test/plugin/Test_MongoDBPlugin.java b/Bench4Q-Agent/src/test/java/org/bench4q/agent/test/plugin/Test_MongoDBPlugin.java new file mode 100644 index 00000000..615ad9e2 --- /dev/null +++ b/Bench4Q-Agent/src/test/java/org/bench4q/agent/test/plugin/Test_MongoDBPlugin.java @@ -0,0 +1,83 @@ +package org.bench4q.agent.test.plugin; + +import static org.junit.Assert.*; + +import java.io.File; +import java.io.IOException; +import java.util.Arrays; +import java.util.LinkedList; +import java.util.UUID; + +import org.apache.commons.io.FileUtils; +import org.bench4q.agent.plugin.basic.MongoDB.MongoDBPlugin; +import org.bench4q.agent.plugin.basic.MongoDB.MongoDBReturn; +import org.bench4q.agent.scenario.Scenario; +import org.bench4q.agent.scenario.VUser; +import org.bench4q.agent.test.TestBase; +import org.bench4q.share.helper.MarshalHelper; +import org.bench4q.share.helper.TestHelper; +import org.bench4q.share.models.agent.ParameterModel; +import org.bench4q.share.models.agent.RunScenarioModel; +import org.bench4q.share.models.agent.scriptrecord.BatchModel; +import org.bench4q.share.models.agent.scriptrecord.BehaviorModel; +import org.bench4q.share.models.agent.scriptrecord.PageModel; +import org.bench4q.share.models.agent.scriptrecord.UsePluginModel; +import org.junit.Test; +import org.junit.runner.RunWith; +import org.springframework.test.context.ContextConfiguration; +import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; + +@RunWith(SpringJUnit4ClassRunner.class) +@ContextConfiguration(locations = { "classpath:application-context.xml" }) +public class Test_MongoDBPlugin extends TestBase { + @Test + public void test_Insert() { + MongoDBPlugin mongoDBPlugin = new MongoDBPlugin(); + MongoDBReturn result = mongoDBPlugin.insert("key1", "value1", + "localhost", 27017, "temp"); + assertTrue(result.getSuccessCount() > 0); + } + + private RunScenarioModel buildUpMongoDBScenario() throws IOException { + RunScenarioModel scenarioModel = new RunScenarioModel(); + scenarioModel.setUsePlugins(new LinkedList()); + UsePluginModel mongoDB = new UsePluginModel(); + String pluginId = "mongo1"; + mongoDB.setId(pluginId); + mongoDB.setName("MongoDBPlugin"); + mongoDB.setParameters(new LinkedList()); + scenarioModel.getUsePlugins().add(mongoDB); + + scenarioModel.setPages(new LinkedList()); + PageModel page1 = new PageModel(); + BatchModel batch = new BatchModel(); + page1.getBatches().add(batch); + scenarioModel.getPages().add(page1); + + batch.getBehaviors().add( + BehaviorModel.UserBehaviorBuilder(0, "Insert", pluginId, Arrays + .asList(ParameterModel.createParameter("key", "key2"), + ParameterModel.createParameter("value", + "value2"), ParameterModel + .createParameter("hostName", + "localhost"), ParameterModel + .createParameter("port", "27017"), + ParameterModel + .createParameter("dbName", "temp")))); + String scriptContent = MarshalHelper.tryMarshal(scenarioModel); + System.out.println(scriptContent); + File targetFile = new File("Scripts" + + System.getProperty("file.separator") + "mongoScript.xml"); + TestHelper.createFileIfNotExist(targetFile); + FileUtils.writeStringToFile(targetFile, scriptContent); + return scenarioModel; + } + + @Test + public void test_BuildScenario() throws IOException { + RunScenarioModel result = buildUpMongoDBScenario(); + VUser vUser = createVUser(Scenario.scenarioBuilderWithCompile(result), + UUID.randomUUID()); + vUser.run(); + } +}