kserve/pkg/logger/batch_immediate.go

/*
Copyright 2025 The KServe Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
    http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package logger

import "context"

// ImmediateBatch emits each received record as a single-record batch.
type ImmediateBatch struct{}

var _ BatchStrategy = &ImmediateBatch{}

// Run reads from in, emits each record as a single-record batch to out,
// closes out when in is closed, and respects ctx cancellation.
func (b *ImmediateBatch) Run(ctx context.Context, in <-chan LogRequest, out chan<- []LogRequest) {
	defer close(out)

	for {
		select {
		case <-ctx.Done():
			return
		case req, ok := <-in:
			if !ok {
				return
			}
			// Emit single-record batch
			select {
			case <-ctx.Done():
				return
			case out <- []LogRequest{req}:
			}
		}
	}
}