Как результат кодогенераций сохранить в архив и сразу его отдать?

Question

[Александр Мелентьев]

Всем привет.
Есть рестовое API, при обращении в определенный эндпойт планируется делать несколько кодогенераций, и результаты в виде уже готовых файлов класть в архив и отдавать запрашиваемому. Как результат кодогенераций сохранить в архив и сразу его отдать?

Answer 1

[WinPooh32]

https://golang.org/pkg/archive/zip/

Пример с одним файлом:

func handle(w http.ResponseWriter, r *http.Request){
	fileToZip, _ := os.Open("filepath")
	defer fileToZip.Close()

	// NewWriter returns a new Writer writing a zip file to w.
	zipWriter := zip.NewWriter(w)
	defer zipWriter.Close()

	info, _ := fileToZip.Stat()

	header, _ := zip.FileInfoHeader(info)

	// Change to deflate to gain better compression
	// see http://golang.org/pkg/archive/zip/#pkg-constants
	header.Method = zip.Deflate

	// CreateHeader adds a file to the zip archive using the provided FileHeader
	// for the file metadata. 
	// This returns a Writer to which the file contents should be written.
	writer, _ := zipWriter.CreateHeader(header)
	io.Copy(writer, fileToZip)
}

Comments

[Александр Мелентьев]

Вот код генерации:

func Module(name string)  {
	const (
		txt = `File {{.Name}}`
	)

	type Module struct {
		Module string
	}

	t := template.Must(template.New(name).Parse(txt))
	
	r := Module{
		Module: name,
	}
	f, err := os.Create("file.txt")
	if err != nil {
		log.Println(err)
	}
	defer f.Close()
	if err := t.Execute(f, r); err != nil {
		log.Println(err)
	}
}

В конце, сгенерированный код кладу в файл, и этот файл, а то и несколько нужно будет переносить в архив

[Александр Мелентьев]

Доработал вариант. Еще раз утверждаюсь в том, что всё можно делать из коробки. Спасибо!