kopia/internal/hashcache/hashcache_writer.go

package hashcache

import (
	"fmt"
	"io"

	"github.com/kopia/kopia/internal/jsonstream"
)

// Writer emits hash cache entries.
type Writer interface {
	WriteEntry(e Entry) error
	Finalize() error
}

type writer struct {
	writer          *jsonstream.Writer
	lastNameWritten string
}

// NewWriter creates new hash cache Writer.
func NewWriter(w io.Writer) Writer {
	hcw := &writer{
		writer: jsonstream.NewWriter(w, hashCacheStreamType),
	}
	return hcw
}

// WriteEntry emits the specified hash cache entry.
func (hcw *writer) WriteEntry(e Entry) error {
	if err := e.ObjectID.Validate(); err != nil {
		panic("invalid object ID: " + err.Error())
	}

	if hcw.lastNameWritten != "" {
		if isLessOrEqual(e.Name, hcw.lastNameWritten) {
			return fmt.Errorf("out-of-order directory entry, previous '%v' current '%v'", hcw.lastNameWritten, e.Name)
		}
		hcw.lastNameWritten = e.Name
	}

	return hcw.writer.Write(&e)
}

// Finalize closes hashcache stream and must be invoked after emitting all entries.
func (hcw *writer) Finalize() error {
	return hcw.writer.Finalize()
}