106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
// Licensed to Elasticsearch B.V. under one or more contributor
|
|
// license agreements. See the NOTICE file distributed with
|
|
// this work for additional information regarding copyright
|
|
// ownership. Elasticsearch B.V. licenses this file to you 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 mongodb
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/elastic/beats/libbeat/logp"
|
|
"github.com/elastic/beats/metricbeat/mb"
|
|
"github.com/elastic/beats/metricbeat/mb/parse"
|
|
|
|
mgo "gopkg.in/mgo.v2"
|
|
)
|
|
|
|
func init() {
|
|
// Register the ModuleFactory function for the "mongodb" module.
|
|
if err := mb.Registry.AddModule("mongodb", NewModule); err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
// NewModule creates a new mb.Module instance and validates that at least one host has been
|
|
// specified
|
|
func NewModule(base mb.BaseModule) (mb.Module, error) {
|
|
// Validate that at least one host has been specified.
|
|
config := struct {
|
|
Hosts []string `config:"hosts" validate:"nonzero,required"`
|
|
}{}
|
|
if err := base.UnpackConfig(&config); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &base, nil
|
|
}
|
|
|
|
// ParseURL parses valid MongoDB URL strings into an mb.HostData instance
|
|
func ParseURL(module mb.Module, host string) (mb.HostData, error) {
|
|
c := struct {
|
|
Username string `config:"username"`
|
|
Password string `config:"password"`
|
|
}{}
|
|
if err := module.UnpackConfig(&c); err != nil {
|
|
return mb.HostData{}, err
|
|
}
|
|
|
|
if parts := strings.SplitN(host, "://", 2); len(parts) != 2 {
|
|
// Add scheme.
|
|
host = fmt.Sprintf("mongodb://%s", host)
|
|
}
|
|
|
|
// This doesn't use URLHostParserBuilder because MongoDB URLs can contain
|
|
// multiple hosts separated by commas (mongodb://host1,host2,host3?options).
|
|
u, err := url.Parse(host)
|
|
if err != nil {
|
|
return mb.HostData{}, fmt.Errorf("error parsing URL: %v", err)
|
|
}
|
|
|
|
parse.SetURLUser(u, c.Username, c.Password)
|
|
|
|
// https://docs.mongodb.com/manual/reference/connection-string/
|
|
_, err = mgo.ParseURL(u.String())
|
|
if err != nil {
|
|
return mb.HostData{}, err
|
|
}
|
|
|
|
return parse.NewHostDataFromURL(u), nil
|
|
}
|
|
|
|
// NewDirectSession estbalishes direct connections with a list of hosts. It uses the supplied
|
|
// dialInfo parameter as a template for establishing more direct connections
|
|
func NewDirectSession(dialInfo *mgo.DialInfo) (*mgo.Session, error) {
|
|
// make a copy
|
|
nodeDialInfo := *dialInfo
|
|
nodeDialInfo.Direct = true
|
|
nodeDialInfo.FailFast = true
|
|
|
|
logp.Debug("mongodb", "Connecting to MongoDB node at %v", nodeDialInfo.Addrs)
|
|
|
|
session, err := mgo.DialWithInfo(&nodeDialInfo)
|
|
if err != nil {
|
|
logp.Err("Error establishing direct connection to mongo node at %v. Error output: %s", nodeDialInfo.Addrs, err.Error())
|
|
return nil, err
|
|
}
|
|
|
|
// Relax consistency mode so reading from a secondary is allowed
|
|
session.SetMode(mgo.Monotonic, true)
|
|
|
|
return session, nil
|
|
}
|