api/graph2/queries/log.go

package queries

import (
	"context"
	"errors"

	"git.aiterp.net/rpdata/api/internal/loader"
	"git.aiterp.net/rpdata/api/model/log"
)

func (r *resolver) Log(ctx context.Context, id string) (log.Log, error) {
	return log.FindID(id)
}

func (r *resolver) Logs(ctx context.Context, filter *log.Filter) ([]log.Log, error) {
	logs, err := log.List(filter)
	if err != nil {
		return nil, err
	}

	if len(logs) >= 100 {
		loader := loader.FromContext(ctx)
		if loader == nil {
			return nil, errors.New("no loader")
		}

		for _, log := range logs {
			loader.PrimeCharacters("id", log.CharacterIDs...)
			loader.PrimeChannels("name", log.ChannelName)
		}
	}

	return logs, nil
}