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.
139 lines
4.5 KiB
Go
139 lines
4.5 KiB
Go
package sub_client
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/seaweedfs/seaweedfs/weed/glog"
|
|
"github.com/seaweedfs/seaweedfs/weed/mq/topic"
|
|
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
|
|
"github.com/seaweedfs/seaweedfs/weed/util"
|
|
)
|
|
|
|
type ProcessorState struct {
|
|
stopCh chan struct{}
|
|
}
|
|
|
|
// Subscribe subscribes to a topic's specified partitions.
|
|
// If a partition is moved to another broker, the subscriber will automatically reconnect to the new broker.
|
|
|
|
func (sub *TopicSubscriber) Subscribe() error {
|
|
|
|
go sub.startProcessors()
|
|
|
|
// loop forever
|
|
// TODO shutdown the subscriber when not needed anymore
|
|
sub.doKeepConnectedToSubCoordinator()
|
|
|
|
return nil
|
|
}
|
|
|
|
func (sub *TopicSubscriber) startProcessors() {
|
|
// listen to the messages from the sub coordinator
|
|
// start one processor per partition
|
|
var wg sync.WaitGroup
|
|
semaphore := make(chan struct{}, sub.SubscriberConfig.MaxPartitionCount)
|
|
|
|
for message := range sub.brokerPartitionAssignmentChan {
|
|
if assigned := message.GetAssignment(); assigned != nil {
|
|
wg.Add(1)
|
|
semaphore <- struct{}{}
|
|
|
|
topicPartition := topic.FromPbPartition(assigned.PartitionAssignment.Partition)
|
|
|
|
// wait until no covering partition is still in progress
|
|
sub.waitUntilNoOverlappingPartitionInFlight(topicPartition)
|
|
|
|
// start a processors
|
|
stopChan := make(chan struct{})
|
|
sub.activeProcessorsLock.Lock()
|
|
sub.activeProcessors[topicPartition] = &ProcessorState{
|
|
stopCh: stopChan,
|
|
}
|
|
sub.activeProcessorsLock.Unlock()
|
|
|
|
go func(assigned *mq_pb.BrokerPartitionAssignment, topicPartition topic.Partition) {
|
|
defer func() {
|
|
sub.activeProcessorsLock.Lock()
|
|
delete(sub.activeProcessors, topicPartition)
|
|
sub.activeProcessorsLock.Unlock()
|
|
|
|
<-semaphore
|
|
wg.Done()
|
|
}()
|
|
glog.V(0).Infof("subscriber %s/%s assigned partition %+v at %v", sub.ContentConfig.Topic, sub.SubscriberConfig.ConsumerGroup, assigned.Partition, assigned.LeaderBroker)
|
|
sub.brokerPartitionAssignmentAckChan <- &mq_pb.SubscriberToSubCoordinatorRequest{
|
|
Message: &mq_pb.SubscriberToSubCoordinatorRequest_AckAssignment{
|
|
AckAssignment: &mq_pb.SubscriberToSubCoordinatorRequest_AckAssignmentMessage{
|
|
Partition: assigned.Partition,
|
|
},
|
|
},
|
|
}
|
|
|
|
executors := util.NewLimitedConcurrentExecutor(int(sub.SubscriberConfig.SlidingWindowSize))
|
|
onDataMessageFn := func(m *mq_pb.SubscribeMessageResponse_Data) {
|
|
executors.Execute(func() {
|
|
if sub.OnDataMessageFunc != nil {
|
|
sub.OnDataMessageFunc(m)
|
|
}
|
|
sub.PartitionOffsetChan <- KeyedTimestamp{
|
|
Key: m.Data.Key,
|
|
TsNs: m.Data.TsNs,
|
|
}
|
|
})
|
|
}
|
|
|
|
err := sub.onEachPartition(assigned, stopChan, onDataMessageFn)
|
|
if err != nil {
|
|
glog.V(0).Infof("subscriber %s/%s partition %+v at %v: %v", sub.ContentConfig.Topic, sub.SubscriberConfig.ConsumerGroup, assigned.Partition, assigned.LeaderBroker, err)
|
|
} else {
|
|
glog.V(0).Infof("subscriber %s/%s partition %+v at %v completed", sub.ContentConfig.Topic, sub.SubscriberConfig.ConsumerGroup, assigned.Partition, assigned.LeaderBroker)
|
|
}
|
|
sub.brokerPartitionAssignmentAckChan <- &mq_pb.SubscriberToSubCoordinatorRequest{
|
|
Message: &mq_pb.SubscriberToSubCoordinatorRequest_AckUnAssignment{
|
|
AckUnAssignment: &mq_pb.SubscriberToSubCoordinatorRequest_AckUnAssignmentMessage{
|
|
Partition: assigned.Partition,
|
|
},
|
|
},
|
|
}
|
|
}(assigned.PartitionAssignment, topicPartition)
|
|
}
|
|
if unAssignment := message.GetUnAssignment(); unAssignment != nil {
|
|
topicPartition := topic.FromPbPartition(unAssignment.Partition)
|
|
sub.activeProcessorsLock.Lock()
|
|
if processor, found := sub.activeProcessors[topicPartition]; found {
|
|
close(processor.stopCh)
|
|
delete(sub.activeProcessors, topicPartition)
|
|
}
|
|
sub.activeProcessorsLock.Unlock()
|
|
}
|
|
}
|
|
|
|
wg.Wait()
|
|
|
|
}
|
|
|
|
func (sub *TopicSubscriber) waitUntilNoOverlappingPartitionInFlight(topicPartition topic.Partition) {
|
|
foundOverlapping := true
|
|
for foundOverlapping {
|
|
sub.activeProcessorsLock.Lock()
|
|
foundOverlapping = false
|
|
var overlappedPartition topic.Partition
|
|
for partition, _ := range sub.activeProcessors {
|
|
if partition.Overlaps(topicPartition) {
|
|
if partition.Equals(topicPartition) {
|
|
continue
|
|
}
|
|
foundOverlapping = true
|
|
overlappedPartition = partition
|
|
break
|
|
}
|
|
}
|
|
sub.activeProcessorsLock.Unlock()
|
|
if foundOverlapping {
|
|
glog.V(0).Infof("subscriber %s new partition %v waiting for partition %+v to complete", sub.ContentConfig.Topic, topicPartition, overlappedPartition)
|
|
time.Sleep(1 * time.Second)
|
|
}
|
|
}
|
|
}
|