github.com/mysteriumnetwork/node@v0.0.0-20240516044423-365054f76801/core/connection/registry.go (about)

     1  /*
     2   * Copyright (C) 2018 The "MysteriumNetwork/node" Authors.
     3   *
     4   * This program is free software: you can redistribute it and/or modify
     5   * it under the terms of the GNU General Public License as published by
     6   * the Free Software Foundation, either version 3 of the License, or
     7   * (at your option) any later version.
     8   *
     9   * This program is distributed in the hope that it will be useful,
    10   * but WITHOUT ANY WARRANTY; without even the implied warranty of
    11   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    12   * GNU General Public License for more details.
    13   *
    14   * You should have received a copy of the GNU General Public License
    15   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
    16   */
    17  
    18  package connection
    19  
    20  // Factory represents a connection constructor
    21  type Factory func() (Connection, error)
    22  
    23  // Registry holds of all plugable connections
    24  type Registry struct {
    25  	creators map[string]Factory
    26  }
    27  
    28  // NewRegistry creates registry of plugable connections
    29  func NewRegistry() *Registry {
    30  	return &Registry{
    31  		creators: make(map[string]Factory),
    32  	}
    33  }
    34  
    35  // Register new plugable connection
    36  func (registry *Registry) Register(serviceType string, creator Factory) {
    37  	registry.creators[serviceType] = creator
    38  }
    39  
    40  // CreateConnection create plugable connection
    41  func (registry *Registry) CreateConnection(serviceType string) (Connection, error) {
    42  	factory, exists := registry.creators[serviceType]
    43  	if !exists {
    44  		return nil, ErrUnsupportedServiceType
    45  	}
    46  
    47  	return factory()
    48  }