plc4go/spi/default/defaultExpectation.go

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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
 *
 *   https://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 _default

import (
	"context"
	"fmt"
	"time"

	"github.com/google/uuid"

	"github.com/apache/plc4x/plc4go/spi"
)

type defaultExpectation struct {
	Uuid            uuid.UUID
	interactionInfo string
	Ctx             context.Context
	CancelFunc      context.CancelCauseFunc
	CreationTime    time.Time
	Expiration      time.Time
	AcceptsMessage  spi.AcceptsMessage
	HandleMessage   spi.HandleMessage
	HandleError     spi.HandleError
}

func newDefaultExpectation(ctx context.Context, interactionInfo string, ttl time.Duration, acceptsMessage spi.AcceptsMessage, handleMessage spi.HandleMessage, handleError spi.HandleError) *defaultExpectation {
	ctx, cancelFunc := context.WithCancelCause(ctx)
	return &defaultExpectation{
		Uuid:            uuid.New(),
		interactionInfo: interactionInfo,
		Ctx:             ctx,
		CancelFunc:      cancelFunc,
		CreationTime:    time.Now(),
		Expiration:      time.Now().Add(ttl),
		AcceptsMessage:  acceptsMessage,
		HandleMessage:   handleMessage,
		HandleError:     handleError,
	}
}

func (d *defaultExpectation) GetContext() context.Context {
	return d.Ctx
}

func (d *defaultExpectation) Cancel(cause error) {
	d.CancelFunc(cause)
}

func (d *defaultExpectation) GetCreationTime() time.Time {
	return d.CreationTime
}

func (d *defaultExpectation) GetExpiration() time.Time {
	return d.Expiration
}

func (d *defaultExpectation) GetAcceptsMessage() spi.AcceptsMessage {
	return d.AcceptsMessage
}

func (d *defaultExpectation) GetHandleMessage() spi.HandleMessage {
	return d.HandleMessage
}

func (d *defaultExpectation) GetHandleError() spi.HandleError {
	return d.HandleError
}

func (d *defaultExpectation) String() string {
	return fmt.Sprintf("Expectation '%s' %s (expires at %v in %s)", d.interactionInfo, d.Uuid, d.Expiration, time.Until(d.Expiration))
}