aboutsummaryrefslogtreecommitdiff
path: root/weed/filer/arangodb/arangodb_store_bucket.go
blob: b35a5c9b37bb5ed8c002d572d85da89da5cb5f28 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
package arangodb

import (
	"context"
	"github.com/arangodb/go-driver"
	"time"

	"github.com/seaweedfs/seaweedfs/weed/filer"

	"github.com/seaweedfs/seaweedfs/weed/glog"
)

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 {
		glog.Errorf("bucket create %s: %v", bucket, err)
	}
}
func (store *ArangodbStore) OnBucketDeletion(bucket string) {
	timeout, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	collection, err := store.ensureBucket(timeout, bucket)
	if err != nil {
		glog.Errorf("bucket delete %s: %v", bucket, err)
		return
	}
	err = collection.Remove(timeout)
	if err != nil && !driver.IsNotFound(err) {
		glog.Errorf("bucket delete %s: %v", bucket, err)
		return
	}
}
func (store *ArangodbStore) CanDropWholeBucket() bool {
	return true
}