v2fly/proxy/testing/proxy.go

package testing

import (
	"fmt"

	"github.com/v2ray/v2ray-core/proxy"
	"github.com/v2ray/v2ray-core/proxy/internal"
)

var count = 0

func randomString() string {
	count++
	return fmt.Sprintf("-%d", count)
}

func RegisterInboundConnectionHandlerCreator(prefix string, creator internal.InboundConnectionHandlerCreator) (string, error) {
	for {
		name := prefix + randomString()
		err := proxy.RegisterInboundConnectionHandlerFactory(name, creator)
		if err != proxy.ErrorNameExists {
			return name, err
		}
	}
}

func RegisterOutboundConnectionHandlerCreator(prefix string, creator internal.OutboundConnectionHandlerCreator) (string, error) {
	for {
		name := prefix + randomString()
		err := proxy.RegisterOutboundConnectionHandlerFactory(name, creator)
		if err != proxy.ErrorNameExists {
			return name, err
		}
	}
}