Achieve capacity limits

Author: duanhongyi

.gitignore

@@ -3,3 +3,4 @@ goofys
 goofys.test
 xout
 s3proxy.jar
+.vscode

api/common/config.go

@@ -27,7 +27,7 @@ import (
 )
 
 type PartSizeConfig struct {
-	PartSize uint64
+	PartSize  uint64
 	PartCount uint64
 }
 
@@ -45,8 +45,12 @@ type FlagStorage struct {
 
 	// Common Backend Config
 	UseContentType bool
-	Endpoint       string
-	Backend        interface{}
+
+	Provider          string
+	Capacity          uint64
+	DiskUsageInterval uint64
+	Endpoint          string
+	Backend           interface{}
 
 	// Tuning
 	MemoryLimit           uint64

go.mod

@@ -16,24 +16,21 @@ require (
 	github.com/aws/aws-sdk-go v1.38.7
 	github.com/google/btree v1.0.0
 	github.com/google/uuid v1.1.2
-	github.com/gopherjs/gopherjs v0.0.0-20210202160940-bed99a852dfe // indirect
 	github.com/jacobsa/fuse v0.0.0-20210818065549-10d864429bf7
-	github.com/jtolds/gls v4.2.0+incompatible // indirect
 	github.com/kardianos/osext v0.0.0-20190222173326-2bc1f35cddc0
 	github.com/kr/pretty v0.1.1-0.20190720101428-71e7e4993750 // indirect
 	github.com/mattn/go-ieproxy v0.0.0-20190805055040-f9202b1cfdeb // indirect
+	github.com/minio/madmin-go v1.4.6
 	github.com/mitchellh/go-homedir v1.1.0
 	github.com/satori/go.uuid v1.2.1-0.20181028125025-b2ce2384e17b
 	github.com/sevlyar/go-daemon v0.1.5
 	github.com/shirou/gopsutil v0.0.0-20190731134726-d80c43f9c984
 	github.com/sirupsen/logrus v1.8.1
-	github.com/smartystreets/assertions v0.0.0-20160201214316-443d812296a8 // indirect
-	github.com/smartystreets/goconvey v1.6.1-0.20160119221636-995f5b2e021c // indirect
 	github.com/urfave/cli v1.21.1-0.20190807111034-521735b7608a
 	golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a
 	google.golang.org/api v0.49.0
 	gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127
-	gopkg.in/ini.v1 v1.46.0
+	gopkg.in/ini.v1 v1.62.0
 )
 
 replace github.com/aws/aws-sdk-go => ./s3ext

go.sum

@@ -27,7 +27,7 @@ import (
 )
 
 type Capabilities struct {
-	MaxMultipartSize    uint64
+	MaxMultipartSize uint64
 	// indicates that the blob store has native support for directories
 	DirBlob bool
 	Name    string
@@ -44,7 +44,7 @@ type BlobItemOutput struct {
 	Size         uint64
 	StorageClass *string
 	// may be nil in list responses for backends that don't return metadata in listings
-	Metadata     map[string]*string
+	Metadata map[string]*string
 }
 
 type HeadBlobOutput struct {
@@ -179,7 +179,7 @@ type MultipartBlobAddInput struct {
 
 type MultipartBlobAddOutput struct {
 	RequestId string
-	PartId *string
+	PartId    *string
 }
 
 type MultipartBlobCopyInput struct {
@@ -192,7 +192,7 @@ type MultipartBlobCopyInput struct {
 
 type MultipartBlobCopyOutput struct {
 	RequestId string
-	PartId *string
+	PartId    *string
 }
 
 type MultipartBlobCommitOutput struct {
@@ -228,6 +228,14 @@ type MakeBucketOutput struct {
 	RequestId string
 }
 
+type GetBucketUsageInput struct {
+}
+
+type GetBucketUsageOutput struct {
+	Size      uint64
+	RequestId string
+}
+
 /// Implementations of all the functions here are expected to be
 /// concurrency-safe, except for
 ///
@@ -256,6 +264,7 @@ type StorageBackend interface {
 	MultipartExpire(param *MultipartExpireInput) (*MultipartExpireOutput, error)
 	RemoveBucket(param *RemoveBucketInput) (*RemoveBucketOutput, error)
 	MakeBucket(param *MakeBucketInput) (*MakeBucketOutput, error)
+	GetBucketUsage(param *GetBucketUsageInput) (*GetBucketUsageOutput, error)
 	Delegate() interface{}
 }
 
@@ -424,6 +433,11 @@ func (s *StorageBackendInitWrapper) MakeBucket(param *MakeBucketInput) (*MakeBuc
 	return s.StorageBackend.MakeBucket(param)
 }
 
+func (s *StorageBackendInitWrapper) GetBucketUsage(param *GetBucketUsageInput) (*GetBucketUsageOutput, error) {
+	s.Init("")
+	return s.StorageBackend.GetBucketUsage(param)
+}
+
 type StorageBackendInitError struct {
 	error
 	cap Capabilities
@@ -547,3 +561,7 @@ func (e StorageBackendInitError) RemoveBucket(param *RemoveBucketInput) (*Remove
 func (e StorageBackendInitError) MakeBucket(param *MakeBucketInput) (*MakeBucketOutput, error) {
 	return nil, e
 }
+
+func (e StorageBackendInitError) GetBucketUsage(param *GetBucketUsageInput) (*GetBucketUsageOutput, error) {
+	return nil, e
+}

internal/backend.go

@@ -152,8 +152,8 @@ func NewADLv1(bucket string, flags *FlagStorage, config *ADLv1Config) (*ADLv1, e
 		bucket:  bucket,
 		cap: Capabilities{
 			//NoParallelMultipart: true,
-			DirBlob:             true,
-			Name:                "adl",
+			DirBlob: true,
+			Name:    "adl",
 			// ADLv1 fails with 404 if we upload data
 			// larger than 30000000 bytes (28.6MB) (28MB
 			// also failed in at one point, but as of
@@ -701,3 +701,7 @@ func (b *ADLv1) mkdir(dir string) error {
 	}
 	return nil
 }
+
+func (b *ADLv1) GetBucketUsage(param *GetBucketUsageInput) (*GetBucketUsageOutput, error) {
+	return &GetBucketUsageOutput{Size: 0}, nil
+}

internal/backend_adlv1.go

@@ -207,6 +207,10 @@ func (b *ADLv2) Capabilities() *Capabilities {
 	return &b.cap
 }
 
+func (b *ADLv2) GetBucketUsage(param *GetBucketUsageInput) (*GetBucketUsageOutput, error) {
+	return &GetBucketUsageOutput{Size: 0}, nil
+}
+
 type ADL2Error struct {
 	adl2.DataLakeStorageError
 }

internal/backend_adlv2.go

@@ -222,7 +222,7 @@ func (b *AZBlob) refreshToken() (*azblob.ContainerURL, error) {
 		// our token totally expired, renew inline before using it
 		b.mu.Unlock()
 		b.tokenRenewGate <- 1
-		defer func() { <- b.tokenRenewGate } ()
+		defer func() { <-b.tokenRenewGate }()
 
 		b.mu.Lock()
 		// check again, because in the mean time maybe it's renewed
@@ -248,7 +248,7 @@ func (b *AZBlob) refreshToken() (*azblob.ContainerURL, error) {
 				if err != nil {
 					azbLog.Errorf("Unable to refresh token: %v", err)
 				}
-				<- b.tokenRenewGate
+				<-b.tokenRenewGate
 			}()
 
 			// if we cannot renew token, treat it as a
@@ -665,7 +665,7 @@ func (b *AZBlob) DeleteBlobs(param *DeleteBlobsInput) (ret *DeleteBlobsOutput, d
 
 		go func(key string) {
 			defer func() {
-				<- SmallActionsGate
+				<-SmallActionsGate
 				wg.Done()
 			}()
 
@@ -947,3 +947,7 @@ func (b *AZBlob) MakeBucket(param *MakeBucketInput) (*MakeBucketOutput, error) {
 	}
 	return &MakeBucketOutput{}, nil
 }
+
+func (b *AZBlob) GetBucketUsage(param *GetBucketUsageInput) (*GetBucketUsageOutput, error) {
+	return &GetBucketUsageOutput{Size: 0}, nil
+}

internal/backend_azblob.go

@@ -23,15 +23,15 @@ import (
 	"sync"
 	"syscall"
 
-	"github.com/jacobsa/fuse"
 	"cloud.google.com/go/storage"
+	"github.com/jacobsa/fuse"
 	"google.golang.org/api/iterator"
 )
 
 // GCS variant of S3
 type GCS3 struct {
 	*S3Backend
-	gcs *storage.Client
+	gcs          *storage.Client
 	jsonCredFile string
 }
 
@@ -103,12 +103,12 @@ func (s *GCS3) ListBlobs(param *ListBlobsInput) (*ListBlobsOutput, error) {
 			})
 		} else {
 			items = append(items, BlobItemOutput{
-				Key: &attrs.Name,
-				ETag: &attrs.Etag,
+				Key:          &attrs.Name,
+				ETag:         &attrs.Etag,
 				LastModified: &attrs.Updated,
-				Size: uint64(attrs.Size),
+				Size:         uint64(attrs.Size),
 				StorageClass: &attrs.StorageClass,
-				Metadata: PMetadata(attrs.Metadata),
+				Metadata:     PMetadata(attrs.Metadata),
 			})
 		}
 		n++
@@ -156,3 +156,7 @@ func (s *GCS3) DeleteBlobs(param *DeleteBlobsInput) (*DeleteBlobsOutput, error)
 func (s *GCS3) MultipartBlobCopy(param *MultipartBlobCopyInput) (*MultipartBlobCopyOutput, error) {
 	return nil, syscall.ENOSYS
 }
+
+func (s *GCS3) GetBucketUsage(param *GetBucketUsageInput) (*GetBucketUsageOutput, error) {
+	return &GetBucketUsageOutput{Size: 0}, nil
+}

internal/backend_gcs3.go

@@ -0,0 +1,68 @@
+// Copyright 2019 Ka-Hing Cheung
+// Copyright 2021 Yandex LLC
+//
+// 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 internal
+
+import (
+	"context"
+	"encoding/json"
+	"io/ioutil"
+	"net/url"
+
+	"github.com/minio/madmin-go"
+	. "github.com/yandex-cloud/geesefs/api/common"
+)
+
+type BucketsUsage struct {
+	BucketsSizes map[string]uint64
+}
+
+type MinioBackend struct {
+	*S3Backend
+}
+
+func NewMinio(bucket string, flags *FlagStorage, config *S3Config) (*MinioBackend, error) {
+	s3Backend, err := NewS3(bucket, flags, config)
+	if err != nil {
+		return nil, err
+	}
+	s3Backend.Capabilities().Name = "minio"
+	s := &MinioBackend{S3Backend: s3Backend}
+	return s, nil
+}
+
+func (s *MinioBackend) GetBucketUsage(param *GetBucketUsageInput) (*GetBucketUsageOutput, error) {
+	value, err := s.Config.Credentials.Get()
+	if err != nil {
+		return nil, err
+	}
+
+	endpointURL, _ := url.Parse(s.Endpoint)
+	madminClient, err := madmin.New(endpointURL.Host, value.AccessKeyID, value.SecretAccessKey, endpointURL.Scheme == "https")
+	if err != nil {
+		return nil, err
+	}
+	resp, err := madminClient.ExecuteMethod(context.Background(), "GET", madmin.RequestData{RelPath: "/v3/datausageinfo"})
+	if err != nil {
+		return nil, err
+	}
+	response, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return nil, err
+	}
+	var result BucketsUsage
+	json.Unmarshal(response, &result)
+	return &GetBucketUsageOutput{Size: result.BucketsSizes[s.Bucket()]}, nil
+}