PMM-3879 MongoDB Profiler Agent.

models/agent_model.go

@@ -37,6 +37,7 @@ const (
 	MongoDBExporterType         AgentType = "mongodb_exporter"
 	QANMySQLPerfSchemaAgentType AgentType = "qan-mysql-perfschema-agent"
 	PostgresExporterType        AgentType = "postgres_exporter"
+	QANMongoDBProfilerAgentType AgentType = "qan-mongodb-profiler-agent"
 )
 
 // Agent represents Agent as stored in database.

services/agents/mongodb.go

@@ -50,6 +50,28 @@ func mongodbExporterConfig(service *models.Service, exporter *models.Agent) *age
 
 	sort.Strings(args)
 
+	connString := mongoDSN(service, exporter)
+
+	return &agentpb.SetStateRequest_AgentProcess{
+		Type:               agentpb.Type_MONGODB_EXPORTER,
+		TemplateLeftDelim:  tdp.left,
+		TemplateRightDelim: tdp.right,
+		Args:               args,
+		Env: []string{
+			fmt.Sprintf("MONGODB_URI=%s", connString),
+		},
+	}
+}
+
+// qanMongoDBProfilerAgentConfig returns desired configuration of qan-mongodb-profiler-agent built-in agent.
+func qanMongoDBProfilerAgentConfig(service *models.Service, exporter *models.Agent) *agentpb.SetStateRequest_BuiltinAgent {
+	return &agentpb.SetStateRequest_BuiltinAgent{
+		Type: agentpb.Type_QAN_MONGODB_PROFILER_AGENT,
+		Dsn:  mongoDSN(service, exporter),
+	}
+}
+
+func mongoDSN(service *models.Service, exporter *models.Agent) string {
 	host := pointer.GetString(service.Address)
 	port := pointer.GetUint16(service.Port)
 
@@ -64,13 +86,5 @@ func mongodbExporterConfig(service *models.Service, exporter *models.Agent) *age
 		connURL.User = url.UserPassword(username, password)
 	}
 
-	return &agentpb.SetStateRequest_AgentProcess{
-		Type:               agentpb.Type_MONGODB_EXPORTER,
-		TemplateLeftDelim:  tdp.left,
-		TemplateRightDelim: tdp.right,
-		Args:               args,
-		Env: []string{
-			fmt.Sprintf("MONGODB_URI=%s", connURL.String()),
-		},
-	}
+	return connURL.String()
 }

services/agents/mysql.go

@@ -98,7 +98,7 @@ func mysqldExporterConfig(service *models.Service, exporter *models.Agent) *agen
 	}
 }
 
-// qanMySQLPerfSchemaAgentConfig returns desired configuration of qan-mysql-perfschema internal agent.
+// qanMySQLPerfSchemaAgentConfig returns desired configuration of qan-mysql-perfschema built-in agent.
 func qanMySQLPerfSchemaAgentConfig(service *models.Service, exporter *models.Agent) *agentpb.SetStateRequest_BuiltinAgent {
 	return &agentpb.SetStateRequest_BuiltinAgent{
 		Type: agentpb.Type_QAN_MYSQL_PERFSCHEMA_AGENT,

services/agents/registry.go

@@ -420,6 +420,18 @@ func (r *Registry) SendSetStateRequest(ctx context.Context, pmmAgentID string) {
 			}
 			agentProcesses[row.AgentID] = postgresExporterConfig(services[0], row)
 
+		case models.QANMongoDBProfilerAgentType:
+			services, err := models.ServicesForAgent(r.db.Querier, row.AgentID)
+			if err != nil {
+				l.Error(err)
+				return
+			}
+			if len(services) != 1 {
+				l.Errorf("Expected exactly one Services, got %d.", len(services))
+				return
+			}
+			builtinAgents[row.AgentID] = qanMongoDBProfilerAgentConfig(services[0], row)
+
 		default:
 			l.Panicf("unhandled Agent type %s", row.AgentType)
 		}