1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
/**
* Verifies that the $collStats aggregation stage includes the shard and hostname for each output
* document when run via mongoS, and that the former is absent when run on a non-shard mongoD.
* @tags: [
* requires_sharding,
* sbe_incompatible,
* ]
*/
(function() {
"use strict";
const conn = MongoRunner.runMongod();
let testDB = conn.getDB(jsTestName());
let testColl = testDB.test;
// getHostName() doesn't include port, db.getMongo().host is 127.0.0.1:<port>
const hostName = (getHostName() + ":" + testDB.getMongo().host.split(":")[1]);
// Test that the shard field is absent and the host field is present when run on mongoD.
assert.eq(testColl
.aggregate([
{$collStats: {latencyStats: {histograms: true}}},
{$group: {_id: {shard: "$shard", host: "$host"}}}
])
.toArray(),
[{_id: {host: hostName}}]);
MongoRunner.stopMongod(conn);
// Test that both shard and hostname are present for $collStats results on a sharded cluster.
const st = new ShardingTest({name: jsTestName(), shards: 2});
testDB = st.s.getDB(jsTestName());
testColl = testDB.test;
assert.commandWorked(testDB.dropDatabase());
// Enable sharding on the test database.
assert.commandWorked(testDB.adminCommand({enableSharding: testDB.getName()}));
// Shard 'testColl' on {_id: 'hashed'}. This will automatically presplit the collection and
// place chunks on each shard.
assert.commandWorked(
testDB.adminCommand({shardCollection: testColl.getFullName(), key: {_id: "hashed"}}));
// Group $collStats result by $shard and $host to confirm that both fields are present.
assert.eq(testColl
.aggregate([
{$collStats: {latencyStats: {histograms: true}}},
{$group: {_id: {shard: "$shard", host: "$host"}}},
{$sort: {_id: 1}}
])
.toArray(),
[
{_id: {shard: st.shard0.shardName, host: st.rs0.getPrimary().host}},
{_id: {shard: st.shard1.shardName, host: st.rs1.getPrimary().host}},
]);
st.stop();
})();
|
