manager/allocator/cnmallocator: add temporary adaptor for constructor

Add an adaptor to help with a signature change in the Idm constructor.

Signed-off-by: Sebastiaan van Stijn <[email protected]>

Author: thaJeztah

manager/allocator/cnmallocator/portallocator.go

@@ -3,6 +3,7 @@ package cnmallocator
 import (
 	"fmt"
 
+	"github.com/docker/docker/libnetwork/datastore"
 	"github.com/docker/docker/libnetwork/idm"
 	"github.com/moby/swarmkit/v2/api"
 )
@@ -117,16 +118,40 @@ func newPortAllocator() (*portAllocator, error) {
 	return &portAllocator{portSpaces: portSpaces}, nil
 }
 
+type (
+	// these are deliberately aliases, otherwise we'd only match the same type, not signature.
+	legacyConstructor = func(ds datastore.DataStore, id string, start, end uint64) (*idm.Idm, error)
+	idmConstructor    = func(id string, start, end uint64) (*idm.Idm, error)
+)
+
+func adaptConstructor(newFn any) (idmConstructor, error) {
+	switch fn := newFn.(type) {
+	case idmConstructor:
+		return fn, nil
+	case legacyConstructor:
+		return func(id string, start, end uint64) (*idm.Idm, error) {
+			return fn(nil, id, start, end)
+		}, nil
+	default:
+		return nil, fmt.Errorf("invalid constructor signature:%T", newFn)
+	}
+}
+
 func newPortSpace(protocol api.PortConfig_Protocol) (*portSpace, error) {
+	fn, err := adaptConstructor(idm.New)
+	if err != nil {
+		return nil, err
+	}
+
 	masterName := fmt.Sprintf("%s-master-ports", protocol)
 	dynamicName := fmt.Sprintf("%s-dynamic-ports", protocol)
 
-	master, err := idm.New(nil, masterName, masterPortStart, masterPortEnd)
+	master, err := fn(masterName, masterPortStart, masterPortEnd)
 	if err != nil {
 		return nil, err
 	}
 
-	dynamic, err := idm.New(nil, dynamicName, dynamicPortStart, dynamicPortEnd)
+	dynamic, err := fn(dynamicName, dynamicPortStart, dynamicPortEnd)
 	if err != nil {
 		return nil, err
 	}