Add ArbOwner impl for resource constraints

Author: MishkaRogachev

arbos/constraints/constraints.go

@@ -23,6 +23,8 @@ type ResourceSet struct {
 	weights [multigas.NumResourceKind]ResourceWeight
 }
 
+const MaxResourceWeight = 1_000_000
+
 // EmptyResourceSet creates a new set with all weights initialized to zero.
 func EmptyResourceSet() ResourceSet {
 	return ResourceSet{

arbos/constraints/storage.go

@@ -129,3 +129,23 @@ func (src *StorageResourceConstraints) Write(rc *ResourceConstraints) error {
 	}
 	return src.storage.Set(data)
 }
+
+// SetConstraint adds or updates a resource constraint in storage.
+func (src *StorageResourceConstraints) SetConstraint(resources ResourceSet, periodSecs uint32, targetPerSec uint64) error {
+	rc, err := src.Load()
+	if err != nil {
+		return err
+	}
+	rc.Set(resources, PeriodSecs(periodSecs), targetPerSec)
+	return src.Write(rc)
+}
+
+// ClearConstraint removes a resource constraint from storage.
+func (src *StorageResourceConstraints) ClearConstraint(resources ResourceSet, periodSecs uint32) error {
+	rc, err := src.Load()
+	if err != nil {
+		return err
+	}
+	rc.Clear(resources, PeriodSecs(periodSecs))
+	return src.Write(rc)
+}

contracts

@@ -10,9 +10,11 @@ import (
 	"fmt"
 	"math/big"
 
+	"github.com/ethereum/go-ethereum/arbitrum/multigas"
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/params"
 
+	"github.com/offchainlabs/nitro/arbos/constraints"
 	"github.com/offchainlabs/nitro/arbos/l1pricing"
 	"github.com/offchainlabs/nitro/arbos/programs"
 	"github.com/offchainlabs/nitro/util/arbmath"
@@ -453,3 +455,43 @@ func (con ArbOwner) SetChainConfig(c ctx, evm mech, serializedChainConfig []byte
 func (con ArbOwner) SetCalldataPriceIncrease(c ctx, _ mech, enable bool) error {
 	return c.State.Features().SetCalldataPriceIncrease(enable)
 }
+
+type resourceWeight struct {
+	ResourceKind multigas.ResourceKind
+	Weight       uint64
+}
+
+func parseResourceWeights(resources []resourceWeight) (constraints.ResourceSet, error) {
+	rs := constraints.EmptyResourceSet()
+	if len(resources) == 0 {
+		return constraints.ResourceSet{}, fmt.Errorf("at least one resource is required")
+	}
+	for _, r := range resources {
+		if r.Weight == 0 || r.Weight > constraints.MaxResourceWeight {
+			return constraints.ResourceSet{}, fmt.Errorf("resource weight for kind %d must be in range [1, %d]", r.ResourceKind, constraints.MaxResourceWeight)
+		}
+		if rs.HasResource(r.ResourceKind) {
+			return constraints.ResourceSet{}, fmt.Errorf("duplicate resource kind %d", r.ResourceKind)
+		}
+		rs = rs.WithResource(r.ResourceKind, constraints.ResourceWeight(r.Weight))
+	}
+	return rs, nil
+}
+
+// SetResourceConstraint adds or updates a resource constraint with specified resource kinds and weights
+func (con ArbOwner) SetResourceConstraint(c ctx, evm mech, resources []resourceWeight, periodSecs uint32, targetPerSec uint64) error {
+	res, err := parseResourceWeights(resources)
+	if err != nil {
+		return err
+	}
+	return c.State.ResourceConstraints().SetConstraint(res, periodSecs, targetPerSec)
+}
+
+// ClearConstraint removes a resource constraint
+func (con ArbOwner) ClearConstraint(c ctx, evm mech, resources []resourceWeight, periodSecs uint32) error {
+	res, err := parseResourceWeights(resources)
+	if err != nil {
+		return err
+	}
+	return c.State.ResourceConstraints().ClearConstraint(res, periodSecs)
+}