Merge pull request #786 from stevvooe/edit-content-command

cmd/dist: add ability to edit content

services/content/ingester.go

@@ -127,6 +127,9 @@ func (rw *remoteWriter) Write(p []byte) (n int, err error) {
 	}
 
 	rw.offset += int64(n)
+	if resp.Digest != "" {
+		rw.digest = resp.Digest
+	}
 	return
 }
 
@@ -149,6 +152,8 @@ func (rw *remoteWriter) Commit(size int64, expected digest.Digest) error {
 		return errors.Errorf("unexpected digest: %v != %v", resp.Digest, expected)
 	}
 
+	rw.digest = resp.Digest
+	rw.offset = resp.Offset
 	return nil
 }
 

services/content/service.go

@@ -304,9 +304,9 @@ func (s *Service) Write(session api.Content_WriteServer) (err error) {
 				if err := wr.Commit(total, expected); err != nil {
 					return err
 				}
-
-				msg.Digest = wr.Digest()
 			}
+
+			msg.Digest = wr.Digest()
 		case api.WriteActionAbort:
 			return s.store.Abort(ref)
 		}