2019-12-12 15:02:37 +01:00
|
|
|
/*
|
|
|
|
* MinIO Cloud Storage, (C) 2019 MinIO, Inc.
|
|
|
|
*
|
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
|
* you may not use this file except in compliance with the License.
|
|
|
|
* You may obtain a copy of the License at
|
|
|
|
*
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
*
|
|
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
|
|
* See the License for the specific language governing permissions and
|
|
|
|
* limitations under the License.
|
|
|
|
*/
|
|
|
|
|
|
|
|
package cmd
|
|
|
|
|
|
|
|
import (
|
|
|
|
"bytes"
|
|
|
|
"context"
|
|
|
|
"encoding/json"
|
2021-02-05 18:57:30 +01:00
|
|
|
"net/http"
|
2019-12-12 15:02:37 +01:00
|
|
|
|
2020-01-08 12:31:43 +01:00
|
|
|
jsoniter "github.com/json-iterator/go"
|
2019-12-12 15:02:37 +01:00
|
|
|
"github.com/minio/minio/cmd/logger"
|
|
|
|
"github.com/minio/minio/pkg/hash"
|
2021-04-03 18:03:42 +02:00
|
|
|
"github.com/minio/minio/pkg/madmin"
|
2019-12-12 15:02:37 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
const (
|
2020-04-28 10:16:57 +02:00
|
|
|
dataUsageRoot = SlashSeparator
|
|
|
|
dataUsageBucket = minioMetaBucket + SlashSeparator + bucketMetaPrefix
|
|
|
|
|
2020-06-12 19:28:21 +02:00
|
|
|
dataUsageObjName = ".usage.json"
|
|
|
|
dataUsageCacheName = ".usage-cache.bin"
|
|
|
|
dataUsageBloomName = ".bloomcycle.bin"
|
2019-12-12 15:02:37 +01:00
|
|
|
)
|
|
|
|
|
2020-03-19 00:19:29 +01:00
|
|
|
// storeDataUsageInBackend will store all objects sent on the gui channel until closed.
|
2021-04-03 18:03:42 +02:00
|
|
|
func storeDataUsageInBackend(ctx context.Context, objAPI ObjectLayer, dui <-chan madmin.DataUsageInfo) {
|
2020-12-07 22:47:48 +01:00
|
|
|
for dataUsageInfo := range dui {
|
2020-03-19 17:47:47 +01:00
|
|
|
dataUsageJSON, err := json.Marshal(dataUsageInfo)
|
2020-03-19 00:19:29 +01:00
|
|
|
if err != nil {
|
|
|
|
logger.LogIf(ctx, err)
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
size := int64(len(dataUsageJSON))
|
pkg/etag: add new package for S3 ETag handling (#11577)
This commit adds a new package `etag` for dealing
with S3 ETags.
Even though ETag is often viewed as MD5 checksum of
an object, handling S3 ETags correctly is a surprisingly
complex task. While it is true that the ETag corresponds
to the MD5 for the most basic S3 API operations, there are
many exceptions in case of multipart uploads or encryption.
In worse, some S3 clients expect very specific behavior when
it comes to ETags. For example, some clients expect that the
ETag is a double-quoted string and fail otherwise.
Non-AWS compliant ETag handling has been a source of many bugs
in the past.
Therefore, this commit adds a dedicated `etag` package that provides
functionality for parsing, generating and converting S3 ETags.
Further, this commit removes the ETag computation from the `hash`
package. Instead, the `hash` package (i.e. `hash.Reader`) should
focus only on computing and verifying the content-sha256.
One core feature of this commit is to provide a mechanism to
communicate a computed ETag from a low-level `io.Reader` to
a high-level `io.Reader`.
This problem occurs when an S3 server receives a request and
has to compute the ETag of the content. However, the server
may also wrap the initial body with several other `io.Reader`,
e.g. when encrypting or compressing the content:
```
reader := Encrypt(Compress(ETag(content)))
```
In such a case, the ETag should be accessible by the high-level
`io.Reader`.
The `etag` provides a mechanism to wrap `io.Reader` implementations
such that the `ETag` can be accessed by a type-check.
This technique is applied to the PUT, COPY and Upload handlers.
2021-02-23 21:31:53 +01:00
|
|
|
r, err := hash.NewReader(bytes.NewReader(dataUsageJSON), size, "", "", size)
|
2020-03-19 00:19:29 +01:00
|
|
|
if err != nil {
|
|
|
|
logger.LogIf(ctx, err)
|
|
|
|
continue
|
|
|
|
}
|
2021-02-10 17:52:50 +01:00
|
|
|
_, err = objAPI.PutObject(ctx, dataUsageBucket, dataUsageObjName, NewPutObjReader(r), ObjectOptions{})
|
2020-04-28 10:16:57 +02:00
|
|
|
if !isErrBucketNotFound(err) {
|
|
|
|
logger.LogIf(ctx, err)
|
|
|
|
}
|
2019-12-12 15:02:37 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-04-03 18:03:42 +02:00
|
|
|
func loadDataUsageFromBackend(ctx context.Context, objAPI ObjectLayer) (madmin.DataUsageInfo, error) {
|
2021-02-05 18:57:30 +01:00
|
|
|
r, err := objAPI.GetObjectNInfo(ctx, dataUsageBucket, dataUsageObjName, nil, http.Header{}, readLock, ObjectOptions{})
|
2019-12-12 15:02:37 +01:00
|
|
|
if err != nil {
|
2020-04-28 10:16:57 +02:00
|
|
|
if isErrObjectNotFound(err) || isErrBucketNotFound(err) {
|
2021-04-03 18:03:42 +02:00
|
|
|
return madmin.DataUsageInfo{}, nil
|
2020-01-21 23:07:49 +01:00
|
|
|
}
|
2021-04-03 18:03:42 +02:00
|
|
|
return madmin.DataUsageInfo{}, toObjectErr(err, dataUsageBucket, dataUsageObjName)
|
2019-12-12 15:02:37 +01:00
|
|
|
}
|
2021-02-05 18:57:30 +01:00
|
|
|
defer r.Close()
|
2019-12-12 15:02:37 +01:00
|
|
|
|
2021-04-03 18:03:42 +02:00
|
|
|
var dataUsageInfo madmin.DataUsageInfo
|
2020-01-08 12:31:43 +01:00
|
|
|
var json = jsoniter.ConfigCompatibleWithStandardLibrary
|
2021-02-05 18:57:30 +01:00
|
|
|
if err = json.NewDecoder(r).Decode(&dataUsageInfo); err != nil {
|
2021-04-03 18:03:42 +02:00
|
|
|
return madmin.DataUsageInfo{}, err
|
2019-12-12 15:02:37 +01:00
|
|
|
}
|
|
|
|
|
2020-05-27 15:45:43 +02:00
|
|
|
// For forward compatibility reasons, we need to add this code.
|
|
|
|
if len(dataUsageInfo.BucketsUsage) == 0 {
|
2021-04-03 18:03:42 +02:00
|
|
|
dataUsageInfo.BucketsUsage = make(map[string]madmin.BucketUsageInfo, len(dataUsageInfo.BucketSizes))
|
2020-05-27 15:45:43 +02:00
|
|
|
for bucket, size := range dataUsageInfo.BucketSizes {
|
2021-04-03 18:03:42 +02:00
|
|
|
dataUsageInfo.BucketsUsage[bucket] = madmin.BucketUsageInfo{Size: size}
|
2020-05-27 15:45:43 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// For backward compatibility reasons, we need to add this code.
|
|
|
|
if len(dataUsageInfo.BucketSizes) == 0 {
|
|
|
|
dataUsageInfo.BucketSizes = make(map[string]uint64, len(dataUsageInfo.BucketsUsage))
|
|
|
|
for bucket, bui := range dataUsageInfo.BucketsUsage {
|
|
|
|
dataUsageInfo.BucketSizes[bucket] = bui.Size
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-12-12 15:02:37 +01:00
|
|
|
return dataUsageInfo, nil
|
|
|
|
}
|