Merge pull request #6311 from dolthub/pavel/bug-6306

This commit is contained in:
Pavel Safronov
2023-07-11 08:38:27 -10:00
committed by GitHub

View File

@@ -679,10 +679,16 @@ type repoStateReader struct {
}
func (r *repoStateReader) CWBHeadRef() (ref.DoltRef, error) {
if r.RepoState == nil && r.RSLoadErr != nil {
return nil, r.RSLoadErr
}
return r.RepoState.CWBHeadRef(), nil
}
func (r *repoStateReader) CWBHeadSpec() (*doltdb.CommitSpec, error) {
if r.RepoState == nil && r.RSLoadErr != nil {
return nil, r.RSLoadErr
}
return r.RepoState.CWBHeadSpec(), nil
}
@@ -695,6 +701,10 @@ type repoStateWriter struct {
}
func (r *repoStateWriter) SetCWBHeadRef(ctx context.Context, marshalableRef ref.MarshalableRef) error {
if r.RepoState == nil && r.RSLoadErr != nil {
return r.RSLoadErr
}
r.RepoState.Head = marshalableRef
err := r.RepoState.Save(r.FS)