dubbo.apache.org/dubbo-go/v3@v3.1.1/common/extension/service_instance_customizer.go (about)

     1  /*
     2   * Licensed to the Apache Software Foundation (ASF) under one or more
     3   * contributor license agreements.  See the NOTICE file distributed with
     4   * this work for additional information regarding copyright ownership.
     5   * The ASF licenses this file to You under the Apache License, Version 2.0
     6   * (the "License"); you may not use this file except in compliance with
     7   * the License.  You may obtain a copy of the License at
     8   *
     9   *     http://www.apache.org/licenses/LICENSE-2.0
    10   *
    11   * Unless required by applicable law or agreed to in writing, software
    12   * distributed under the License is distributed on an "AS IS" BASIS,
    13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14   * See the License for the specific language governing permissions and
    15   * limitations under the License.
    16   */
    17  
    18  package extension
    19  
    20  import (
    21  	"sort"
    22  )
    23  
    24  import (
    25  	"dubbo.apache.org/dubbo-go/v3/registry"
    26  )
    27  
    28  var customizers = make([]registry.ServiceInstanceCustomizer, 0, 8)
    29  
    30  // AddCustomizers will put the customizer into slices and then sort them;
    31  // this method will be invoked several time, so we sort them here.
    32  func AddCustomizers(cus registry.ServiceInstanceCustomizer) {
    33  	customizers = append(customizers, cus)
    34  	sort.Stable(customizerSlice(customizers))
    35  }
    36  
    37  // GetCustomizers will return the sorted customizer
    38  // the result won't be nil
    39  func GetCustomizers() []registry.ServiceInstanceCustomizer {
    40  	return customizers
    41  }
    42  
    43  type customizerSlice []registry.ServiceInstanceCustomizer
    44  
    45  // nolint
    46  func (c customizerSlice) Len() int {
    47  	return len(c)
    48  }
    49  
    50  // nolint
    51  func (c customizerSlice) Swap(i, j int) { c[i], c[j] = c[j], c[i] }
    52  
    53  // nolint
    54  func (c customizerSlice) Less(i, j int) bool { return c[i].GetPriority() < c[j].GetPriority() }