Add tls options #3

Merged
luuvansinh merged 2 commits from add-tls-options into master 2022-02-24 08:06:55 +00:00
1 changed files with 32 additions and 1 deletions
Showing only changes of commit 2b6be44627 - Show all commits

View File

@ -7,10 +7,41 @@ import (
"github.com/logrusorgru/aurora"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"go.mongodb.org/mongo-driver/mongo/readpref"
)
// ConnectTLSOpts ...
type ConnectTLSOpts struct {
Host string
DBName string
ReplSet string
CaFilePath string
CertificateKeyFilePath string
CertificateKeyFilePassword string
}
var db *mongo.Database
// ConnectWithTLS ...
func ConnectWithTLS(opts ConnectTLSOpts) (*mongo.Database, error) {
ctx := context.Background()
uri := fmt.Sprintf("%s/?tls=true&tlsCAFile=%s&tlsCertificateKeyFile=%s&tlsCertificateKeyFilePassword=%s", opts.Host, opts.CaFilePath, opts.CertificateKeyFilePath, opts.CertificateKeyFilePassword)
readPref := readpref.SecondaryPreferred()
credential := options.Credential{
AuthMechanism: "MONGODB-X509",
}
clientOpts := options.Client().SetAuth(credential).SetReadPreference(readPref).SetReplicaSet(opts.ReplSet).ApplyURI(uri)
client, err := mongo.Connect(ctx, clientOpts)
if err != nil {
return nil, err
}
if err := client.Ping(ctx, readpref.SecondaryPreferred()); err != nil {
return nil, err
}
db = client.Database(opts.DBName)
return db, err
}
// Connect to mongo server
func Connect(host, user, password, dbName, mechanism, source string) (*mongo.Database, error) {
connectOptions := options.ClientOptions{}
@ -31,7 +62,7 @@ func Connect(host, user, password, dbName, mechanism, source string) (*mongo.Dat
return nil, err
}
fmt.Println(aurora.Green("*** CONNECTED TO MONGODB: " + host))
fmt.Println(aurora.Green("*** CONNECTED TO MONGODB: " + host + " --- DB: " + dbName))
// Set data
db = client.Database(dbName)