Как печатать байты во время загрузки файла? -golang
Мне интересно, можно ли подсчитать и распечатать количество байтов, загруженных во время загрузки файла.
out, err := os.Create("file.txt")
defer out.Close()
if err != nil {
fmt.Println(fmt.Sprint(err) )
panic(err)
}
resp, err := http.Get("http://example.com/zip")
defer resp.Body.Close()
if err != nil {
fmt.Println(fmt.Sprint(err) )
panic(err)
}
n, er := io.Copy(out, resp.Body)
if er != nil {
fmt.Println(fmt.Sprint(err) )
}
fmt.Println(n, "bytes ")
по некоторым причинам, поэтому не позволяет мне представить вопрос, потому что "похоже, что ваш пост в основном код; Пожалуйста, добавьте некоторые подробности", поэтому я просто пишу это
3 ответов
если я правильно вас понимаю, вы хотите отобразить количество прочитанных байтов во время передачи данных. Предположительно, чтобы поддерживать какой-то прогресс-бар или что-то в этом роде.
В этом случае вы можете использовать структуры композиционных данных Go, чтобы обернуть средство чтения или записи в пользовательский io.Reader
или io.Writer
реализация.
он просто пересылает соответствующие Read
или Write
вызов базового потока, выполняя некоторую дополнительную работу с (int, error)
значения, возвращаемые их. Вот пример, который вы можете запустить на Go playground.
package main
import (
"bytes"
"fmt"
"io"
"os"
"strings"
)
// PassThru wraps an existing io.Reader.
//
// It simply forwards the Read() call, while displaying
// the results from individual calls to it.
type PassThru struct {
io.Reader
total int64 // Total # of bytes transferred
}
// Read 'overrides' the underlying io.Reader's Read method.
// This is the one that will be called by io.Copy(). We simply
// use it to keep track of byte counts and then forward the call.
func (pt *PassThru) Read(p []byte) (int, error) {
n, err := pt.Reader.Read(p)
pt.total += int64(n)
if err == nil {
fmt.Println("Read", n, "bytes for a total of", pt.total)
}
return n, err
}
func main() {
var src io.Reader // Source file/url/etc
var dst bytes.Buffer // Destination file/buffer/etc
// Create some random input data.
src = bytes.NewBufferString(strings.Repeat("Some random input data", 1000))
// Wrap it with our custom io.Reader.
src = &PassThru{Reader: src}
count, err := io.Copy(&dst, src)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
fmt.Println("Transferred", count, "bytes")
}
вывод, который он генерирует, таков:
Read 512 bytes for a total of 512
Read 1024 bytes for a total of 1536
Read 2048 bytes for a total of 3584
Read 4096 bytes for a total of 7680
Read 8192 bytes for a total of 15872
Read 6128 bytes for a total of 22000
Transferred 22000 bytes
в stdlib сейчас дает что-то вроде
// WriteCounter counts the number of bytes written to it.
type WriteCounter struct {
Total int64 // Total # of bytes transferred
}
// Write implements the io.Writer interface.
//
// Always completes and never returns an error.
func (wc *WriteCounter) Write(p []byte) (int, error) {
n := len(p)
wc.Total += int64(n)
fmt.Printf("Read %d bytes for a total of %d\n", n, wc.Total)
return n, nil
}
func main() {
// ...
// Wrap it with our custom io.Reader.
src = io.TeeReader(src, &WriteCounter{})
// ...
}
на схватила GO пакет реализует обновления прогресса (и многие другие функции) для загрузки файлов.
пример обновления хода печати во время загрузки включен в следующее пошаговое руководство:http://cavaliercoder.com/blog/downloading-large-files-in-go.html
вообще можно назвать grab.GetAsync
который загружает в новую процедуру Go, а затем контролирует BytesTransferred
или Progress
вернулся grab.Response
от вызывающий поток.