trpc.group/trpc-go/trpc-go@v1.0.3/healthcheck/options.go (about)

     1  //
     2  //
     3  // Tencent is pleased to support the open source community by making tRPC available.
     4  //
     5  // Copyright (C) 2023 THL A29 Limited, a Tencent company.
     6  // All rights reserved.
     7  //
     8  // If you have downloaded a copy of the tRPC source code from Tencent,
     9  // please note that tRPC source code is licensed under the  Apache 2.0 License,
    10  // A copy of the Apache 2.0 License is included in this file.
    11  //
    12  //
    13  
    14  package healthcheck
    15  
    16  // Opt modifies HealthCheck.
    17  type Opt func(*HealthCheck)
    18  
    19  // WithUnregisteredServiceStatus changes the default status of unregistered service to status.
    20  func WithUnregisteredServiceStatus(status Status) Opt {
    21  	return func(hc *HealthCheck) {
    22  		hc.unregisteredServiceStatus = status
    23  	}
    24  }
    25  
    26  // WithStatusWatchers returns an Option which set watchers for HealthCheck.
    27  func WithStatusWatchers(watchers map[string][]func(status Status)) Opt {
    28  	return func(hc *HealthCheck) {
    29  		hc.serviceWatchers = watchers
    30  	}
    31  }