snapshot of https://github.com/seaweedfs/seaweedfs.git (master, 2026-09-16; blobs > 131072B omitted)

weed/filer/tarantool/doc.go

@@ -0,0 +1,7 @@
+/*
+Package tarantool is for Tarantool filer store.
+
+The referenced "github.com/tarantool/go-tarantool/v2" library is too big when compiled.
+So this is only compiled in "make full_install".
+*/
+package tarantool

weed/filer/tarantool/readme.md

@@ -0,0 +1,11 @@
+## Tarantool
+
+database: https://www.tarantool.io/
+
+go driver: https://github.com/tarantool/go-tarantool/
+
+To set up local env:
+`make -C docker test_tarantool`
+
+Run tests:
+`RUN_TARANTOOL_TESTS=1 go test -tags=tarantool ./weed/filer/tarantool`

weed/filer/tarantool/tarantool_slog.go

@@ -0,0 +1,75 @@
+//go:build tarantool
+
+package tarantool
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"strings"
+
+	"github.com/seaweedfs/seaweedfs/weed/glog"
+)
+
+// glogSlogHandler forwards go-tarantool/v3's structured logs into weed/glog,
+// which is the logging facility used everywhere else in SeaweedFS.
+type glogSlogHandler struct {
+	preformatted string
+}
+
+func newGlogLogger() *slog.Logger {
+	return slog.New(&glogSlogHandler{})
+}
+
+func (h *glogSlogHandler) Enabled(context.Context, slog.Level) bool {
+	return true
+}
+
+func (h *glogSlogHandler) Handle(ctx context.Context, r slog.Record) error {
+	msg := r.Message
+	var attrs []string
+	if h.preformatted != "" {
+		attrs = append(attrs, h.preformatted)
+	}
+	r.Attrs(func(a slog.Attr) bool {
+		attrs = append(attrs, fmt.Sprintf("%s=%v", a.Key, a.Value.Any()))
+		return true
+	})
+	if len(attrs) > 0 {
+		msg = msg + " " + strings.Join(attrs, " ")
+	}
+
+	switch {
+	case r.Level >= slog.LevelError:
+		glog.ErrorfCtx(ctx, "tarantool: %s", msg)
+	case r.Level >= slog.LevelWarn:
+		glog.WarningfCtx(ctx, "tarantool: %s", msg)
+	case r.Level >= slog.LevelInfo:
+		glog.InfofCtx(ctx, "tarantool: %s", msg)
+	default:
+		glog.V(1).InfofCtx(ctx, "tarantool: %s", msg)
+	}
+	return nil
+}
+
+func (h *glogSlogHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
+	if len(attrs) == 0 {
+		return h
+	}
+	var sb strings.Builder
+	if h.preformatted != "" {
+		sb.WriteString(h.preformatted)
+		sb.WriteString(" ")
+	}
+	for i, a := range attrs {
+		if i > 0 {
+			sb.WriteString(" ")
+		}
+		sb.WriteString(fmt.Sprintf("%s=%v", a.Key, a.Value.Any()))
+	}
+	return &glogSlogHandler{preformatted: sb.String()}
+}
+
+func (h *glogSlogHandler) WithGroup(name string) slog.Handler {
+	return h
+}

weed/filer/tarantool/tarantool_store.go

