gql.go

package tgproducer

import (
	"reflect"

	"github.com/graphql-go/graphql"
	"github.com/usnistgov/ndn-dpdk/app/tg/tggql"
	"github.com/usnistgov/ndn-dpdk/core/gqlserver"
	"github.com/usnistgov/ndn-dpdk/iface"
	"github.com/usnistgov/ndn-dpdk/ndn"
)

// GqlRetrieveByFaceID returns *Producer associated with a face.
// It is assigned during package tg initialization.
var GqlRetrieveByFaceID func(id iface.ID) *Producer

// GraphQL types.
var (
	GqlReplyInput          *graphql.InputObject
	GqlPatternInput        *graphql.InputObject
	GqlConfigInput         *graphql.InputObject
	GqlPatternCountersType *graphql.Object
	GqlCountersType        *graphql.Object
	GqlProducerType        *gqlserver.NodeType[*Producer]
)

func init() {
	GqlReplyInput = graphql.NewInputObject(graphql.InputObjectConfig{
		Name:        "TgpReplyInput",
		Description: "Traffic generator producer reply definition.",
		Fields: gqlserver.BindInputFields[Reply](gqlserver.FieldTypes{
			reflect.TypeFor[ndn.Name](): gqlserver.NonNullString,
		}),
	})
	GqlPatternInput = graphql.NewInputObject(graphql.InputObjectConfig{
		Name:        "TgpPatternInput",
		Description: "Traffic generator producer pattern definition.",
		Fields: gqlserver.BindInputFields[Pattern](gqlserver.FieldTypes{
			reflect.TypeFor[ndn.Name](): gqlserver.NonNullString,
			reflect.TypeFor[Reply]():    GqlReplyInput,
		}),
	})
	GqlConfigInput = graphql.NewInputObject(graphql.InputObjectConfig{
		Name:        "TgpConfigInput",
		Description: "Traffic generator producer config.",
		Fields: gqlserver.BindInputFields[Config](gqlserver.FieldTypes{
			reflect.TypeFor[iface.PktQueueConfig](): iface.GqlPktQueueInput,
			reflect.TypeFor[Pattern]():              GqlPatternInput,
		}),
	})

	GqlPatternCountersType = graphql.NewObject(graphql.ObjectConfig{
		Name:   "TgpPatternCounters",
		Fields: gqlserver.BindFields[PatternCounters](nil),
	})
	GqlCountersType = graphql.NewObject(graphql.ObjectConfig{
		Name: "TgpCounters",
		Fields: gqlserver.BindFields[Counters](gqlserver.FieldTypes{
			reflect.TypeFor[PatternCounters](): GqlPatternCountersType,
		}),
	})

	GqlProducerType = gqlserver.NewNodeType(graphql.ObjectConfig{
		Name:        "TgProducer",
		Description: "Traffic generator producer.",
		Fields: tggql.CommonFields(graphql.Fields{
			"patterns": &graphql.Field{
				Description: "Traffic patterns.",
				Type:        gqlserver.NonNullJSON,
				Resolve: func(p graphql.ResolveParams) (any, error) {
					return p.Source.(*Producer).Patterns(), nil
				},
			},
			"counters": &graphql.Field{
				Description: "Counters.",
				Type:        graphql.NewNonNull(GqlCountersType),
				Resolve: func(p graphql.ResolveParams) (any, error) {
					return p.Source.(*Producer).Counters(), nil
				},
			},
		}),
	}, tggql.NodeConfig(&GqlRetrieveByFaceID))
}