context cancellation experiment

tomyl · Sep 20, 2022 · acb08bd · acb08bd
1 parent f85532f
commit acb08bd
Show file tree

Hide file tree

Showing 4 changed files with 50 additions and 25 deletions.
diff --git a/examples/drpc/client/main.go b/examples/drpc/client/main.go
@@ -6,6 +6,7 @@ package main
 import (
 	"context"
 	"fmt"
+	"log"
 	"net"
 	"time"
 
@@ -37,19 +38,31 @@ func Main(ctx context.Context) error {
 	// make a drpc proto-specific client
 	client := pb.NewDRPCCookieMonsterClient(conn)
 
-	// set a deadline for the operation
-	ctx, cancel := context.WithTimeout(ctx, time.Second)
-	defer cancel()
+	for _, d := range []time.Duration{1, 3} {
+		select {
+		case <-conn.Closed():
+			log.Print("closed")
+		default:
+			log.Print("not closed")
+		}
+		func() {
+			// set a deadline for the operation
+			ctx, cancel := context.WithTimeout(ctx, d*time.Second)
+			defer cancel()
 
-	// run the RPC
-	crumbs, err := client.EatCookie(ctx, &pb.Cookie{
-		Type: pb.Cookie_Oatmeal,
-	})
-	if err != nil {
-		return err
+			// run the RPC
+			crumbs, err := client.EatCookie(ctx, &pb.Cookie{
+				Type: pb.Cookie_Oatmeal,
+			})
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			// check the results
+			fmt.Println(crumbs.Cookie.Type.String())
+		}()
 	}
 
-	// check the results
-	_, err = fmt.Println(crumbs.Cookie.Type.String())
-	return err
+	return nil
 }
diff --git a/examples/drpc/server/main.go b/examples/drpc/server/main.go
@@ -6,6 +6,7 @@ package main
 import (
 	"context"
 	"net"
+	"time"
 
 	"storj.io/drpc/drpcmux"
 	"storj.io/drpc/drpcserver"
@@ -20,6 +21,7 @@ type CookieMonsterServer struct {
 
 // EatCookie turns a cookie into crumbs.
 func (s *CookieMonsterServer) EatCookie(ctx context.Context, cookie *pb.Cookie) (*pb.Crumbs, error) {
+	time.Sleep(2 * time.Second)
 	return &pb.Crumbs{
 		Cookie: cookie,
 	}, nil

diff --git a/examples/grpc/client/main.go b/examples/grpc/client/main.go
@@ -6,6 +6,7 @@ package main
 import (
 	"context"
 	"fmt"
+	"log"
 	"time"
 
 	"google.golang.org/grpc"
@@ -31,19 +32,26 @@ func Main(ctx context.Context) error {
 	// make a grpc proto-specific client
 	client := pb.NewCookieMonsterClient(conn)
 
-	// set a deadline for the operation
-	ctx, cancel := context.WithTimeout(ctx, time.Second)
-	defer cancel()
-
-	// run the RPC
-	crumbs, err := client.EatCookie(ctx, &pb.Cookie{
-		Type: pb.Cookie_Oatmeal,
-	})
-	if err != nil {
-		return err
+	for _, d := range []time.Duration{1, 3} {
+		log.Println(conn.GetState())
+		func() {
+			// set a deadline for the operation
+			ctx, cancel := context.WithTimeout(ctx, d*time.Second)
+			defer cancel()
+
+			// run the RPC
+			crumbs, err := client.EatCookie(ctx, &pb.Cookie{
+				Type: pb.Cookie_Oatmeal,
+			})
+			if err != nil {
+				log.Println(err)
+				return
+			}
+
+			// check the results
+			_, err = fmt.Println(crumbs.Cookie.Type.String())
+		}()
 	}
 
-	// check the results
-	_, err = fmt.Println(crumbs.Cookie.Type.String())
-	return err
+	return nil
 }
diff --git a/examples/grpc/server/main.go b/examples/grpc/server/main.go
@@ -6,6 +6,7 @@ package main
 import (
 	"context"
 	"net"
+	"time"
 
 	"google.golang.org/grpc"
 
@@ -19,6 +20,7 @@ type CookieMonsterServer struct {
 
 // EatCookie turns a cookie into crumbs.
 func (s *CookieMonsterServer) EatCookie(ctx context.Context, cookie *pb.Cookie) (*pb.Crumbs, error) {
+	time.Sleep(2 * time.Second)
 	return &pb.Crumbs{
 		Cookie: cookie,
 	}, nil