batch_request_collection.go

package msgraphgocore

import (
	"context"
	"errors"
	abstractions "github.com/microsoft/kiota-abstractions-go"
)

type BatchRequestCollection struct {
	batchRequest *batchRequest
	batchLimit   int
}

const MaxBatchRequests = 4

// NewBatchRequestCollection creates an instance of a BatchRequestCollection with a default request limit
func NewBatchRequestCollection(adapter abstractions.RequestAdapter) *BatchRequestCollection {
	return NewBatchRequestCollectionWithLimit(adapter, MaxBatchRequests)
}

// NewBatchRequestCollectionWithLimit creates an instance of a BatchRequestCollection with a defined limit in requests
func NewBatchRequestCollectionWithLimit(adapter abstractions.RequestAdapter, batchLimit int) *BatchRequestCollection {
	return &BatchRequestCollection{
		batchRequest: &batchRequest{
			adapter: adapter,
		},
		batchLimit: batchLimit,
	}
}

// AddBatchRequestStep converts RequestInformation to a BatchItem and adds it to a BatchRequestCollection
func (b *BatchRequestCollection) AddBatchRequestStep(reqInfo abstractions.RequestInformation) (BatchItem, error) {
	return b.batchRequest.addLimitedBatchRequestStep(reqInfo, -1)
}

// Send serializes and sends the batch request to the server
func (b *BatchRequestCollection) Send(ctx context.Context, adapter abstractions.RequestAdapter) (BatchResponse, error) {
	// spit request with a max of 19
	requestItems := chunkSlice(b.batchRequest.requests, 19)

	if len(requestItems) > b.batchLimit {
		return nil, errors.New("exceeded max number of batch requests")
	}

	// execute requests
	response := NewBatchResponse()
	for _, requests := range requestItems {
		batch := NewBatchRequest(b.batchRequest.adapter)
		batch.SetRequests(requests)
		res, err := batch.Send(ctx, adapter)
		if err != nil {
			return nil, err
		}
		response.AddResponses(res.GetResponses())
	}

	return response, nil
}

func chunkSlice[T interface{}](slice []T, chunkSize int) [][]T {
	var chunks [][]T
	for i := 0; i < len(slice); i += chunkSize {
		end := i + chunkSize
		if end > len(slice) {
			end = len(slice)
		}

		chunks = append(chunks, slice[i:end])
	}
	return chunks
}