Files
s3/weed/shell/command_mq_topic_truncate.go
zooqueen dd74d3e6d6 Replace AGPL MinIO fork with Apache-2.0 SeaweedFS fork; rebrand CLI weed->s3
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.
2026-06-18 23:41:25 -07:00

141 lines
4.1 KiB
Go

package shell
import (
"context"
"flag"
"fmt"
"io"
"strings"
"github.com/seaweedfs/seaweedfs/weed/mq/topic"
"github.com/seaweedfs/seaweedfs/weed/pb/filer_pb"
"github.com/seaweedfs/seaweedfs/weed/util"
)
func init() {
Commands = append(Commands, &commandMqTopicTruncate{})
}
type commandMqTopicTruncate struct {
}
func (c *commandMqTopicTruncate) Name() string {
return "mq.topic.truncate"
}
func (c *commandMqTopicTruncate) Help() string {
return `clear all data from a topic while preserving topic structure
Example:
mq.topic.truncate -namespace <namespace> -topic <topic_name>
This command removes all log files and parquet files from all partitions
of the specified topic, while keeping the topic configuration intact.
`
}
func (c *commandMqTopicTruncate) HasTag(CommandTag) bool {
return false
}
func (c *commandMqTopicTruncate) Do(args []string, commandEnv *CommandEnv, writer io.Writer) error {
// parse parameters
mqCommand := flag.NewFlagSet(c.Name(), flag.ContinueOnError)
namespace := mqCommand.String("namespace", "", "namespace name")
topicName := mqCommand.String("topic", "", "topic name")
if err := mqCommand.Parse(args); err != nil {
return err
}
if *namespace == "" {
return fmt.Errorf("namespace is required")
}
if *topicName == "" {
return fmt.Errorf("topic name is required")
}
// Verify topic exists by trying to read its configuration
t := topic.NewTopic(*namespace, *topicName)
err := commandEnv.WithFilerClient(false, func(client filer_pb.SeaweedFilerClient) error {
_, err := t.ReadConfFile(client)
if err != nil {
return fmt.Errorf("topic %s.%s does not exist or cannot be read: %v", *namespace, *topicName, err)
}
return nil
})
if err != nil {
return err
}
fmt.Fprintf(writer, "Truncating topic %s.%s...\n", *namespace, *topicName)
// Discover and clear all partitions using centralized logic
partitions, err := t.DiscoverPartitions(context.Background(), commandEnv)
if err != nil {
return fmt.Errorf("failed to discover topic partitions: %v", err)
}
if len(partitions) == 0 {
fmt.Fprintf(writer, "No partitions found for topic %s.%s\n", *namespace, *topicName)
return nil
}
fmt.Fprintf(writer, "Found %d partitions, clearing data...\n", len(partitions))
// Clear data from each partition
totalFilesDeleted := 0
for _, partitionPath := range partitions {
filesDeleted, err := c.clearPartitionData(commandEnv, partitionPath, writer)
if err != nil {
fmt.Fprintf(writer, "Warning: failed to clear partition %s: %v\n", partitionPath, err)
continue
}
totalFilesDeleted += filesDeleted
fmt.Fprintf(writer, "Cleared partition: %s (%d files)\n", partitionPath, filesDeleted)
}
fmt.Fprintf(writer, "Successfully truncated topic %s.%s - deleted %d files from %d partitions\n",
*namespace, *topicName, totalFilesDeleted, len(partitions))
return nil
}
// clearPartitionData deletes all data files (log files, parquet files) from a partition directory
// Returns the number of files deleted
func (c *commandMqTopicTruncate) clearPartitionData(commandEnv *CommandEnv, partitionPath string, writer io.Writer) (int, error) {
filesDeleted := 0
err := filer_pb.ReadDirAllEntries(context.Background(), commandEnv, util.FullPath(partitionPath), "", func(entry *filer_pb.Entry, isLast bool) error {
if entry.IsDirectory {
return nil // Skip subdirectories
}
fileName := entry.Name
// Preserve configuration files
if strings.HasSuffix(fileName, ".conf") ||
strings.HasSuffix(fileName, ".config") ||
fileName == "topic.conf" ||
fileName == "partition.conf" {
fmt.Fprintf(writer, " Preserving config file: %s\n", fileName)
return nil
}
// Delete all data files (log files, parquet files, offset files, etc.)
deleteErr := filer_pb.Remove(context.Background(), commandEnv, partitionPath, fileName, false, true, true, false, nil)
if deleteErr != nil {
fmt.Fprintf(writer, " Warning: failed to delete %s/%s: %v\n", partitionPath, fileName, deleteErr)
// Continue with other files rather than failing entirely
} else {
fmt.Fprintf(writer, " Deleted: %s\n", fileName)
filesDeleted++
}
return nil
})
return filesDeleted, err
}