examples_test.go

Documentation: google.golang.org/api/transport/bytestream

     1  // Copyright 2016 Google LLC.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  package bytestream
     6  
     7  import (
     8  	"bytes"
     9  	"context"
    10  	"fmt"
    11  	"io"
    12  	"log"
    13  
    14  	"google.golang.org/grpc"
    15  )
    16  
    17  func ExampleNewClient(serverPort int, resourceName string) {
    18  	ctx := context.Background()
    19  	conn, err := grpc.Dial(fmt.Sprintf("localhost:%d", serverPort), grpc.WithInsecure())
    20  	if err != nil {
    21  		log.Printf("grpc.Dial: %v", err)
    22  		return
    23  	}
    24  	client := NewClient(conn)
    25  	reader, err := client.NewReader(ctx, resourceName)
    26  	if err != nil {
    27  		log.Printf("NewReader(%q): %v", resourceName, err)
    28  	}
    29  	var buf bytes.Buffer
    30  	n, err := buf.ReadFrom(reader)
    31  	if err != nil && err != io.EOF {
    32  		log.Printf("Read %d bytes, got err=%v", n, err)
    33  	}
    34  	log.Printf("read %q", buf.String())
    35  }
    36  
    37  func ExampleNewReader(serverPort int, resourceName string) {
    38  	ctx := context.Background()
    39  	conn, err := grpc.Dial(fmt.Sprintf("localhost:%d", serverPort), grpc.WithInsecure())
    40  	if err != nil {
    41  		log.Printf("grpc.Dial: %v", err)
    42  		return
    43  	}
    44  	client := NewClient(conn)
    45  	reader, err := client.NewReader(ctx, resourceName)
    46  	if err != nil {
    47  		log.Printf("NewReader(%q): %v", resourceName, err)
    48  	}
    49  	var buf bytes.Buffer
    50  	n, err := buf.ReadFrom(reader)
    51  	if err != nil && err != io.EOF {
    52  		log.Printf("Read %d bytes, got err=%v", n, err)
    53  	}
    54  	log.Printf("read %q", buf.String())
    55  }
    56  
    57  func ExampleNewWriter(serverPort int, resourceName string) {
    58  	ctx := context.Background()
    59  	conn, err := grpc.Dial(fmt.Sprintf("localhost:%d", serverPort), grpc.WithInsecure())
    60  	if err != nil {
    61  		log.Printf("grpc.Dial: %v", err)
    62  		return
    63  	}
    64  	client := NewClient(conn)
    65  
    66  	w, err := client.NewWriter(ctx, resourceName)
    67  	if err != nil {
    68  		log.Printf("NewWriter: %v", err)
    69  		return
    70  	}
    71  	defer func() {
    72  		err := w.Close()
    73  		if err != nil {
    74  			log.Printf("Close: %v", err)
    75  		}
    76  	}()
    77  
    78  	buf := []byte("hello world")
    79  	n, err := w.Write(buf)
    80  	if err != nil {
    81  		log.Printf("Write: %v", err)
    82  	}
    83  	log.Printf("Wrote %d bytes", n)
    84  }
    85
View as plain text