feat: expose costs

connect/src/wg/cosmo/node/v1/node_pb.ts

@@ -838,6 +838,11 @@ export class DataSourceConfiguration extends Message<DataSourceConfiguration> {
    */
   interfaceObjects: EntityInterfaceConfiguration[] = [];
 
+  /**
+   * @generated from field: optional wg.cosmo.node.v1.CostConfiguration cost_configuration = 16;
+   */
+  costConfiguration?: CostConfiguration;
+
   constructor(data?: PartialMessage<DataSourceConfiguration>) {
     super();
     proto3.util.initPartial(data, this);
@@ -861,6 +866,7 @@ export class DataSourceConfiguration extends Message<DataSourceConfiguration> {
     { no: 13, name: "custom_events", kind: "message", T: DataSourceCustomEvents },
     { no: 14, name: "entity_interfaces", kind: "message", T: EntityInterfaceConfiguration, repeated: true },
     { no: 15, name: "interface_objects", kind: "message", T: EntityInterfaceConfiguration, repeated: true },
+    { no: 16, name: "cost_configuration", kind: "message", T: CostConfiguration, opt: true },
   ]);
 
   static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): DataSourceConfiguration {
@@ -880,6 +886,183 @@ export class DataSourceConfiguration extends Message<DataSourceConfiguration> {
   }
 }
 
