transport.go 500 B

123456789101112131415161718192021222324252627282930313233
  1. package net
  2. import (
  3. "io"
  4. )
  5. const (
  6. bufferSize = 32 * 1024
  7. )
  8. func ReaderToChan(stream chan<- []byte, reader io.Reader) error {
  9. for {
  10. buffer := make([]byte, bufferSize)
  11. nBytes, err := reader.Read(buffer)
  12. if nBytes > 0 {
  13. stream <- buffer[:nBytes]
  14. }
  15. if err != nil {
  16. return err
  17. }
  18. }
  19. return nil
  20. }
  21. func ChanToWriter(writer io.Writer, stream <-chan []byte) error {
  22. for buffer := range stream {
  23. _, err := writer.Write(buffer)
  24. if err != nil {
  25. return err
  26. }
  27. }
  28. return nil
  29. }