Exemplo n.º 1
0
// buildFile saves a data file from the tar archive to disk,
// then returns a struct with data we'll need to construct the
// GenericFile object in Fedora later.
func (reader *Reader) saveWithChecksums(gf *models.GenericFile) {
	// Set up a MultiWriter to stream data ONCE to file,
	// md5 and sha256. We don't want to process the stream
	// three separate times.
	err := os.MkdirAll(filepath.Dir(gf.IngestLocalPath), 0755)
	if err != nil {
		gf.IngestErrorMessage = err.Error()
		return
	}
	outputWriter, err := os.OpenFile(gf.IngestLocalPath, os.O_CREATE|os.O_WRONLY, 0644)
	if outputWriter != nil {
		defer outputWriter.Close()
	}
	if err != nil {
		gf.IngestErrorMessage = fmt.Sprintf("Error opening %s for writing: %v", gf.IngestLocalPath, err)
		return
	}
	md5Hash := md5.New()
	shaHash := sha256.New()
	multiWriter := io.MultiWriter(md5Hash, shaHash, outputWriter)
	io.Copy(multiWriter, reader.tarReader)
	gf.IngestMd5 = fmt.Sprintf("%x", md5Hash.Sum(nil))
	gf.IngestSha256 = fmt.Sprintf("%x", shaHash.Sum(nil))
	gf.IngestSha256GeneratedAt = time.Now().UTC()
	gf.FileFormat, _ = platform.GuessMimeType(gf.IngestLocalPath) // on err, defaults to application/binary
	return
}
Exemplo n.º 2
0
// For partner build, we just want to make sure this doesn't throw
// an execption. For our own server build, we also want to make sure
// it gets the mime type right. GuessMimeType is defined in mime.go,
// with a dummy version in partner builds in nomime.go.
func TestGuessMimeType(t *testing.T) {
	pathToTempFile := setupMimeTest(t)
	defer teardownMimeTest(pathToTempFile)
	mimetype, err := platform.GuessMimeType(pathToTempFile)
	if err != nil {
		t.Error(err)
	}
	if !platform.IsPartnerBuild {
		assert.Equal(t, "text/plain", mimetype)
	}
}