feat: enforce shared application resource quotas across namespaces

sdk/go/examples/application-resources/main.go

@@ -0,0 +1,101 @@
+// application-resources 演示同一应用的两个命名空间共享积压额度。
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"errors"
+	"fmt"
+	"os"
+	"time"
+
+	orbit "git.yuntuo.hk/k2safe/orbit/sdk/go"
+)
+
+func main() {
+	if err := run(); err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+}
+func client(path string) (*orbit.Client, string, error) {
+	raw, err := os.ReadFile(path)
+	if err != nil {
+		return nil, "", err
+	}
+	var v struct {
+		Token   string `json:"token"`
+		QueueID string `json:"queue_id"`
+	}
+	if err := json.Unmarshal(raw, &v); err != nil {
+		return nil, "", err
+	}
+	c, err := orbit.NewClient(os.Getenv("ORBIT_URL"), orbit.Options{Token: v.Token})
+	return c, v.QueueID, err
+}
+func run() (err error) {
+	a, qa, err := client(os.Getenv("ORBIT_CREDENTIALS_FILE"))
+	if err != nil {
+		return err
+	}
+	b, qb, err := client(os.Getenv("ORBIT_SECOND_CREDENTIALS_FILE"))
+	if err != nil {
+		return err
+	}
+	ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
+	defer cancel()
+	before, err := a.GetApplicationResources(ctx)
+	if err != nil {
+		return err
+	}
+	peer, err := b.GetApplicationResources(ctx)
+	if err != nil {
+		return err
+	}
+	if before.ApplicationID != peer.ApplicationID || qa == qb || before.UnfinishedTasks != 0 {
+		return errors.New("use two namespaces in the same idle, dedicated example application")
+	}
+	op, err := orbit.NewWorkerMessageID()
+	if err != nil {
+		return err
+	}
+	config, err := a.ConfigureApplicationResources(ctx, "config-"+op, orbit.ConfigureApplicationResourcesRequest{ExpectedRevision: before.Revision, Reason: "cross namespace resource example", MaxUnfinishedTasks: 1, MaxConcurrency: 2, RatePerSecond: 10})
+	if err != nil {
+		return err
+	}
+	defer func() {
+		cleanup, stop := context.WithTimeout(context.Background(), 8*time.Second)
+		defer stop()
+		_, restoreErr := a.ConfigureApplicationResources(cleanup, "restore-"+op, orbit.ConfigureApplicationResourcesRequest{ExpectedRevision: config.Resources.Revision, Reason: "restore example configuration", MaxUnfinishedTasks: before.MaxUnfinishedTasks, MaxConcurrency: before.MaxConcurrency, RatePerSecond: before.RatePerSecond})
+		err = errors.Join(err, restoreErr)
+	}()
+	input := orbit.SubmitTaskRequest{QueueID: qa, Handler: "object.prepare", HandlerVersion: "v1", PayloadVersion: "v1", Payload: json.RawMessage(`{"object_ref":"objects/example"}`), TaskTimeoutMS: 20000, Retry: orbit.RetryPolicy{MaxAttempts: 1, AttemptTimeoutMS: 10000}}
+	first, err := a.SubmitTask(ctx, "first-"+op, input)
+	if err != nil {
+		return err
+	}
+	input.QueueID = qb
+	_, err = b.SubmitTask(ctx, "second-"+op, input)
+	var full *orbit.APIError
+	if !errors.As(err, &full) || full.Code != "APPLICATION_FULL" {
+		return fmt.Errorf("expected APPLICATION_FULL, got %v", err)
+	}
+	fmt.Printf("cross_namespace_backpressure=%s http=%d\n", full.Code, full.StatusCode)
+	if _, err := a.CancelTask(ctx, first.Task.ID, "cancel-first-"+op, orbit.CancelTaskRequest{ExpectedRevision: 1, Reason: "free shared slot"}); err != nil {
+		return err
+	}
+	second, err := b.SubmitTask(ctx, "second-"+op, input)
+	if err != nil {
+		return err
+	}
+	fmt.Printf("same_identity_retry_accepted=%t\n", second.Task.State == orbit.TaskWaiting)
+	if _, err := b.CancelTask(ctx, second.Task.ID, "cancel-second-"+op, orbit.CancelTaskRequest{ExpectedRevision: 1, Reason: "complete example"}); err != nil {
+		return err
+	}
+	after, err := a.GetApplicationResources(ctx)
+	if err != nil {
+		return err
+	}
+	fmt.Printf("unfinished=%d active=%d pending_starts=%d\n", after.UnfinishedTasks, after.ActiveAttempts, after.PendingStarts)
+	return nil
+}