Compare commits

...

2 commits

3 changed files with 101 additions and 26 deletions

View file

@ -15,6 +15,7 @@ var maxMemAllocPercent int = 80
var mongoInitdbRootUsername = "sasha"
var mongoInitdbRootPassword = "12345"
var verbosePtr = flag.Bool("v", false, "boolean")
var localIpPtr = flag.Bool("localip", false, "boolean")
func ExecSystem(command string) {

View file

@ -4,41 +4,93 @@ import (
"context"
"time"
"fmt"
"log"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
"go.mongodb.org/mongo-driver/bson"
)
func initReplMongo() {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
// Create a new replica set client
client, err := mongo.Connect(ctx, options.Client().ApplyURI("mongodb://sasha:12345@localhost:30000/?directConnection=true"))
if err != nil {
panic(err)
}
if *localIpPtr{
db := client.Database("admin")
members := make([]bson.M, numberOfContainers)
for i := 0; i < numberOfContainers; i++ {
members[i] = bson.M{
"_id": i,
"host": fmt.Sprintf("mongo%d", i),
ips, err := GetLocalIPs()
if err != nil {
log.Fatal(err)
}
}
config := bson.M{
"_id": replicasetName,
"members": members,
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
// Create a new replica set client
uri := fmt.Sprintf("mongodb://%s:%s@localhost:30000/?directConnection=true", mongoInitdbRootUsername, mongoInitdbRootPassword)
client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
if err != nil {
panic(err)
}
db := client.Database("admin")
members := make([]bson.M, numberOfContainers)
for i := 0; i < numberOfContainers; i++ {
members[i] = bson.M{
"_id": i,
"host": fmt.Sprintf("%s:%d", ips[0], 30000+i),
}
}
config := bson.M{
"_id": replicasetName,
"members": members,
}
if *verbosePtr{
fmt.Println(config)
}
command := bson.D{{"replSetInitiate", config}}
var result bson.M
err = db.RunCommand(context.TODO(), command).Decode(&result)
if err != nil {
panic(err)
}
client.Disconnect(ctx)
} else {
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
// Create a new replica set client
uri := fmt.Sprintf("mongodb://%s:%s@localhost:30000/?directConnection=true", mongoInitdbRootUsername, mongoInitdbRootPassword)
client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
if err != nil {
panic(err)
}
db := client.Database("admin")
members := make([]bson.M, numberOfContainers)
for i := 0; i < numberOfContainers; i++ {
members[i] = bson.M{
"_id": i,
"host": fmt.Sprintf("mongo%d", i),
}
}
config := bson.M{
"_id": replicasetName,
"members": members,
}
if *verbosePtr{
fmt.Println(config)
}
command := bson.D{{"replSetInitiate", config}}
var result bson.M
err = db.RunCommand(context.TODO(), command).Decode(&result)
if err != nil {
panic(err)
}
client.Disconnect(ctx)
}
command := bson.D{{"replSetInitiate", config}}
var result bson.M
err = db.RunCommand(context.TODO(), command).Decode(&result)
if err != nil {
panic(err)
}
client.Disconnect(ctx)
}

22
network.go Normal file
View file

@ -0,0 +1,22 @@
package main
import (
"net"
)
func GetLocalIPs() ([]net.IP, error) {
var ips []net.IP
addresses, err := net.InterfaceAddrs()
if err != nil {
return nil, err
}
for _, addr := range addresses {
if ipnet, ok := addr.(*net.IPNet); ok && !ipnet.IP.IsLoopback() {
if ipnet.IP.To4() != nil {
ips = append(ips, ipnet.IP)
}
}
}
return ips, nil
}