chrislu
3 years ago
10 changed files with 137 additions and 72 deletions
-
11Makefile
-
72weed/command/filer_meta_tail.go
-
82weed/command/filer_meta_tail_elastic.go
-
14weed/command/filer_meta_tail_non_elastic.go
-
9weed/filer/elastic/v7/doc.go
-
3weed/filer/elastic/v7/elastic_store.go
-
3weed/filer/elastic/v7/elastic_store_kv.go
-
9weed/notification/gocdk_pub_sub/doc.go
-
3weed/notification/gocdk_pub_sub/gocdk_pub_sub.go
-
3weed/replication/sub/notification_gocdk_pub_sub.go
@ -0,0 +1,11 @@ |
|||
BINARY = weed |
|||
|
|||
SOURCE_DIR = . |
|||
|
|||
all: install |
|||
|
|||
install: |
|||
cd weed; go install |
|||
|
|||
full_install: |
|||
cd weed; go install -tags "elastic gocdk" |
@ -0,0 +1,82 @@ |
|||
//go:build elastic
|
|||
// +build elastic
|
|||
|
|||
package command |
|||
|
|||
import ( |
|||
"context" |
|||
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb" |
|||
"github.com/chrislusf/seaweedfs/weed/util" |
|||
jsoniter "github.com/json-iterator/go" |
|||
elastic "github.com/olivere/elastic/v7" |
|||
"strings" |
|||
) |
|||
|
|||
type EsDocument struct { |
|||
Dir string `json:"dir,omitempty"` |
|||
Name string `json:"name,omitempty"` |
|||
IsDirectory bool `json:"isDir,omitempty"` |
|||
Size uint64 `json:"size,omitempty"` |
|||
Uid uint32 `json:"uid,omitempty"` |
|||
Gid uint32 `json:"gid,omitempty"` |
|||
UserName string `json:"userName,omitempty"` |
|||
Collection string `json:"collection,omitempty"` |
|||
Crtime int64 `json:"crtime,omitempty"` |
|||
Mtime int64 `json:"mtime,omitempty"` |
|||
Mime string `json:"mime,omitempty"` |
|||
} |
|||
|
|||
func toEsEntry(event *filer_pb.EventNotification) (*EsDocument, string) { |
|||
entry := event.NewEntry |
|||
dir, name := event.NewParentPath, entry.Name |
|||
id := util.Md5String([]byte(util.NewFullPath(dir, name))) |
|||
esEntry := &EsDocument{ |
|||
Dir: dir, |
|||
Name: name, |
|||
IsDirectory: entry.IsDirectory, |
|||
Size: entry.Attributes.FileSize, |
|||
Uid: entry.Attributes.Uid, |
|||
Gid: entry.Attributes.Gid, |
|||
UserName: entry.Attributes.UserName, |
|||
Collection: entry.Attributes.Collection, |
|||
Crtime: entry.Attributes.Crtime, |
|||
Mtime: entry.Attributes.Mtime, |
|||
Mime: entry.Attributes.Mime, |
|||
} |
|||
return esEntry, id |
|||
} |
|||
|
|||
func sendToElasticSearchFunc(servers string, esIndex string) (func(resp *filer_pb.SubscribeMetadataResponse) error, error) { |
|||
options := []elastic.ClientOptionFunc{} |
|||
options = append(options, elastic.SetURL(strings.Split(servers, ",")...)) |
|||
options = append(options, elastic.SetSniff(false)) |
|||
options = append(options, elastic.SetHealthcheck(false)) |
|||
client, err := elastic.NewClient(options...) |
|||
if err != nil { |
|||
return nil, err |
|||
} |
|||
return func(resp *filer_pb.SubscribeMetadataResponse) error { |
|||
event := resp.EventNotification |
|||
if event.OldEntry != nil && |
|||
(event.NewEntry == nil || resp.Directory != event.NewParentPath || event.OldEntry.Name != event.NewEntry.Name) { |
|||
// delete or not update the same file
|
|||
dir, name := resp.Directory, event.OldEntry.Name |
|||
id := util.Md5String([]byte(util.NewFullPath(dir, name))) |
|||
println("delete", id) |
|||
_, err := client.Delete().Index(esIndex).Id(id).Do(context.Background()) |
|||
return err |
|||
} |
|||
if event.NewEntry != nil { |
|||
// add a new file or update the same file
|
|||
esEntry, id := toEsEntry(event) |
|||
value, err := jsoniter.Marshal(esEntry) |
|||
if err != nil { |
|||
return err |
|||
} |
|||
println(string(value)) |
|||
_, err = client.Index().Index(esIndex).Id(id).BodyJson(string(value)).Do(context.Background()) |
|||
return err |
|||
} |
|||
return nil |
|||
}, nil |
|||
} |
@ -0,0 +1,14 @@ |
|||
//go:build !elastic
|
|||
// +build !elastic
|
|||
|
|||
package command |
|||
|
|||
import ( |
|||
"github.com/chrislusf/seaweedfs/weed/pb/filer_pb" |
|||
) |
|||
|
|||
func sendToElasticSearchFunc(servers string, esIndex string) (func(resp *filer_pb.SubscribeMetadataResponse) error, error) { |
|||
return func(resp *filer_pb.SubscribeMetadataResponse) error { |
|||
return nil |
|||
}, nil |
|||
} |
@ -0,0 +1,9 @@ |
|||
/* |
|||
|
|||
Package elastic is for elastic filer store. |
|||
|
|||
The referenced "github.com/olivere/elastic/v7" library is too big when compiled. |
|||
So this is only compiled in "make full_install". |
|||
|
|||
*/ |
|||
package elastic |
@ -0,0 +1,9 @@ |
|||
/* |
|||
|
|||
Package gocdk_pub_sub is for Azure Service Bus and RabbitMQ. |
|||
|
|||
The referenced "gocloud.dev/pubsub" library is too big when compiled. |
|||
So this is only compiled in "make full_install". |
|||
|
|||
*/ |
|||
package gocdk_pub_sub |
Write
Preview
Loading…
Cancel
Save
Reference in new issue