seaweedfs/weed/filer/arangodb/arangodb_store_bucket.go

44 lines
1.1 KiB
Go
Raw Normal View History

2022-03-18 02:12:25 +00:00
package arangodb
import (
"context"
2022-04-20 23:21:56 +00:00
"github.com/arangodb/go-driver"
2022-03-18 02:12:25 +00:00
"time"
"github.com/seaweedfs/seaweedfs/weed/filer"
"github.com/seaweedfs/seaweedfs/weed/glog"
2022-03-18 02:12:25 +00:00
)
var _ filer.BucketAware = (*ArangodbStore)(nil)
func (store *ArangodbStore) OnBucketCreation(bucket string) {
timeout, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
// create the collection && add to cache
_, err := store.ensureBucket(timeout, bucket)
if err != nil {
2022-04-20 23:21:56 +00:00
glog.Errorf("bucket create %s: %v", bucket, err)
}
2022-03-18 02:12:25 +00:00
}
func (store *ArangodbStore) OnBucketDeletion(bucket string) {
timeout, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()
collection, err := store.ensureBucket(timeout, bucket)
2022-03-18 02:12:25 +00:00
if err != nil {
2022-04-20 23:21:56 +00:00
glog.Errorf("bucket delete %s: %v", bucket, err)
return
}
err = collection.Remove(timeout)
if err != nil && !driver.IsNotFound(err) {
2022-04-20 23:21:56 +00:00
glog.Errorf("bucket delete %s: %v", bucket, err)
return
2022-03-18 02:12:25 +00:00
}
store.mu.Lock()
delete(store.buckets, bucket)
store.mu.Unlock()
2022-03-18 02:12:25 +00:00
}
func (store *ArangodbStore) CanDropWholeBucket() bool {
return true
}