@@ -4,12 +4,10 @@ import (
4
4
"context"
5
5
"fmt"
6
6
stdlog "log"
7
- "math/rand"
8
7
"os"
9
8
"time"
10
9
11
10
"github.com/alecthomas/kingpin"
12
- "go.mongodb.org/mongo-driver/bson/primitive"
13
11
"go.mongodb.org/mongo-driver/mongo"
14
12
15
13
"github.com/percona/percona-backup-mongodb/pbm/compress"
@@ -63,8 +61,6 @@ func main() {
63
61
* sampleSizeF = 1
64
62
}
65
63
66
- rand .Seed (time .Now ().UnixNano ())
67
-
68
64
switch cmd {
69
65
case compressionCmd .FullCommand ():
70
66
fmt .Print ("Test started " )
@@ -90,7 +86,7 @@ func testCompression(mURL string, compression compress.CompressionType, level *i
90
86
var cn * mongo.Client
91
87
92
88
if collection != "" {
93
- cn , err := connect .MongoConnect (ctx , mURL , & connect.MongoConnectOptions { Direct : true } )
89
+ cn , err := connect .MongoConnect (ctx , mURL , connect .Direct ( true ) )
94
90
if err != nil {
95
91
stdlog .Fatalln ("Error: connect to mongodb-node:" , err )
96
92
}
@@ -112,23 +108,19 @@ func testCompression(mURL string, compression compress.CompressionType, level *i
112
108
}
113
109
114
110
func testStorage (mURL string , compression compress.CompressionType , level * int , sizeGb float64 , collection string ) {
115
- ctx := context .Background ()
116
-
117
- sess , err := connect .MongoConnect (ctx , mURL , & connect.MongoConnectOptions {Direct : true })
111
+ sess , err := connect .MongoConnect (context .Background (), mURL , connect .Direct (true ))
118
112
if err != nil {
119
113
stdlog .Fatalln ("Error: connect to mongodb-node:" , err )
120
114
}
121
- defer sess .Disconnect (ctx ) //nolint:errcheck
115
+ defer sess .Disconnect (context . Background () ) //nolint:errcheck
122
116
123
- client , err := connect .Connect (ctx , mURL , & connect. ConnectOptions { AppName : "pbm-speed-test" } )
117
+ client , err := connect .Connect (context . Background () , mURL , "pbm-speed-test" )
124
118
if err != nil {
125
119
stdlog .Fatalln ("Error: connect to mongodb-pbm:" , err )
126
120
}
127
- defer client .Disconnect (ctx ) //nolint:errcheck
121
+ defer client .Disconnect (context . Background () ) //nolint:errcheck
128
122
129
- l := log .FromContext (ctx ).
130
- NewEvent ("" , "" , "" , primitive.Timestamp {})
131
- stg , err := util .GetStorage (ctx , client , l )
123
+ stg , err := util .GetStorage (context .Background (), client , log .DiscardEvent )
132
124
if err != nil {
133
125
stdlog .Fatalln ("Error: get storage:" , err )
134
126
}
0 commit comments