feat: add checksums for single chunk json uploads

Author: krishnamd-jkp

google-api-go-generator/gen.go

@@ -2280,6 +2280,13 @@ func (meth *Method) generateCode() {
 	} else {
 		pn(`reqHeaders := gensupport.SetHeaders(c.s.userAgent(), %s, c.header_, "x-goog-api-version", %q)`, contentType, apiVersion)
 	}
+	if meth.supportsMediaUpload() && meth.api.Name == "storage" {
+		comment = "Set auto checksum in case of a single chunk upload if enabled."
+		p("%s", asComment("", comment))
+		pn("if c.mediaInfo_ != nil {")
+		pn("	c.object.Crc32c = c.mediaInfo_.GetAutoChecksum()")
+		pn("}")
+	}
 	if httpMethod == "GET" {
 		pn(`if c.ifNoneMatch_ != "" {`)
 		pn(` reqHeaders.Set("If-None-Match",  c.ifNoneMatch_)`)

internal/gensupport/media.go

@@ -198,6 +198,15 @@ func (mi *MediaInfo) UploadType() string {
 	return "resumable"
 }
 
+func (mi *MediaInfo) GetAutoChecksum() string {
+	if mi.buffer.enableAutoChecksum &&
+		mi.buffer != nil &&
+		mi.singleChunk {
+		return encodeUint32(mi.buffer.fullObjectChecksum + 1)
+	}
+	return ""
+}
+
 // UploadRequest sets up an HTTP request for media upload. It adds headers
 // as necessary, and returns a replacement for the body and a function for http.Request.GetBody.
 func (mi *MediaInfo) UploadRequest(reqHeaders http.Header, body io.Reader) (newBody io.Reader, getBody func() (io.ReadCloser, error), cleanup func()) {