debug: add log for debug discovery

Author: YangruiEmma

client/middlewares.go

@@ -95,6 +95,7 @@ func newResolveMWBuilder(lbf *lbcache.BalancerFactory) endpoint.MiddlewareBuilde
 				if remote.GetInstance() != nil {
 					return next(ctx, request, response)
 				}
+				klog.CtxErrorf(ctx, "KITEX-DEBUG: start get balancer")
 				lb, err := lbf.Get(ctx, dest)
 				if err != nil {
 					return kerrors.ErrServiceDiscovery.WithCause(err)
@@ -108,10 +109,13 @@ func newResolveMWBuilder(lbf *lbcache.BalancerFactory) endpoint.MiddlewareBuilde
 					default:
 					}
 
+					klog.CtxErrorf(ctx, "KITEX-DEBUG: start get picker")
 					// we always need to get a new picker every time, because when downstream update deployment,
 					// we may get an old picker that include all outdated instances which will cause connect always failed.
 					picker := lb.GetPicker()
+					klog.CtxErrorf(ctx, "KITEX-DEBUG: start get instance")
 					ins := picker.Next(ctx, request)
+					klog.CtxErrorf(ctx, "KITEX-DEBUG: end get instance")
 					if ins == nil {
 						err = kerrors.ErrNoMoreInstance.WithCause(fmt.Errorf("last error: %w", lastErr))
 					} else {

pkg/loadbalance/lbcache/cache.go

@@ -163,7 +163,9 @@ func (b *BalancerFactory) Get(ctx context.Context, target rpcinfo.EndpointInfo)
 		return val.(*Balancer), nil
 	}
 	val, err, _ := b.sfg.Do(desc, func() (interface{}, error) {
+		klog.CtxErrorf(ctx, "KITEX-DEBUG: start Resolve")
 		res, err := b.resolver.Resolve(ctx, desc)
+		klog.CtxErrorf(ctx, "KITEX-DEBUG: end Resolve")
 		if err != nil {
 			return nil, err
 		}
@@ -173,7 +175,9 @@ func (b *BalancerFactory) Get(ctx context.Context, target rpcinfo.EndpointInfo)
 			target: desc,
 		}
 		bl.res.Store(res)
+		klog.CtxErrorf(ctx, "KITEX-DEBUG: start getSharedTicker")
 		bl.sharedTicker = getSharedTicker(bl, b.opts.RefreshInterval)
+		klog.CtxErrorf(ctx, "KITEX-DEBUG: end getSharedTicker")
 		b.cache.Store(desc, bl)
 		return bl, nil
 	})

pkg/loadbalance/weighted_balancer.go

@@ -63,7 +63,9 @@ func (wb *weightedBalancer) GetPicker(e discovery.Result) Picker {
 	picker, ok := wb.pickerCache.Load(e.CacheKey)
 	if !ok {
 		picker, _, _ = wb.sfg.Do(e.CacheKey, func() (interface{}, error) {
+			klog.Error("KITEX-DEBUG: start createPicker")
 			p := wb.createPicker(e)
+			klog.Error("KITEX-DEBUG: end createPicker")
 			wb.pickerCache.Store(e.CacheKey, p)
 			return p, nil
 		})