@@ -0,0 +1,328 @@
+//go:build tarantool
+
+package tarantool
+
+import (
+	"context"
+	"fmt"
+	"reflect"
+	"strings"
+	"time"
+
+	"github.com/seaweedfs/seaweedfs/weed/filer"
+	"github.com/seaweedfs/seaweedfs/weed/glog"
+	"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
+	"github.com/seaweedfs/seaweedfs/weed/util"
+	weed_util "github.com/seaweedfs/seaweedfs/weed/util"
+	"github.com/tarantool/go-option"
+	"github.com/tarantool/go-tarantool/v3"
+	"github.com/tarantool/go-tarantool/v3/crud"
+	"github.com/tarantool/go-tarantool/v3/pool"
+)
+
+const (
+	tarantoolSpaceName = "filer_metadata"
+)
+
+func init() {
+	filer.Stores = append(filer.Stores, &TarantoolStore{})
+}
+
+type TarantoolStore struct {
+	pool *pool.Pool
+}
+
+func (store *TarantoolStore) GetName() string {
+	return "tarantool"
+}
+
+func (store *TarantoolStore) Initialize(configuration weed_util.Configuration, prefix string) error {
+
+	configuration.SetDefault(prefix+"address", "localhost:3301")
+	configuration.SetDefault(prefix+"user", "guest")
+	configuration.SetDefault(prefix+"password", "")
+	configuration.SetDefault(prefix+"timeout", "5s")
+	configuration.SetDefault(prefix+"maxReconnects", "1000")
+
+	address := configuration.GetString(prefix + "address")
+	user := configuration.GetString(prefix + "user")
+	password := configuration.GetString(prefix + "password")
+
+	timeoutStr := configuration.GetString(prefix + "timeout")
+	timeout, err := time.ParseDuration(timeoutStr)
+	if err != nil {
+		return fmt.Errorf("parse tarantool store timeout: %w", err)
+	}
+
+	if maxReconnects := configuration.GetInt(prefix + "maxReconnects"); maxReconnects != 1000 {
+		glog.Warningf("tarantool store: \"maxReconnects\" is no longer supported with go-tarantool v3 (the connection pool manages reconnection internally); ignoring configured value %d", maxReconnects)
+	}
+
+	addresses := strings.Split(address, ",")
+
+	return store.initialize(addresses, user, password, timeout)
+}
+
+func (store *TarantoolStore) initialize(addresses []string, user string, password string, timeout time.Duration) error {
+
+	opts := tarantool.Opts{
+		Timeout: timeout,
+	}
+
+	poolInstances := makePoolInstances(addresses, user, password, opts)
+	poolOpts := pool.Opts{
+		CheckTimeout: time.Second,
+		Logger:       newGlogLogger(),
+	}
+
+	ctx := context.Background()
+	p, err := pool.NewWithOpts(ctx, poolInstances, poolOpts)
+	if err != nil {
+		return fmt.Errorf("Can't create connection pool: %w", err)
+	}
+
+	_, err = p.Do(tarantool.NewPingRequest(), pool.ModeAny).Get()
+	if err != nil {
+		return err
+	}
+
+	store.pool = p
+
+	return nil
+}
+
+func makePoolInstances(addresses []string, user string, password string, opts tarantool.Opts) []pool.Instance {
+	poolInstances := make([]pool.Instance, 0, len(addresses))
+	for i, address := range addresses {
+		poolInstances = append(poolInstances, makePoolInstance(address, user, password, opts, i))
+	}
+	return poolInstances
+}
+
+func makePoolInstance(address string, user string, password string, opts tarantool.Opts, serial int) pool.Instance {
+	return pool.Instance{
+		Name: fmt.Sprintf("instance%d", serial),
+		Dialer: tarantool.NetDialer{
+			Address:  address,
+			User:     user,
+			Password: password,
+		},
+		Opts: opts,
+	}
+}
+
+func (store *TarantoolStore) BeginTransaction(ctx context.Context) (context.Context, error) {
+	return ctx, nil
+}
+
+func (store *TarantoolStore) CommitTransaction(ctx context.Context) error {
+	return nil
+}
+
+func (store *TarantoolStore) RollbackTransaction(ctx context.Context) error {
+	return nil
+}
+
+func (store *TarantoolStore) InsertEntry(ctx context.Context, entry *filer.Entry) (err error) {
+	dir, name := entry.FullPath.DirAndName()
+	meta, err := entry.EncodeAttributesAndChunks()
+	if err != nil {
+		return fmt.Errorf("encode %s: %s", entry.FullPath, err)
+	}
+
+	if len(entry.GetChunks()) > filer.CountEntryChunksForGzip {
+		meta = util.MaybeGzipData(meta)
+	}
+
+	var ttl int64
+	if entry.TtlSec > 0 {
+		ttl = time.Now().Unix() + int64(entry.TtlSec)
+	} else {
+		ttl = 0
+	}
+
+	var operations = []crud.Operation{
+		{
+			Operator: crud.Assign,
+			Field:    "data",
+			Value:    string(meta),
+		},
+	}
+
+	req := crud.NewUpsertRequest(tarantoolSpaceName).
+		Tuple([]interface{}{dir, nil, name, ttl, string(meta)}).
+		Operations(operations).
+		Context(ctx)
+
+	ret := crud.Result{}
+
+	if err := store.pool.Do(req, pool.ModeRW).GetTyped(&ret); err != nil {
+		return fmt.Errorf("insert %s: %s", entry.FullPath, err)
+	}
+
+	return nil
+}
+
+func (store *TarantoolStore) UpdateEntry(ctx context.Context, entry *filer.Entry) (err error) {
+	return store.InsertEntry(ctx, entry)
+}
+
+func (store *TarantoolStore) FindEntry(ctx context.Context, fullpath weed_util.FullPath) (entry *filer.Entry, err error) {
+	dir, name := fullpath.DirAndName()
+
+	findEntryGetOpts := crud.GetOpts{
+		Fields:        option.SomeAny([]interface{}{"data"}),
+		Mode:          option.SomeString("read"),
+		PreferReplica: option.SomeBool(true),
+		Balance:       option.SomeBool(true),
+	}
+
+	req := crud.NewGetRequest(tarantoolSpaceName).
+		Key([]interface{}{dir, name}).
+		Opts(findEntryGetOpts).
+		Context(ctx)
+
+	resp := crud.Result{}
+
+	err = store.pool.Do(req, pool.ModePreferRO).GetTyped(&resp)
+	if err != nil {
+		return nil, err
+	}
+
+	results, ok := resp.Rows.([]interface{})
+	if !ok || len(results) != 1 {
+		return nil, filer_pb.ErrNotFound
+	}
+
+	rows, ok := results[0].([]interface{})
+	if !ok || len(rows) != 1 {
+		return nil, filer_pb.ErrNotFound
+	}
+
+	row, ok := rows[0].(string)
+	if !ok {
+		return nil, fmt.Errorf("Can't convert rows[0] field to string. Actual type: %v, value: %v", reflect.TypeOf(rows[0]), rows[0])
+	}
+
+	entry = &filer.Entry{
+		FullPath: fullpath,
+	}
+
+	err = entry.DecodeAttributesAndChunks(weed_util.MaybeDecompressData([]byte(row)))
+	if err != nil {
+		return entry, fmt.Errorf("decode %s : %v", entry.FullPath, err)
+	}
+
+	return entry, nil
+}
+
+func (store *TarantoolStore) DeleteEntry(ctx context.Context, fullpath weed_util.FullPath) (err error) {
+	dir, name := fullpath.DirAndName()
+
+	delOpts := crud.DeleteOpts{
+		Noreturn: option.SomeBool(true),
+	}
+
+	req := crud.NewDeleteRequest(tarantoolSpaceName).
+		Key([]interface{}{dir, name}).
+		Opts(delOpts).
+		Context(ctx)
+
+	if _, err := store.pool.Do(req, pool.ModeRW).Get(); err != nil {
+		return fmt.Errorf("delete %s : %v", fullpath, err)
+	}
+
+	return nil
+}
+
+func (store *TarantoolStore) DeleteFolderChildren(ctx context.Context, fullpath weed_util.FullPath) (err error) {
+	req := tarantool.NewCallRequest("filer_metadata.delete_by_directory_idx").
+		Args([]interface{}{fullpath}).
+		Context(ctx)
+
+	if _, err := store.pool.Do(req, pool.ModeRW).Get(); err != nil {
+		return fmt.Errorf("delete %s : %v", fullpath, err)
+	}
+
+	return nil
+}
+
+func (store *TarantoolStore) ListDirectoryPrefixedEntries(ctx context.Context, dirPath weed_util.FullPath, startFileName string, includeStartFile bool, limit int64, prefix string, eachEntryFunc filer.ListEachEntryFunc) (lastFileName string, err error) {
+	return lastFileName, filer.ErrUnsupportedListDirectoryPrefixed
+}
+
+func (store *TarantoolStore) ListDirectoryEntries(ctx context.Context, dirPath weed_util.FullPath, startFileName string, includeStartFile bool, limit int64, eachEntryFunc filer.ListEachEntryFunc) (lastFileName string, err error) {
+
+	req := tarantool.NewCallRequest("filer_metadata.find_by_directory_idx_and_name").
+		Args([]interface{}{string(dirPath), startFileName, includeStartFile, limit}).
+		Context(ctx)
+
+	results, err := store.pool.Do(req, pool.ModePreferRO).Get()
+	if err != nil {
+		return
+	}
+
+	if len(results) < 1 {
+		glog.ErrorfCtx(ctx, "Can't find results, data is empty")
+		return
+	}
+
+	rows, ok := results[0].([]interface{})
+	if !ok {
+		glog.ErrorfCtx(ctx, "Can't convert results[0] to list")
+		return
+	}
+
+	for _, result := range rows {
+		row, ok := result.([]interface{})
+		if !ok {
+			glog.ErrorfCtx(ctx, "Can't convert result to list")
+			return
+		}
+
+		if len(row) < 5 {
+			glog.ErrorfCtx(ctx, "Length of result is less than needed: %v", len(row))
+			return
+		}
+
+		nameRaw := row[2]
+		name, ok := nameRaw.(string)
+		if !ok {
+			glog.ErrorfCtx(ctx, "Can't convert name field to string. Actual type: %v, value: %v", reflect.TypeOf(nameRaw), nameRaw)
+			return
+		}
+
+		dataRaw := row[4]
+		data, ok := dataRaw.(string)
+		if !ok {
+			glog.ErrorfCtx(ctx, "Can't convert data field to string. Actual type: %v, value: %v", reflect.TypeOf(dataRaw), dataRaw)
+			return
+		}
+
+		entry := &filer.Entry{
+			FullPath: util.NewFullPath(string(dirPath), name),
+		}
+		lastFileName = name
+		if decodeErr := entry.DecodeAttributesAndChunks(util.MaybeDecompressData([]byte(data))); decodeErr != nil {
+			err = decodeErr
+			glog.V(0).InfofCtx(ctx, "list %s : %v", entry.FullPath, err)
+			break
+		}
+
+		resEachEntryFunc, resEachEntryFuncErr := eachEntryFunc(entry)
+		if resEachEntryFuncErr != nil {
+			err = fmt.Errorf("failed to process eachEntryFunc: %w", resEachEntryFuncErr)
+			break
+		}
+
+		if !resEachEntryFunc {
+			break
+		}
+	}
+
+	return lastFileName, err
+}
+
+func (store *TarantoolStore) Shutdown() {
+	store.pool.Close()
+}

