Rework write deadline logic

go-sql-driver · May 9, 2017 · d5f60b5 · d5f60b5
1 parent 221eb6e
commit d5f60b5
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/packets.go b/packets.go
@@ -116,15 +116,13 @@ func (mc *mysqlConn) writePacket(ctx mysqlContext, data []byte) error {
 		data[3] = mc.sequence
 
 		// Write packet
-		var timeNow = time.Now()
-		var deadline = timeNow
-		if mc.writeTimeout > 0 {
-			deadline = timeNow.Add(mc.writeTimeout)
-			if isCtxDeadlineSet && deadline.After(ctxDeadline) {
-				deadline = ctxDeadline
-			}
+		var deadline time.Time
+		if ctxDeadline, ok := ctx.Deadline(); ok {
+			deadline = ctxDeadline
+		} else if mc.writeTimeout > 0 {
+			deadline = time.Now().Add(mc.writeTimeout)
 		}
-		if deadline.After(timeNow) {
+		if !deadline.IsZero() {
 			if err := mc.netConn.SetWriteDeadline(deadline); err != nil {
 				return err
 			}