mirror of
https://github.com/hanzoai/s3.git
synced 2026-11-02 13:11:24 +00:00
zooqueen
dd74d3e6d6
hanzoai/s3 was an AGPL-3.0 MinIO fork whose network-copyleft blocks embedding
and reselling as a managed service. Replace it wholesale with a fork of
SeaweedFS (github.com/seaweedfs/seaweedfs, Apache-2.0, Go) which ships a
built-in S3 gateway + Filer.
- License: AGPL-3.0 -> Apache-2.0 (upstream LICENSE kept verbatim, incl.
"Copyright 2025 Chris Lu"); NOTICE added crediting SeaweedFS. No AGPL/BUSL/
SSPL/GPL or enterprise ("ee") files remain.
- Module: go.mod path set to github.com/hanzoai/s3 with a self-replace
(replace github.com/seaweedfs/seaweedfs => ./) so the ~1800 internal Go files
keep their upstream import paths with zero churn. Internal identifiers and
import paths are intentionally NOT renamed.
- CLI rebrand: the binary is built as `s3` (was `weed`); root usage, per-command
help/usage/flag strings, shell autocomplete program name, and the product name
in CLI output now read "Hanzo S3" / `s3`. Users never see `weed`.
(e.g. `s3 server`, `s3 filer`, `s3 master`, `s3 volume`, `s3 s3`, `s3 mount`.)
- Build: `make` (or `go build -o s3 ./weed`) builds the `s3` binary green;
`cmd/weed-db` and `cmd/weed-sql` also build; `go vet ./weed/command/` clean.
- README/LLM.md rewritten for Hanzo S3; AGENTS.md/CLAUDE.md symlinks preserved.
- Dropped a 33MB prebuilt Mach-O `stress` binary upstream had committed; `.gitignore`
updated to ignore the `s3` build output.
120 lines
3.4 KiB
Go
120 lines
3.4 KiB
Go
package wdclient
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/master_pb"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
"google.golang.org/grpc/status"
|
|
)
|
|
|
|
// fakeLookupServer returns Unavailable for the first N calls, then succeeds.
|
|
type fakeLookupServer struct {
|
|
master_pb.UnimplementedSeaweedServer
|
|
unavailableCount int32
|
|
callCount atomic.Int32
|
|
}
|
|
|
|
func (s *fakeLookupServer) LookupVolume(_ context.Context, req *master_pb.LookupVolumeRequest) (*master_pb.LookupVolumeResponse, error) {
|
|
n := s.callCount.Add(1)
|
|
if n <= s.unavailableCount {
|
|
return nil, status.Errorf(codes.Unavailable, "master is warming up")
|
|
}
|
|
resp := &master_pb.LookupVolumeResponse{}
|
|
for _, vid := range req.VolumeOrFileIds {
|
|
resp.VolumeIdLocations = append(resp.VolumeIdLocations, &master_pb.LookupVolumeResponse_VolumeIdLocation{
|
|
VolumeOrFileId: vid,
|
|
Locations: []*master_pb.Location{
|
|
{Url: "127.0.0.1:8080", PublicUrl: "127.0.0.1:8080"},
|
|
},
|
|
})
|
|
}
|
|
return resp, nil
|
|
}
|
|
|
|
func startFakeMasterServer(t *testing.T, srv master_pb.SeaweedServer) pb.ServerAddress {
|
|
t.Helper()
|
|
lis, err := net.Listen("tcp", "127.0.0.1:0")
|
|
if err != nil {
|
|
t.Fatalf("listen: %v", err)
|
|
}
|
|
grpcServer := grpc.NewServer()
|
|
master_pb.RegisterSeaweedServer(grpcServer, srv)
|
|
go func() { _ = grpcServer.Serve(lis) }()
|
|
t.Cleanup(grpcServer.GracefulStop)
|
|
|
|
_, port, _ := net.SplitHostPort(lis.Addr().String())
|
|
return pb.ServerAddress(fmt.Sprintf("127.0.0.1:0.%s", port))
|
|
}
|
|
|
|
func TestLookupVolumeIdsRetriesOnUnavailable(t *testing.T) {
|
|
srv := &fakeLookupServer{unavailableCount: 3}
|
|
addr := startFakeMasterServer(t, srv)
|
|
|
|
mc := NewMasterClient(
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
"", "test", "", "", "",
|
|
pb.ServerDiscovery{},
|
|
)
|
|
mc.setCurrentMaster(addr)
|
|
mc.grpcTimeout = 5 * time.Second
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
defer cancel()
|
|
|
|
provider := &masterVolumeProvider{masterClient: mc}
|
|
result, err := provider.LookupVolumeIds(ctx, []string{"1"})
|
|
|
|
if err != nil {
|
|
t.Fatalf("expected success after retries, got error: %v", err)
|
|
}
|
|
if _, ok := result["1"]; !ok {
|
|
t.Error("expected volume 1 in result")
|
|
}
|
|
if calls := srv.callCount.Load(); calls != 4 {
|
|
t.Errorf("expected 4 calls (3 unavailable + 1 success), got %d", calls)
|
|
}
|
|
}
|
|
|
|
func TestLookupVolumeIdsStopsOnContextCancel(t *testing.T) {
|
|
srv := &fakeLookupServer{unavailableCount: 1000}
|
|
addr := startFakeMasterServer(t, srv)
|
|
|
|
mc := NewMasterClient(
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
"", "test", "", "", "",
|
|
pb.ServerDiscovery{},
|
|
)
|
|
mc.setCurrentMaster(addr)
|
|
mc.grpcTimeout = 5 * time.Second
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
|
|
defer cancel()
|
|
|
|
provider := &masterVolumeProvider{masterClient: mc}
|
|
start := time.Now()
|
|
_, err := provider.LookupVolumeIds(ctx, []string{"1"})
|
|
elapsed := time.Since(start)
|
|
|
|
// Verify the error is from context deadline
|
|
if !errors.Is(err, context.DeadlineExceeded) {
|
|
t.Fatalf("expected context.DeadlineExceeded, got: %v", err)
|
|
}
|
|
// Verify the loop actually retried (not just an immediate failure)
|
|
if calls := srv.callCount.Load(); calls <= 1 {
|
|
t.Errorf("expected multiple retry attempts, got %d calls", calls)
|
|
}
|
|
if elapsed > 5*time.Second {
|
|
t.Errorf("took %v, expected to stop near context deadline of 2s", elapsed)
|
|
}
|
|
}
|