Skip to content

Commit

Permalink
[#9027] Update mongodb plugin
Browse files Browse the repository at this point in the history
  • Loading branch information
@jaehong-kim
jaehong-kim committed Jul 11, 2022
1 parent 549daa1 commit 3a14ae8
Show file tree
Hide file tree
Showing 46 changed files with 2,666 additions and 1,139 deletions.
31 changes: 21 additions & 10 deletions plugins-it/mongodb-it/pom.xml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -29,11 +29,22 @@
<packaging>jar</packaging>

<properties>
<jdk.version>1.7</jdk.version>
<jdk.home>${env.JAVA_7_HOME}</jdk.home>
<jdk.version>1.8</jdk.version>
<jdk.home>${env.JAVA_8_HOME}</jdk.home>
</properties>

<dependencies>
<dependency>
<groupId>com.navercorp.pinpoint</groupId>
<artifactId>pinpoint-plugin-it-utils</artifactId>
<version>${project.version}</version>
</dependency>
<dependency>
<groupId>com.navercorp.pinpoint</groupId>
<artifactId>pinpoint-plugin-it-jdbc-test</artifactId>
<version>${project.version}</version>
</dependency>

<dependency>
<groupId>com.navercorp.pinpoint</groupId>
<artifactId>pinpoint-mongodb-driver-plugin</artifactId>
Expand All @@ -51,18 +62,18 @@
<artifactId>mongodb-driver</artifactId>
<scope>test</scope>
</dependency>

<dependency>
<groupId>de.flapdoodle.embed</groupId>
<artifactId>de.flapdoodle.embed.mongo</artifactId>
<version>1.50.5</version>
<groupId>org.testcontainers</groupId>
<artifactId>testcontainers</artifactId>
<version>${testcontainers.version}</version>
<scope>test</scope>
</dependency>


<dependency>
<groupId>com.navercorp.pinpoint</groupId>
<artifactId>pinpoint-plugin-it-utils</artifactId>
<version>${project.version}</version>
<groupId>org.testcontainers</groupId>
<artifactId>mongodb</artifactId>
<version>${testcontainers.version}</version>
<scope>test</scope>
</dependency>
</dependencies>

Expand Down
316 changes: 22 additions & 294 deletions plugins-it/mongodb-it/src/test/java/com/navercorp/pinpoint/plugin/mongodb/MongoDBITBase.java
View file

Large diffs are not rendered by default.

290 changes: 290 additions & 0 deletions ...ns-it/mongodb-it/src/test/java/com/navercorp/pinpoint/plugin/mongodb/MongoDBITHelper.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,290 @@
/*
* Copyright 2022 NAVER Corp.
*
* Licensed 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 com.navercorp.pinpoint.plugin.mongodb;

import com.mongodb.WriteConcern;
import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoCursor;
import com.mongodb.client.MongoDatabase;
import com.mongodb.client.result.DeleteResult;
import com.navercorp.pinpoint.bootstrap.plugin.test.ExpectedAnnotation;
import com.navercorp.pinpoint.bootstrap.plugin.test.PluginTestVerifier;
import com.navercorp.pinpoint.bootstrap.plugin.test.PluginTestVerifierHolder;
import com.navercorp.pinpoint.common.util.StringStringValue;
import com.navercorp.pinpoint.plugin.mongo.MongoConstants;
import com.navercorp.pinpoint.plugin.mongo.MongoUtil;
import com.navercorp.pinpoint.plugin.mongo.NormalizedBson;
import org.bson.BsonArray;
import org.bson.BsonBinary;
import org.bson.BsonBinarySubType;
import org.bson.BsonBoolean;
import org.bson.BsonDateTime;
import org.bson.BsonDbPointer;
import org.bson.BsonDecimal128;
import org.bson.BsonDocument;
import org.bson.BsonDouble;
import org.bson.BsonInt32;
import org.bson.BsonInt64;
import org.bson.BsonJavaScript;
import org.bson.BsonJavaScriptWithScope;
import org.bson.BsonNull;
import org.bson.BsonObjectId;
import org.bson.BsonRegularExpression;
import org.bson.BsonString;
import org.bson.BsonSymbol;
import org.bson.BsonTimestamp;
import org.bson.BsonUndefined;
import org.bson.BsonValue;
import org.bson.Document;
import org.bson.conversions.Bson;
import org.bson.types.Decimal128;
import org.bson.types.ObjectId;
import org.junit.Assert;

import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.Date;

import static com.mongodb.client.model.Filters.and;
import static com.mongodb.client.model.Filters.eq;
import static com.mongodb.client.model.Filters.exists;
import static com.mongodb.client.model.Filters.nin;
import static com.navercorp.pinpoint.bootstrap.plugin.test.Expectations.event;

public class MongoDBITHelper {
protected static final String MONGO_EXECUTE_QUERY = "MONGO_EXECUTE_QUERY";

public void testConnection30(MongoDBITBase mongoDBITBase, String address, final MongoDatabase database, Class<?> mongoDatabaseImplClass, String secondCollectionDefaultOption) throws Exception {
PluginTestVerifier verifier = PluginTestVerifierHolder.getInstance();

MongoCollection<Document> collection = database.getCollection("customers");
MongoCollection<Document> collection2 = database.getCollection("customers2").withWriteConcern(WriteConcern.ACKNOWLEDGED);

insertComlexBsonValueData30(verifier, address, collection, mongoDatabaseImplClass, "customers", "MAJORITY");
insertData(verifier, address, collection, mongoDatabaseImplClass, "customers", "MAJORITY");
insertData(verifier, address, collection2, mongoDatabaseImplClass, "customers2", secondCollectionDefaultOption);
updateData(verifier, address, collection, mongoDatabaseImplClass);
readData(verifier, address, collection, mongoDatabaseImplClass);
filterData(verifier, address, collection, mongoDatabaseImplClass);
filterData2(verifier, address, collection, mongoDatabaseImplClass);
deleteData(verifier, address, collection, mongoDatabaseImplClass);

stopDB(collection);
}

public void testConnection34(MongoDBITBase mongoDBITBase, String address, final MongoDatabase database, Class<?> mongoDatabaseImplClass, String secondCollectionDefaultOption) throws Exception {
PluginTestVerifier verifier = PluginTestVerifierHolder.getInstance();

MongoCollection<Document> collection = database.getCollection("customers");
MongoCollection<Document> collection2 = database.getCollection("customers2").withWriteConcern(WriteConcern.ACKNOWLEDGED);

insertComlexBsonValueData34(verifier, address, collection, mongoDatabaseImplClass, "customers", "MAJORITY");
insertData(verifier, address, collection, mongoDatabaseImplClass, "customers", "MAJORITY");
insertData(verifier, address, collection2, mongoDatabaseImplClass, "customers2", secondCollectionDefaultOption);
updateData(verifier, address, collection, mongoDatabaseImplClass);
readData(verifier, address, collection, mongoDatabaseImplClass);
filterData(verifier, address, collection, mongoDatabaseImplClass);
filterData2(verifier, address, collection, mongoDatabaseImplClass);
deleteData(verifier, address, collection, mongoDatabaseImplClass);

stopDB(collection);
}

public void stopDB(MongoCollection<Document> collection) throws Exception {
try {
collection.drop();
} catch (Exception ex) {
throw new RuntimeException("drop() failure", ex);
}
}

public void insertComlexBsonValueData30(PluginTestVerifier verifier, String address, MongoCollection<Document> collection, Class<?> mongoDatabaseImpl, String collectionInfo, String collectionOption) {
//insert Data
Document document = createComplexDocument();

collection.insertOne(document);

Method insertOneMethod = getMethod(mongoDatabaseImpl, "insertOne", Object.class);
NormalizedBson parsedBson = parseBson(document);

verifier.verifyTrace(event(MONGO_EXECUTE_QUERY, insertOneMethod, null, address, null
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_INFO.getName(), collectionInfo)
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_OPTION.getName(), collectionOption)
, new ExpectedAnnotation(MongoConstants.MONGO_JSON_DATA.getName(), new StringStringValue(parsedBson.getNormalizedBson(), parsedBson.getParameter()))));
}

public NormalizedBson parseBson(Object... documents) {
Object[] objects = Arrays.copyOf(documents, documents.length);
return MongoUtil.parseBson(objects, true);
}

private Document createComplexDocument() {
//insert Data
BsonValue a = new BsonString("stest");
BsonValue b = new BsonDouble(111);
BsonValue c = new BsonBoolean(true);

Document document = new Document()
.append("int32", new BsonInt32(12))
.append("int64", new BsonInt64(77L))
.append("bo\"olean", new BsonBoolean(true))
.append("date", new BsonDateTime(new Date().getTime()))
.append("double", new BsonDouble(12.3))
.append("string", new BsonString("pinpoint"))
.append("objectId", new BsonObjectId(new ObjectId()))
.append("code", new BsonJavaScript("int i = 10;"))
.append("codeWithScope", new BsonJavaScriptWithScope("int x = y", new BsonDocument("y", new BsonInt32(1))))
.append("regex", new BsonRegularExpression("^test.*regex.*xyz$", "big"))
.append("symbol", new BsonSymbol("wow"))
.append("timestamp", new BsonTimestamp(0x12345678, 5))
.append("undefined", new BsonUndefined())
.append("binary1", new BsonBinary(new byte[]{(byte) 0xe0, 0x4f, (byte) 0xd0, 0x20}))
.append("oldBinary", new BsonBinary(BsonBinarySubType.OLD_BINARY, new byte[]{1, 1, 1, 1, 1}))
.append("arrayInt", new BsonArray(Arrays.asList(a, b, c, new BsonInt32(7))))
.append("document", new BsonDocument("a", new BsonInt32(77)))
.append("dbPointer", new BsonDbPointer("db.coll", new ObjectId()))
.append("null", new BsonNull());
return document;
}

private Method getMethod(Class<?> mongoDatabaseImpl, String name, Class<?>... parameterTypes) {
try {
return mongoDatabaseImpl.getDeclaredMethod(name, parameterTypes);
} catch (NoSuchMethodException ex) {
throw new RuntimeException(ex);
}
}

public void insertComlexBsonValueData34(PluginTestVerifier verifier, String address, MongoCollection<Document> collection, Class<?> mongoDatabaseImpl, String collectionInfo, String collectionOption) {
//insert Data
Document document = createComplexDocument();
document.append("decimal128", new BsonDecimal128(new Decimal128(55)));

collection.insertOne(document);

Method insertOneMethod = getMethod(mongoDatabaseImpl, "insertOne", Object.class);
NormalizedBson parsedBson = parseBson(document);

verifier.verifyTrace(event(MONGO_EXECUTE_QUERY, insertOneMethod, null, address, null
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_INFO.getName(), collectionInfo)
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_OPTION.getName(), collectionOption)
, new ExpectedAnnotation(MongoConstants.MONGO_JSON_DATA.getName(), new StringStringValue(parsedBson.getNormalizedBson(), parsedBson.getParameter()))));
}

public void insertData(PluginTestVerifier verifier, String address, MongoCollection<Document> collection, Class<?> mongoDatabaseImpl, String collectionInfo, String collectionOption) {
//insert Data
Document doc = new Document("name", "Roy").append("company", "Naver");
collection.insertOne(doc);

Method insertOneMethod = getMethod(mongoDatabaseImpl, "insertOne", Object.class);
NormalizedBson parsedBson = parseBson(doc);

verifier.verifyTrace(event(MONGO_EXECUTE_QUERY, insertOneMethod, null, address, null
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_INFO.getName(), collectionInfo)
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_OPTION.getName(), collectionOption)
, new ExpectedAnnotation(MongoConstants.MONGO_JSON_DATA.getName(), new StringStringValue(parsedBson.getNormalizedBson(), parsedBson.getParameter()))));
}

public void updateData(PluginTestVerifier verifier, String address, MongoCollection<Document> collection, Class<?> mongoDatabaseImpl) {
//update Data
Document doc = new Document("name", "Roy").append("company", "Naver");
Document doc2 = new Document("$set", new Document("name", "Roy3"));
System.out.println("## Update Result=" + collection.updateOne(doc, doc2));

Method updateOne = getMethod(mongoDatabaseImpl, "updateOne", Bson.class, Bson.class);
NormalizedBson parsedBson = parseBson(doc, doc2);

verifier.verifyTrace(event(MONGO_EXECUTE_QUERY, updateOne, null, address, null
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_INFO.getName(), "customers")
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_OPTION.getName(), "MAJORITY")
, new ExpectedAnnotation(MongoConstants.MONGO_JSON_DATA.getName(), new StringStringValue(parsedBson.getNormalizedBson(), parsedBson.getParameter()))));
}


public void readData(PluginTestVerifier verifier, String address, MongoCollection<Document> collection, Class<?> mongoDatabaseImpl) {
//read data
MongoCursor<Document> cursor = collection.find().iterator();

Method find = getMethod(mongoDatabaseImpl, "find");

verifier.verifyTrace(event(MONGO_EXECUTE_QUERY, find, null, address, null
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_INFO.getName(), "customers")
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_OPTION.getName(), "secondaryPreferred")));

assertResultSize("Unexpected read data",2, cursor);
}

private void assertResultSize(String message, int expected, MongoCursor<Document> cursor) {
int resultCount = 0;
try {
while (cursor.hasNext()) {
resultCount++;
Document document = cursor.next();
System.out.println("## Result=" + document);
}
} finally {
cursor.close();
}
Assert.assertEquals(message, expected, resultCount);
}

public void deleteData(PluginTestVerifier verifier, String address, MongoCollection<Document> collection, Class<?> mongoDatabaseImpl) {
//delete data
Document doc = new Document("name", "Roy3");
DeleteResult deleteResult = collection.deleteMany(doc);

Method deleteMany = getMethod(mongoDatabaseImpl, "deleteMany", Bson.class);
NormalizedBson parsedBson = parseBson(doc);

verifier.verifyTrace(event(MONGO_EXECUTE_QUERY, deleteMany, null, address, null
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_INFO.getName(), "customers")
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_OPTION.getName(), "MAJORITY")
, new ExpectedAnnotation(MongoConstants.MONGO_JSON_DATA.getName(), new StringStringValue(parsedBson.getNormalizedBson(), parsedBson.getParameter()))));

Assert.assertEquals("unexcepted delete count",1, deleteResult.getDeletedCount());
}

public void filterData(PluginTestVerifier verifier, String address, MongoCollection<Document> collection, Class<?> mongoDatabaseImpl) {
Method find = getMethod(mongoDatabaseImpl, "find", Bson.class);
Bson bson = eq("name", "Roy3");
NormalizedBson parsedBson = parseBson(bson);

MongoCursor<Document> cursor = collection.find(bson).iterator();

verifier.verifyTrace(event(MONGO_EXECUTE_QUERY, find, null, address, null
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_INFO.getName(), "customers")
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_OPTION.getName(), "secondaryPreferred")
, new ExpectedAnnotation(MongoConstants.MONGO_JSON_DATA.getName(), new StringStringValue(parsedBson.getNormalizedBson(), parsedBson.getParameter()))));

assertResultSize("Unexpected filter data",1, cursor);
}

public void filterData2(PluginTestVerifier verifier, String address, MongoCollection<Document> collection, Class<?> mongoDatabaseImpl) {
Method find = getMethod(mongoDatabaseImpl, "find", Bson.class);
Bson bson = and(exists("name"), nin("name", 5, 15));
NormalizedBson parsedBson = parseBson(bson);

MongoCursor<Document> cursor = collection.find(bson).iterator();

verifier.verifyTrace(event(MONGO_EXECUTE_QUERY, find, null, address, null
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_INFO.getName(), "customers")
, new ExpectedAnnotation(MongoConstants.MONGO_COLLECTION_OPTION.getName(), "secondaryPreferred")
, new ExpectedAnnotation(MongoConstants.MONGO_JSON_DATA.getName(), new StringStringValue(parsedBson.getNormalizedBson(), parsedBson.getParameter()))));

assertResultSize("Unexpected filter data2",1, cursor);
}
}
Loading

0 comments on commit 3a14ae8

Please sign in to comment.