weed/filer/tarantool/tarantool_store_kv.go

@@ -0,0 +1,98 @@
+//go:build tarantool
+
+package tarantool
+
+import (
+	"context"
+	"fmt"
+	"reflect"
+
+	"github.com/seaweedfs/seaweedfs/weed/filer"
+	"github.com/tarantool/go-option"
+	"github.com/tarantool/go-tarantool/v3/crud"
+	"github.com/tarantool/go-tarantool/v3/pool"
+)
+
+const (
+	tarantoolKVSpaceName = "key_value"
+)
+
+func (store *TarantoolStore) KvPut(ctx context.Context, key []byte, value []byte) (err error) {
+
+	var operations = []crud.Operation{
+		{
+			Operator: crud.Assign,
+			Field:    "value",
+			Value:    string(value),
+		},
+	}
+
+	req := crud.NewUpsertRequest(tarantoolKVSpaceName).
+		Tuple([]interface{}{string(key), nil, string(value)}).
+		Operations(operations).
+		Context(ctx)
+
+	ret := crud.Result{}
+	if err := store.pool.Do(req, pool.ModeRW).GetTyped(&ret); err != nil {
+		return fmt.Errorf("kv put: %w", err)
+	}
+
+	return nil
+}
+
+func (store *TarantoolStore) KvGet(ctx context.Context, key []byte) (value []byte, err error) {
+
+	getOpts := crud.GetOpts{
+		Fields:        option.SomeAny([]interface{}{"value"}),
+		Mode:          option.SomeString("read"),
+		PreferReplica: option.SomeBool(true),
+		Balance:       option.SomeBool(true),
+	}
+
+	req := crud.NewGetRequest(tarantoolKVSpaceName).
+		Key([]interface{}{string(key)}).
+		Opts(getOpts).
+		Context(ctx)
+
+	resp := crud.Result{}
+
+	err = store.pool.Do(req, pool.ModePreferRO).GetTyped(&resp)
+	if err != nil {
+		return nil, err
+	}
+
+	results, ok := resp.Rows.([]interface{})
+	if !ok || len(results) != 1 {
+		return nil, filer.ErrKvNotFound
+	}
+
+	rows, ok := results[0].([]interface{})
+	if !ok || len(rows) != 1 {
+		return nil, filer.ErrKvNotFound
+	}
+
+	row, ok := rows[0].(string)
+	if !ok {
+		return nil, fmt.Errorf("Can't convert rows[0] field to string. Actual type: %v, value: %v", reflect.TypeOf(rows[0]), rows[0])
+	}
+
+	return []byte(row), nil
+}
+
+func (store *TarantoolStore) KvDelete(ctx context.Context, key []byte) (err error) {
+
+	delOpts := crud.DeleteOpts{
+		Noreturn: option.SomeBool(true),
+	}
+
+	req := crud.NewDeleteRequest(tarantoolKVSpaceName).
+		Key([]interface{}{string(key)}).
+		Opts(delOpts).
+		Context(ctx)
+
+	if _, err := store.pool.Do(req, pool.ModeRW).Get(); err != nil {
+		return fmt.Errorf("kv delete: %w", err)
+	}
+
+	return nil
+}

weed/filer/tarantool/tarantool_store_test.go

@@ -0,0 +1,23 @@
+//go:build tarantool
+
+package tarantool
+
+import (
+	"os"
+	"testing"
+	"time"
+
+	"github.com/seaweedfs/seaweedfs/weed/filer/store_test"
+)
+
+func TestStore(t *testing.T) {
+	// run "make test_tarantool" under docker folder.
+	// to set up local env
+	if os.Getenv("RUN_TARANTOOL_TESTS") != "1" {
+		t.Skip("Tarantool tests are disabled. Set RUN_TARANTOOL_TESTS=1 to enable.")
+	}
+	store := &TarantoolStore{}
+	addresses := []string{"127.0.1:3303"}
+	store.initialize(addresses, "client", "client", 5*time.Second)
+	store_test.TestFilerStore(t, store)
+}