mirror of https://github.com/minio/minio.git
				
				
				
			
		
			
				
	
	
		
			70 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			Go
		
	
	
	
			
		
		
	
	
			70 lines
		
	
	
		
			1.7 KiB
		
	
	
	
		
			Go
		
	
	
	
// Copyright (c) 2015-2023 MinIO, Inc.
 | 
						|
//
 | 
						|
// This file is part of MinIO Object Storage stack
 | 
						|
//
 | 
						|
// This program is free software: you can redistribute it and/or modify
 | 
						|
// it under the terms of the GNU Affero General Public License as published by
 | 
						|
// the Free Software Foundation, either version 3 of the License, or
 | 
						|
// (at your option) any later version.
 | 
						|
//
 | 
						|
// This program is distributed in the hope that it will be useful
 | 
						|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
 | 
						|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 | 
						|
// GNU Affero General Public License for more details.
 | 
						|
//
 | 
						|
// You should have received a copy of the GNU Affero General Public License
 | 
						|
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
 | 
						|
 | 
						|
package cmd
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"sync"
 | 
						|
 | 
						|
	"github.com/minio/madmin-go/v3"
 | 
						|
	"github.com/minio/minio/internal/pubsub"
 | 
						|
)
 | 
						|
 | 
						|
const bootstrapTraceLimit = 4 << 10
 | 
						|
 | 
						|
type bootstrapTracer struct {
 | 
						|
	mu   sync.RWMutex
 | 
						|
	info []madmin.TraceInfo
 | 
						|
}
 | 
						|
 | 
						|
var globalBootstrapTracer = &bootstrapTracer{}
 | 
						|
 | 
						|
func (bs *bootstrapTracer) Record(info madmin.TraceInfo) {
 | 
						|
	bs.mu.Lock()
 | 
						|
	defer bs.mu.Unlock()
 | 
						|
 | 
						|
	if len(bs.info) > bootstrapTraceLimit {
 | 
						|
		return
 | 
						|
	}
 | 
						|
	bs.info = append(bs.info, info)
 | 
						|
}
 | 
						|
 | 
						|
func (bs *bootstrapTracer) Events() []madmin.TraceInfo {
 | 
						|
	traceInfo := make([]madmin.TraceInfo, 0, bootstrapTraceLimit)
 | 
						|
 | 
						|
	bs.mu.RLock()
 | 
						|
	for _, i := range bs.info {
 | 
						|
		traceInfo = append(traceInfo, i)
 | 
						|
	}
 | 
						|
	bs.mu.RUnlock()
 | 
						|
 | 
						|
	return traceInfo
 | 
						|
}
 | 
						|
 | 
						|
func (bs *bootstrapTracer) Publish(ctx context.Context, trace *pubsub.PubSub[madmin.TraceInfo, madmin.TraceType]) {
 | 
						|
	for _, bsEvent := range bs.Events() {
 | 
						|
		if bsEvent.Message != "" {
 | 
						|
			select {
 | 
						|
			case <-ctx.Done():
 | 
						|
			default:
 | 
						|
				trace.Publish(bsEvent)
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 |