github.com/lianghucheng/zrddz@v0.0.0-20200923083010-c71f680932e2/src/gopkg.in/mgo.v2/export_test.go (about) 1 package mgo 2 3 import ( 4 "time" 5 ) 6 7 func HackPingDelay(newDelay time.Duration) (restore func()) { 8 globalMutex.Lock() 9 defer globalMutex.Unlock() 10 11 oldDelay := pingDelay 12 restore = func() { 13 globalMutex.Lock() 14 pingDelay = oldDelay 15 globalMutex.Unlock() 16 } 17 pingDelay = newDelay 18 return 19 } 20 21 func HackSyncSocketTimeout(newTimeout time.Duration) (restore func()) { 22 globalMutex.Lock() 23 defer globalMutex.Unlock() 24 25 oldTimeout := syncSocketTimeout 26 restore = func() { 27 globalMutex.Lock() 28 syncSocketTimeout = oldTimeout 29 globalMutex.Unlock() 30 } 31 syncSocketTimeout = newTimeout 32 return 33 }