mirror of
https://github.com/gomods/athens
synced 2026-02-03 11:00:32 +00:00
c6644f82f9
* Checking the number of bytes written to Mongo GridFS * adding missing import * formatting imports
51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package mongo
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
|
|
"github.com/gomods/athens/pkg/errors"
|
|
"github.com/gomods/athens/pkg/observ"
|
|
"github.com/gomods/athens/pkg/storage"
|
|
)
|
|
|
|
// Save stores a module in mongo storage.
|
|
func (s *ModuleStore) Save(ctx context.Context, module, version string, mod []byte, zip io.Reader, info []byte) error {
|
|
const op errors.Op = "mongo.Save"
|
|
ctx, span := observ.StartSpan(ctx, op.String())
|
|
defer span.End()
|
|
|
|
zipName := s.gridFileName(module, version)
|
|
fs := s.s.DB(s.d).GridFS("fs")
|
|
f, err := fs.Create(zipName)
|
|
if err != nil {
|
|
return errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
defer f.Close()
|
|
|
|
numBytesWritten, err := io.Copy(f, zip)
|
|
if err != nil {
|
|
return errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
if numBytesWritten <= 0 {
|
|
e := fmt.Errorf("copied %d bytes to Mongo GridFS", numBytesWritten)
|
|
return errors.E(op, e, errors.M(module), errors.V(version))
|
|
}
|
|
|
|
m := &storage.Module{
|
|
Module: module,
|
|
Version: version,
|
|
Mod: mod,
|
|
Info: info,
|
|
}
|
|
|
|
c := s.s.DB(s.d).C(s.c)
|
|
err = c.Insert(m)
|
|
if err != nil {
|
|
return errors.E(op, err, errors.M(module), errors.V(version))
|
|
}
|
|
|
|
return nil
|
|
}
|