+/**
+ * @generated from message wg.cosmo.node.v1.CostConfiguration
+ */
+export class CostConfiguration extends Message<CostConfiguration> {
+  /**
+   * @generated from field: repeated wg.cosmo.node.v1.FieldWeightConfiguration field_weights = 1;
+   */
+  fieldWeights: FieldWeightConfiguration[] = [];
+
+  /**
+   * @generated from field: repeated wg.cosmo.node.v1.FieldListSizeConfiguration list_sizes = 2;
+   */
+  listSizes: FieldListSizeConfiguration[] = [];
+
+  /**
+   * @generated from field: map<string, int32> type_weights = 3;
+   */
+  typeWeights: { [key: string]: number } = {};
+
+  /**
+   * @generated from field: map<string, int32> directive_argument_weights = 4;
+   */
+  directiveArgumentWeights: { [key: string]: number } = {};
+
+  constructor(data?: PartialMessage<CostConfiguration>) {
+    super();
+    proto3.util.initPartial(data, this);
+  }
+
+  static readonly runtime: typeof proto3 = proto3;
+  static readonly typeName = "wg.cosmo.node.v1.CostConfiguration";
+  static readonly fields: FieldList = proto3.util.newFieldList(() => [
+    { no: 1, name: "field_weights", kind: "message", T: FieldWeightConfiguration, repeated: true },
+    { no: 2, name: "list_sizes", kind: "message", T: FieldListSizeConfiguration, repeated: true },
+    { no: 3, name: "type_weights", kind: "map", K: 9 /* ScalarType.STRING */, V: {kind: "scalar", T: 5 /* ScalarType.INT32 */} },
+    { no: 4, name: "directive_argument_weights", kind: "map", K: 9 /* ScalarType.STRING */, V: {kind: "scalar", T: 5 /* ScalarType.INT32 */} },
+  ]);
+
+  static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): CostConfiguration {
+    return new CostConfiguration().fromBinary(bytes, options);
+  }
+
+  static fromJson(jsonValue: JsonValue, options?: Partial<JsonReadOptions>): CostConfiguration {
+    return new CostConfiguration().fromJson(jsonValue, options);
+  }
+
+  static fromJsonString(jsonString: string, options?: Partial<JsonReadOptions>): CostConfiguration {
+    return new CostConfiguration().fromJsonString(jsonString, options);
+  }
+
+  static equals(a: CostConfiguration | PlainMessage<CostConfiguration> | undefined, b: CostConfiguration | PlainMessage<CostConfiguration> | undefined): boolean {
+    return proto3.util.equals(CostConfiguration, a, b);
+  }
+}
+
+/**
+ * @generated from message wg.cosmo.node.v1.FieldWeightConfiguration
+ */
+export class FieldWeightConfiguration extends Message<FieldWeightConfiguration> {
+  /**
+   * @generated from field: string type_name = 1;
+   */
+  typeName = "";
+
+  /**
+   * @generated from field: string field_name = 2;
+   */
+  fieldName = "";
+
+  /**
+   * @generated from field: optional int32 weight = 3;
+   */
+  weight?: number;
+
+  /**
+   * @generated from field: map<string, int32> argument_weights = 4;
+   */
+  argumentWeights: { [key: string]: number } = {};
+
+  constructor(data?: PartialMessage<FieldWeightConfiguration>) {
+    super();
+    proto3.util.initPartial(data, this);
+  }
+
+  static readonly runtime: typeof proto3 = proto3;
+  static readonly typeName = "wg.cosmo.node.v1.FieldWeightConfiguration";
+  static readonly fields: FieldList = proto3.util.newFieldList(() => [
+    { no: 1, name: "type_name", kind: "scalar", T: 9 /* ScalarType.STRING */ },
+    { no: 2, name: "field_name", kind: "scalar", T: 9 /* ScalarType.STRING */ },
+    { no: 3, name: "weight", kind: "scalar", T: 5 /* ScalarType.INT32 */, opt: true },
+    { no: 4, name: "argument_weights", kind: "map", K: 9 /* ScalarType.STRING */, V: {kind: "scalar", T: 5 /* ScalarType.INT32 */} },
+  ]);
+
+  static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): FieldWeightConfiguration {
+    return new FieldWeightConfiguration().fromBinary(bytes, options);
+  }
+
+  static fromJson(jsonValue: JsonValue, options?: Partial<JsonReadOptions>): FieldWeightConfiguration {
+    return new FieldWeightConfiguration().fromJson(jsonValue, options);
+  }
+
+  static fromJsonString(jsonString: string, options?: Partial<JsonReadOptions>): FieldWeightConfiguration {
+    return new FieldWeightConfiguration().fromJsonString(jsonString, options);
+  }
+
+  static equals(a: FieldWeightConfiguration | PlainMessage<FieldWeightConfiguration> | undefined, b: FieldWeightConfiguration | PlainMessage<FieldWeightConfiguration> | undefined): boolean {
+    return proto3.util.equals(FieldWeightConfiguration, a, b);
+  }
+}
+
+/**
+ * @generated from message wg.cosmo.node.v1.FieldListSizeConfiguration
+ */
+export class FieldListSizeConfiguration extends Message<FieldListSizeConfiguration> {
+  /**
+   * @generated from field: string type_name = 1;
+   */
+  typeName = "";
+
+  /**
+   * @generated from field: string field_name = 2;
+   */
+  fieldName = "";
+
+  /**
+   * @generated from field: optional int32 assumed_size = 3;
+   */
+  assumedSize?: number;
+
+  /**
+   * @generated from field: repeated string slicing_arguments = 4;
+   */
+  slicingArguments: string[] = [];
+
+  /**
+   * @generated from field: repeated string sized_fields = 5;
+   */
+  sizedFields: string[] = [];
+
+  /**
+   * @generated from field: optional bool require_one_slicing_argument = 6;
+   */
+  requireOneSlicingArgument?: boolean;
+
+  constructor(data?: PartialMessage<FieldListSizeConfiguration>) {
+    super();
+    proto3.util.initPartial(data, this);
+  }
+
+  static readonly runtime: typeof proto3 = proto3;
+  static readonly typeName = "wg.cosmo.node.v1.FieldListSizeConfiguration";
+  static readonly fields: FieldList = proto3.util.newFieldList(() => [
+    { no: 1, name: "type_name", kind: "scalar", T: 9 /* ScalarType.STRING */ },
+    { no: 2, name: "field_name", kind: "scalar", T: 9 /* ScalarType.STRING */ },
+    { no: 3, name: "assumed_size", kind: "scalar", T: 5 /* ScalarType.INT32 */, opt: true },
+    { no: 4, name: "slicing_arguments", kind: "scalar", T: 9 /* ScalarType.STRING */, repeated: true },
+    { no: 5, name: "sized_fields", kind: "scalar", T: 9 /* ScalarType.STRING */, repeated: true },
+    { no: 6, name: "require_one_slicing_argument", kind: "scalar", T: 8 /* ScalarType.BOOL */, opt: true },
+  ]);
+
+  static fromBinary(bytes: Uint8Array, options?: Partial<BinaryReadOptions>): FieldListSizeConfiguration {
+    return new FieldListSizeConfiguration().fromBinary(bytes, options);
+  }
+
+  static fromJson(jsonValue: JsonValue, options?: Partial<JsonReadOptions>): FieldListSizeConfiguration {
+    return new FieldListSizeConfiguration().fromJson(jsonValue, options);
+  }
+
+  static fromJsonString(jsonString: string, options?: Partial<JsonReadOptions>): FieldListSizeConfiguration {
+    return new FieldListSizeConfiguration().fromJsonString(jsonString, options);
+  }
+
+  static equals(a: FieldListSizeConfiguration | PlainMessage<FieldListSizeConfiguration> | undefined, b: FieldListSizeConfiguration | PlainMessage<FieldListSizeConfiguration> | undefined): boolean {
+    return proto3.util.equals(FieldListSizeConfiguration, a, b);
+  }
+}
+
 /**
  * @generated from message wg.cosmo.node.v1.ArgumentConfiguration
  */

demo/pkg/subgraphs/employees/subgraph/schema.graphqls

@@ -25,11 +25,13 @@ directive @goField(
 ) on INPUT_FIELD_DEFINITION | FIELD_DEFINITION
 
 directive @openfed__requireFetchReasons repeatable on FIELD_DEFINITION | INTERFACE | OBJECT
