[PRODCRE-792] Per-method onchain config for capabilities V2

Author: bolekk

pkg/capabilities/capabilities.go

@@ -493,6 +493,12 @@ type RemoteExecutableConfig struct {
 	RequestHashExcludedAttributes []string
 	RegistrationRefresh           time.Duration
 	RegistrationExpiry            time.Duration
+
+	Schedule                  string
+	DeltaStage                time.Duration
+	RequestTimeout            time.Duration
+	ServerMaxParallelRequests int
+	RequestHasherType         string
 }
 
 // NOTE: consider splitting this config into values stored in Registry (KS-118)
@@ -541,4 +547,12 @@ type CapabilityConfiguration struct {
 	RemoteTriggerConfig    *RemoteTriggerConfig
 	RemoteTargetConfig     *RemoteTargetConfig
 	RemoteExecutableConfig *RemoteExecutableConfig
+
+	// v2 / "NoDAG" capabilities
+	CapabilityMethodConfig map[string]CapabilityMethodConfig
+}
+
+type CapabilityMethodConfig struct {
+	RemoteTriggerConfig    *RemoteTriggerConfig
+	RemoteExecutableConfig *RemoteExecutableConfig
 }

pkg/capabilities/pb/capabilities.pb.go

@@ -16,20 +16,48 @@ message RemoteTriggerConfig {
   google.protobuf.Duration batchCollectionPeriod = 6;
 }
 
+// deprecated - v1 only
 message RemoteTargetConfig {
   // A collection of dot seperated paths to attributes that should be excluded from the request sent to the remote target
   // when calculating the hash of the request.  This is useful for excluding attributes that are not deterministic to ensure
   // that the hash of logically identical requests is consistent.
   repeated string requestHashExcludedAttributes = 1;
 }
 
+enum TransmissionSchedule {
+  AllAtOnce = 0;
+  OneAtATime = 1;
+}
+
+enum RequestHasherType {
+  Simple = 0;
+  WriteReportExcludeSignatures = 1;
+}
+
 message RemoteExecutableConfig {
   // A collection of dot seperated paths to attributes that should be excluded from the request sent to the remote executable capability
   // when calculating the hash of the request.  This is useful for excluding attributes that are not deterministic to ensure
   // that the hash of logically identical requests is consistent.
-  repeated string requestHashExcludedAttributes = 1;
-  google.protobuf.Duration registrationRefresh = 2;
-  google.protobuf.Duration registrationExpiry = 3;
+  repeated string requestHashExcludedAttributes = 1; // deprecated - v1 only
+  google.protobuf.Duration registrationRefresh = 2;  // unused
+  google.protobuf.Duration registrationExpiry = 3;  // unused
+
+  TransmissionSchedule transmission_schedule = 4;
+  google.protobuf.Duration delta_stage = 5;
+  google.protobuf.Duration request_timeout = 6;
+  RequestHasherType request_hasher_type = 7;
+}
+
+message AggregatorConfig {
+  enum AggregatorType aggregator_type = 1;
+}
+
+message CapabilityMethodConfig {
+  oneof remote_config {
+    RemoteTriggerConfig remote_trigger_config = 1;
+    RemoteExecutableConfig remote_transmission_config = 2;
+  }
+  AggregatorConfig aggregator_config = 3;
 }
 
 message CapabilityConfig {
@@ -43,5 +71,8 @@ message CapabilityConfig {
 
   values.v1.Map restricted_config = 5;
   repeated string restricted_keys = 6;
+
+  // v2 ("NoDAG") capabilities
+  map<string, CapabilityMethodConfig> method_configs = 7;
 }