github.com/cs3org/reva/v2@v2.27.7/pkg/metrics/driver/registry/registry.go (about)

     1  // Copyright 2018-2021 CERN
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  //
    15  // In applying this license, CERN does not waive the privileges and immunities
    16  // granted to it by virtue of its status as an Intergovernmental Organization
    17  // or submit itself to any jurisdiction.
    18  
    19  package registry
    20  
    21  import (
    22  	"github.com/cs3org/reva/v2/pkg/metrics/reader"
    23  )
    24  
    25  var drivers map[string]reader.Reader // map key is driver type name
    26  
    27  // Register register a driver
    28  func Register(driverName string, r reader.Reader) {
    29  	if drivers == nil {
    30  		drivers = make(map[string]reader.Reader)
    31  	}
    32  	drivers[driverName] = r
    33  }
    34  
    35  // GetDriver returns the registered driver for the specified driver name, or nil if it is not registered
    36  func GetDriver(driverName string) reader.Reader {
    37  	driver, found := drivers[driverName]
    38  	if found {
    39  		return driver
    40  	}
    41  	return nil
    42  }