+directive @cost(weight: Int!) on ARGUMENT_DEFINITION | ENUM | FIELD_DEFINITION | INPUT_FIELD_DEFINITION | OBJECT | SCALAR
+directive @listSize( assumedSize: Int, slicingArguments: [String!], sizedFields: [String!], requireOneSlicingArgument: Boolean = true ) on FIELD_DEFINITION
 
 type Query {
-  employee(id: Int!): Employee @openfed__requireFetchReasons
+  employee(id: Int! @cost(weight: 2)): Employee @cost(weight: 5) @openfed__requireFetchReasons
   employeeAsList(id: Int!): [Employee]
-  employees: [Employee]
+  employees: [Employee] @listSize(assumedSize: 50)
   products: [Products!]!
   teammates(team: Department!): [Employee!]!
   firstEmployee: Employee! @tag(name: "internal")
@@ -65,7 +67,7 @@ type Subscription {
   countFor(count: Int!): Int!
 }
 
-enum Department {
+enum Department @cost(weight: 1) {
   ENGINEERING
   MARKETING
   OPERATIONS
@@ -179,7 +181,7 @@ type Consultancy @key(fields: "upc") {
   isLeadAvailable: Boolean @requires(fields: "lead { isAvailable }")
 }
 
-type Cosmo implements IProduct @key(fields: "upc") {
+type Cosmo implements IProduct @key(fields: "upc") @cost(weight: 5) {
   upc: ID!
   engineers: [Employee!]!
   lead: Employee!

demo/pkg/subgraphs/products/subgraph/schema.graphqls

@@ -1,16 +1,19 @@
 extend schema
 @link(url: "https://specs.apollo.dev/federation/v2.5", import: ["@authenticated", "@composeDirective", "@external", "@extends", "@inaccessible", "@interfaceObject", "@override", "@provides", "@key", "@requires", "@requiresScopes", "@shareable", "@tag"])
 
+directive @cost(weight: Int!) on ARGUMENT_DEFINITION | ENUM | FIELD_DEFINITION | INPUT_FIELD_DEFINITION | OBJECT | SCALAR
+directive @listSize( assumedSize: Int, slicingArguments: [String!], sizedFields: [String!], requireOneSlicingArgument: Boolean = true ) on FIELD_DEFINITION
+
 schema {
   query: Queries
   mutation: Mutation
 }
 
 type Queries {
-  productTypes: [Products!]!
+  productTypes: [Products!]! @listSize(assumedSize: 50)
   topSecretFederationFacts: [TopSecretFact!]! @requiresScopes(scopes: [["read:fact"], ["read:all"]])
   factTypes: [TopSecretFactType!]
-  sharedThings(numOfA: Int! numOfB: Int!): [Thing!]! @shareable
+  sharedThings(numOfA: Int! numOfB: Int!): [Thing!]! @listSize(slicingArguments: ["numOfA"]) @shareable
 }
 
 type Mutation {
@@ -38,7 +41,7 @@ interface TopSecretFact @authenticated {
   factType: TopSecretFactType
 }
 
-scalar FactContent @requiresScopes(scopes: [["read:scalar"], ["read:all"]])
+scalar FactContent @cost(weight: 10) @requiresScopes(scopes: [["read:scalar"], ["read:all"]])
 
 type DirectiveFact implements TopSecretFact @authenticated {
   title: String!
@@ -81,7 +84,7 @@ type Consultancy @key(fields: "upc") {
   name: ProductName!
 }
 
-type Cosmo @key(fields: "upc") {
+type Cosmo @key(fields: "upc") @cost(weight: 8) {
   upc: ID!
   name: ProductName!
   repositoryURL: String!

proto/wg/cosmo/node/v1/node.proto

@@ -90,6 +90,30 @@ message DataSourceConfiguration {
   DataSourceCustomEvents custom_events = 13;
   repeated EntityInterfaceConfiguration entity_interfaces = 14;
   repeated EntityInterfaceConfiguration interface_objects = 15;
+  optional CostConfiguration cost_configuration = 16;
+}
+
+message CostConfiguration {
+  repeated FieldWeightConfiguration field_weights = 1;
+  repeated FieldListSizeConfiguration list_sizes = 2;
+  map<string, int32> type_weights = 3;
+  map<string, int32> directive_argument_weights = 4;
+}
+
+message FieldWeightConfiguration {
+  string type_name = 1;
+  string field_name = 2;
+  optional int32 weight = 3;
+  map<string, int32> argument_weights = 4;
+}
+
+message FieldListSizeConfiguration {
+  string type_name = 1;
+  string field_name = 2;
+  optional int32 assumed_size = 3;
+  repeated string slicing_arguments = 4;
+  repeated string sized_fields = 5;
+  optional bool require_one_slicing_argument = 6;
 }
 
 message ArgumentConfiguration {