Это скорее общий вопрос Go, связанный с интерфейсами / функциями / замыканиями / типами / ..., но с конкретным вариантом использования выполнения ввода-вывода.

Учитывая переменную io.Reader, я хочу создать io.ReadCloser с пользовательской реализацией Close().

var file os.File
file = ... // some File which needs to be closed

var reader io.Reader
reader = ... // get Reader from somewhere

var readCloser io.ReadCloser
readCloser = ... // how to create a ReadCloser here?

На Java я бы написал что-то вроде:

readCloser = new io.ReadCloser() {
    Read(p []byte) {
        reader.Read(p);
    }

    Close() {
        // my custom close implementation
        file.Close()
    }    
}

// now pass readCloser around as needed

Как это сделать с Go?

go
0
mstrap 15 Сен 2018 в 14:33

2 ответа

Лучший ответ

Взяв подсказку от http.HandlerFunc, я мог придумать следующее

package main

import (
    "bytes"
    "fmt"
    "io"
    "strings"
)

// ReadData reads data from an io.Reader
func ReadData(r io.Reader) {
    buf := new(bytes.Buffer)
    buf.ReadFrom(r)
    s := buf.String()
    fmt.Printf("read : %s \n", s)
}

// ReaderFunc takes a function and returns io.Reader
type ReaderFunc func([]byte) (int, error)

func (r ReaderFunc) Read(b []byte) (int, error) {
    return r(b)
}

func main() {
    // invoking ReadData with a regular io.Reader
    r := strings.NewReader("data 123")
    ReadData(r)

    // invoking ReadData with an anonymous io.Reader
    r = strings.NewReader("data 456")
    ReadData(ReaderFunc(func(b []byte) (int, error) {
        return r.Read(b)
    }))
}
2
weima 20 Сен 2018 в 03:00

Один из способов - создать собственный тип, сочетающий в себе io.Reader и a io.Closer и реализует интерфейс io.ReadCloser. Что-то вроде этого:

type ReadCloseCombiner struct {
    io.Reader
    io.Closer
}

func (rcc ReadCloseCombiner) Read(p []byte) (n int, err error) {
    return rcc.Reader.Read(p)
}

func (rcc ReadCloseCombiner) Close() error {
    return rcc.Closer.Close()
}

И используйте это так:

var file *os.File
file = ... // some File which needs to be closed

var reader io.Reader
reader = ... // get Reader from somewhere

var readCloser io.ReadCloser
readCloser = ReadCloseCombiner{reader, file}

// now pass readCloser around as needed
readCloser.Read(...)
readCloser.Close()

Если вам нужно что-то более гибкое, я бы подумал о том, чтобы иметь тип, который вместо этого принимает функцию чтения и закрытия, а затем вы можете передать ему анонимные функции (или даже reader.Read / file.Close в вашем случае).

2
Mattias Wadman 15 Сен 2018 в 14:47