@@ -4,12 +4,10 @@ import (
44 "context"
55 "fmt"
66 stdlog "log"
7- "math/rand"
87 "os"
98 "time"
109
1110 "github.com/alecthomas/kingpin"
12- "go.mongodb.org/mongo-driver/bson/primitive"
1311 "go.mongodb.org/mongo-driver/mongo"
1412
1513 "github.com/percona/percona-backup-mongodb/pbm/compress"
@@ -63,8 +61,6 @@ func main() {
6361 * sampleSizeF = 1
6462 }
6563
66- rand .Seed (time .Now ().UnixNano ())
67-
6864 switch cmd {
6965 case compressionCmd .FullCommand ():
7066 fmt .Print ("Test started " )
@@ -90,7 +86,7 @@ func testCompression(mURL string, compression compress.CompressionType, level *i
9086 var cn * mongo.Client
9187
9288 if collection != "" {
93- cn , err := connect .MongoConnect (ctx , mURL , & connect.MongoConnectOptions { Direct : true } )
89+ cn , err := connect .MongoConnect (ctx , mURL , connect .Direct ( true ) )
9490 if err != nil {
9591 stdlog .Fatalln ("Error: connect to mongodb-node:" , err )
9692 }
@@ -112,23 +108,19 @@ func testCompression(mURL string, compression compress.CompressionType, level *i
112108}
113109
114110func 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 ))
118112 if err != nil {
119113 stdlog .Fatalln ("Error: connect to mongodb-node:" , err )
120114 }
121- defer sess .Disconnect (ctx ) //nolint:errcheck
115+ defer sess .Disconnect (context . Background () ) //nolint:errcheck
122116
123- client , err := connect .Connect (ctx , mURL , & connect. ConnectOptions { AppName : "pbm-speed-test" } )
117+ client , err := connect .Connect (context . Background () , mURL , "pbm-speed-test" )
124118 if err != nil {
125119 stdlog .Fatalln ("Error: connect to mongodb-pbm:" , err )
126120 }
127- defer client .Disconnect (ctx ) //nolint:errcheck
121+ defer client .Disconnect (context . Background () ) //nolint:errcheck
128122
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 )
132124 if err != nil {
133125 stdlog .Fatalln ("Error: get storage:" , err )
134126 }
0 commit comments