dsn_helpers.go

// pmm-managed
// Copyright (C) 2017 Percona LLC
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.

package models

import (
	"time"

	"google.golang.org/grpc/codes"
	"google.golang.org/grpc/status"
	"gopkg.in/reform.v1"
)

// FindDSNByServiceIDandPMMAgentID resolves DSN and Files by service id.
func FindDSNByServiceIDandPMMAgentID(q *reform.Querier, serviceID, pmmAgentID, db string) (string, *Agent, error) {
	// FIXME This function is problematic:
	//
	// * it will return error in case we run multiple exporters for the same service with different credentials;
	//
	// * MySQLdExporter's DSN does not use ParseTime that could be helpful for query actions,
	//   but we can't change if for mysqld_exporter for compatibility reasons.
	//
	// rewrite logic to use agent_id instead of service_id?

	svc, err := FindServiceByID(q, serviceID)
	if err != nil {
		return "", nil, err
	}

	var agentTypes []AgentType
	switch svc.ServiceType {
	case MySQLServiceType:
		agentTypes = append(
			agentTypes,
			QANMySQLSlowlogAgentType,
			QANMySQLPerfSchemaAgentType,
			MySQLdExporterType)
	case PostgreSQLServiceType:
		agentTypes = append(
			agentTypes,
			QANPostgreSQLPgStatementsAgentType,
			QANPostgreSQLPgStatMonitorAgentType,
			PostgresExporterType)
	case MongoDBServiceType:
		agentTypes = append(
			agentTypes,
			QANMongoDBProfilerAgentType,
			MongoDBExporterType)
	default:
		return "", nil, status.Errorf(codes.FailedPrecondition, "Couldn't resolve dsn, as service is unsupported")
	}

	for _, agentType := range agentTypes {
		fexp, err := FindAgents(q, AgentFilters{
			ServiceID:  serviceID,
			AgentType:  &agentType, //nolint:gosec
			PMMAgentID: pmmAgentID,
		})
		if err != nil {
			return "", nil, err
		}
		if len(fexp) == 1 {
			return fexp[0].DSN(svc, time.Second, db, nil), fexp[0], nil
		}
		if len(fexp) > 1 {
			return "", nil, status.Errorf(codes.FailedPrecondition, "Couldn't resolve dsn, as there should be only one agent")
		}
	}

	return "", nil, status.Errorf(codes.FailedPrecondition, "Couldn't resolve dsn, as there should be one agent")
}