mirror of
https://github.com/dolthub/dolt.git
synced 2026-01-31 03:18:43 -06:00
61 lines
2.2 KiB
Go
61 lines
2.2 KiB
Go
package sync
|
|
|
|
import (
|
|
"io"
|
|
|
|
"github.com/attic-labs/noms/chunks"
|
|
"github.com/attic-labs/noms/d"
|
|
"github.com/attic-labs/noms/datas"
|
|
"github.com/attic-labs/noms/dataset"
|
|
"github.com/attic-labs/noms/ref"
|
|
"github.com/attic-labs/noms/types"
|
|
"github.com/attic-labs/noms/walk"
|
|
)
|
|
|
|
func validateRefAsCommit(r ref.Ref, cs chunks.ChunkSource) datas.Commit {
|
|
v := types.ReadValue(r, cs)
|
|
|
|
d.Exp.NotNil(v, "%v cannot be found", r)
|
|
|
|
// TODO: Replace this weird recover stuff below once we have a way to determine if a Value is an instance of a custom struct type. BUG #133
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
d.Exp.Fail("Not a Commit:", "%+v", v)
|
|
}
|
|
}()
|
|
return datas.CommitFromVal(v)
|
|
}
|
|
|
|
// DiffHeadsByRef takes two Refs, validates that both refer to Heads in the given ChunkSource, and then returns the set of Refs that can be reached from 'big', but not 'small'.
|
|
func DiffHeadsByRef(small, big ref.Ref, cs chunks.ChunkSource) []ref.Ref {
|
|
if small != (ref.Ref{}) {
|
|
validateRefAsCommit(small, cs)
|
|
}
|
|
validateRefAsCommit(big, cs)
|
|
return walk.Difference(small, big, cs)
|
|
}
|
|
|
|
// CopyChunks reads each Ref in refs out of src and writes it into sink.
|
|
func CopyChunks(refs []ref.Ref, src chunks.ChunkSource, sink chunks.ChunkSink) {
|
|
copy := func(ref ref.Ref) {
|
|
reader := src.Get(ref)
|
|
defer reader.Close()
|
|
d.Exp.NotNil(reader, "Attempt to copy ref which wasn't found: %+v", ref)
|
|
|
|
writer := sink.Put()
|
|
defer writer.Close()
|
|
_, err := io.Copy(writer, reader)
|
|
d.Exp.NoError(err)
|
|
}
|
|
for _, ref := range refs {
|
|
copy(ref)
|
|
}
|
|
return
|
|
}
|
|
|
|
// SetNewHead takes the Ref of the desired new Heads of ds, the chunk for which should already exist in the Dataset. It validates that the Ref points to an existing chunk that decodes to the correct type of value and then commits it to ds, returning a new Dataset with newHeadRef set and ok set to true. In the event that the commit fails, ok is set to false and a new up-to-date Dataset is returned WITHOUT newHeadRef in it. The caller should try again using this new Dataset.
|
|
func SetNewHead(newHeadRef ref.Ref, ds dataset.Dataset) (dataset.Dataset, bool) {
|
|
commit := validateRefAsCommit(newHeadRef, ds.Store())
|
|
return ds.CommitWithParents(commit.Value(), datas.SetOfCommitFromVal(commit.Parents()))
|
|
}
|