Add ExtendedBatchGetLatestValues

pkg/contractreader/extended.go

@@ -44,6 +44,13 @@ type Extended interface {
 		confidenceLevel primitives.ConfidenceLevel,
 		params, returnVal any,
 	) error
+	ExtendedBatchGetLatestValues(
+		ctx context.Context,
+		contractName string,
+		methodName string,
+		params []any,
+		retValType any,
+	) (types.ContractBatchResults, error)
 }
 
 type ExtendedBoundContract struct {
@@ -122,6 +129,54 @@ func (e *extendedContractReader) ExtendedGetLatestValue(
 	)
 }
 
+// ExtendedBatchGetLatestValue calls a single method of a contract once for
+// each set of parameters passed in params slice. The requests will be processed in
+// parallel, and all results returned as types.ContractBatchResults aka []BatchReadResult
+// Nothing is returned in retValType, it will determine the type of each
+// BatchReadResult.returnValue
+func (e *extendedContractReader) ExtendedBatchGetLatestValues(
+	ctx context.Context,
+	contractName string,
+	methodName string,
+	params []any,
+	retValType any,
+) (types.ContractBatchResults, error) {
+	binding, err := e.getOneBinding(contractName)
+	if err != nil {
+		return nil, fmt.Errorf("ExtendedBatchGetLatestValues: %w", err)
+	}
+
+	readIdentifier := binding.Binding.ReadIdentifier(methodName)
+
+	reqs := make(types.ContractBatch, 0, len(params))
+	for p := range params {
+		req := types.BatchRead{
+			ReadName:  readIdentifier,
+			Params:    p,
+			ReturnVal: retValType,
+		}
+		reqs = append(reqs, req)
+	}
+
+	batchReq := types.BatchGetLatestValuesRequest{
+		binding.Binding: reqs,
+	}
+
+	res, err := e.BatchGetLatestValues(
+		ctx,
+		batchReq,
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	extRes, ok := res[binding.Binding]
+	if !ok {
+		return nil, fmt.Errorf("unexpected result from BatchGetLatestValues: %v", res)
+	}
+	return extRes, nil
+}
+
 func (e *extendedContractReader) Bind(ctx context.Context, allBindings []types.BoundContract) error {
 	validBindings := slicelib.Filter(allBindings, func(b types.BoundContract) bool { return !e.bindingExists(b) })
 	if len(validBindings) == 0 {