mirror of
				https://github.com/minio/minio.git
				synced 2025-11-04 10:11:09 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			270 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			270 lines
		
	
	
		
			8.5 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
// Copyright (c) 2015-2021 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"
 | 
						|
	"encoding/json"
 | 
						|
	"fmt"
 | 
						|
	"io"
 | 
						|
	"net/http"
 | 
						|
	"net/url"
 | 
						|
	"reflect"
 | 
						|
	"runtime"
 | 
						|
	"time"
 | 
						|
 | 
						|
	"github.com/gorilla/mux"
 | 
						|
	"github.com/minio/minio-go/v7/pkg/set"
 | 
						|
	xhttp "github.com/minio/minio/internal/http"
 | 
						|
	"github.com/minio/minio/internal/logger"
 | 
						|
	"github.com/minio/minio/internal/rest"
 | 
						|
	"github.com/minio/pkg/env"
 | 
						|
)
 | 
						|
 | 
						|
const (
 | 
						|
	bootstrapRESTVersion       = "v1"
 | 
						|
	bootstrapRESTVersionPrefix = SlashSeparator + bootstrapRESTVersion
 | 
						|
	bootstrapRESTPrefix        = minioReservedBucketPath + "/bootstrap"
 | 
						|
	bootstrapRESTPath          = bootstrapRESTPrefix + bootstrapRESTVersionPrefix
 | 
						|
)
 | 
						|
 | 
						|
const (
 | 
						|
	bootstrapRESTMethodHealth = "/health"
 | 
						|
	bootstrapRESTMethodVerify = "/verify"
 | 
						|
)
 | 
						|
 | 
						|
// To abstract a node over network.
 | 
						|
type bootstrapRESTServer struct{}
 | 
						|
 | 
						|
// ServerSystemConfig - captures information about server configuration.
 | 
						|
type ServerSystemConfig struct {
 | 
						|
	MinioPlatform  string
 | 
						|
	MinioEndpoints EndpointServerPools
 | 
						|
	MinioEnv       map[string]string
 | 
						|
}
 | 
						|
 | 
						|
// Diff - returns error on first difference found in two configs.
 | 
						|
func (s1 ServerSystemConfig) Diff(s2 ServerSystemConfig) error {
 | 
						|
	if s1.MinioPlatform != s2.MinioPlatform {
 | 
						|
		return fmt.Errorf("Expected platform '%s', found to be running '%s'",
 | 
						|
			s1.MinioPlatform, s2.MinioPlatform)
 | 
						|
	}
 | 
						|
	if s1.MinioEndpoints.NEndpoints() != s2.MinioEndpoints.NEndpoints() {
 | 
						|
		return fmt.Errorf("Expected number of endpoints %d, seen %d", s1.MinioEndpoints.NEndpoints(),
 | 
						|
			s2.MinioEndpoints.NEndpoints())
 | 
						|
	}
 | 
						|
 | 
						|
	for i, ep := range s1.MinioEndpoints {
 | 
						|
		if ep.SetCount != s2.MinioEndpoints[i].SetCount {
 | 
						|
			return fmt.Errorf("Expected set count %d, seen %d", ep.SetCount,
 | 
						|
				s2.MinioEndpoints[i].SetCount)
 | 
						|
		}
 | 
						|
		if ep.DrivesPerSet != s2.MinioEndpoints[i].DrivesPerSet {
 | 
						|
			return fmt.Errorf("Expected drives pet set %d, seen %d", ep.DrivesPerSet,
 | 
						|
				s2.MinioEndpoints[i].DrivesPerSet)
 | 
						|
		}
 | 
						|
		for j, endpoint := range ep.Endpoints {
 | 
						|
			if endpoint.String() != s2.MinioEndpoints[i].Endpoints[j].String() {
 | 
						|
				return fmt.Errorf("Expected endpoint %s, seen %s", endpoint,
 | 
						|
					s2.MinioEndpoints[i].Endpoints[j])
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if !reflect.DeepEqual(s1.MinioEnv, s2.MinioEnv) {
 | 
						|
		var missing []string
 | 
						|
		var mismatching []string
 | 
						|
		for k, v := range s1.MinioEnv {
 | 
						|
			ev, ok := s2.MinioEnv[k]
 | 
						|
			if !ok {
 | 
						|
				missing = append(missing, k)
 | 
						|
			} else if v != ev {
 | 
						|
				mismatching = append(mismatching, k)
 | 
						|
			}
 | 
						|
		}
 | 
						|
		if len(mismatching) > 0 {
 | 
						|
			return fmt.Errorf(`Expected same MINIO_ environment variables and values across all servers: Missing environment values: %s / Mismatch environment values: %s`, missing, mismatching)
 | 
						|
		}
 | 
						|
		return fmt.Errorf(`Expected same MINIO_ environment variables and values across all servers: Missing environment values: %s`, missing)
 | 
						|
	}
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
var skipEnvs = map[string]struct{}{
 | 
						|
	"MINIO_OPTS":         {},
 | 
						|
	"MINIO_CERT_PASSWD":  {},
 | 
						|
	"MINIO_SERVER_DEBUG": {},
 | 
						|
	"MINIO_DSYNC_TRACE":  {},
 | 
						|
}
 | 
						|
 | 
						|
func getServerSystemCfg() ServerSystemConfig {
 | 
						|
	envs := env.List("MINIO_")
 | 
						|
	envValues := make(map[string]string, len(envs))
 | 
						|
	for _, envK := range envs {
 | 
						|
		// skip certain environment variables as part
 | 
						|
		// of the whitelist and could be configured
 | 
						|
		// differently on each nodes, update skipEnvs()
 | 
						|
		// map if there are such environment values
 | 
						|
		if _, ok := skipEnvs[envK]; ok {
 | 
						|
			continue
 | 
						|
		}
 | 
						|
		envValues[envK] = env.Get(envK, "")
 | 
						|
	}
 | 
						|
	return ServerSystemConfig{
 | 
						|
		MinioPlatform:  fmt.Sprintf("OS: %s | Arch: %s", runtime.GOOS, runtime.GOARCH),
 | 
						|
		MinioEndpoints: globalEndpoints,
 | 
						|
		MinioEnv:       envValues,
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
// HealthHandler returns success if request is valid
 | 
						|
func (b *bootstrapRESTServer) HealthHandler(w http.ResponseWriter, r *http.Request) {}
 | 
						|
 | 
						|
func (b *bootstrapRESTServer) VerifyHandler(w http.ResponseWriter, r *http.Request) {
 | 
						|
	ctx := newContext(r, w, "VerifyHandler")
 | 
						|
	cfg := getServerSystemCfg()
 | 
						|
	logger.LogIf(ctx, json.NewEncoder(w).Encode(&cfg))
 | 
						|
}
 | 
						|
 | 
						|
// registerBootstrapRESTHandlers - register bootstrap rest router.
 | 
						|
func registerBootstrapRESTHandlers(router *mux.Router) {
 | 
						|
	server := &bootstrapRESTServer{}
 | 
						|
	subrouter := router.PathPrefix(bootstrapRESTPrefix).Subrouter()
 | 
						|
 | 
						|
	subrouter.Methods(http.MethodPost).Path(bootstrapRESTVersionPrefix + bootstrapRESTMethodHealth).HandlerFunc(
 | 
						|
		httpTraceHdrs(server.HealthHandler))
 | 
						|
 | 
						|
	subrouter.Methods(http.MethodPost).Path(bootstrapRESTVersionPrefix + bootstrapRESTMethodVerify).HandlerFunc(
 | 
						|
		httpTraceHdrs(server.VerifyHandler))
 | 
						|
}
 | 
						|
 | 
						|
// client to talk to bootstrap NEndpoints.
 | 
						|
type bootstrapRESTClient struct {
 | 
						|
	endpoint   Endpoint
 | 
						|
	restClient *rest.Client
 | 
						|
}
 | 
						|
 | 
						|
// Wrapper to restClient.Call to handle network errors, in case of network error the connection is marked disconnected
 | 
						|
// permanently. The only way to restore the connection is at the xl-sets layer by xlsets.monitorAndConnectEndpoints()
 | 
						|
// after verifying format.json
 | 
						|
func (client *bootstrapRESTClient) callWithContext(ctx context.Context, method string, values url.Values, body io.Reader, length int64) (respBody io.ReadCloser, err error) {
 | 
						|
	if values == nil {
 | 
						|
		values = make(url.Values)
 | 
						|
	}
 | 
						|
 | 
						|
	respBody, err = client.restClient.Call(ctx, method, values, body, length)
 | 
						|
	if err == nil {
 | 
						|
		return respBody, nil
 | 
						|
	}
 | 
						|
 | 
						|
	return nil, err
 | 
						|
}
 | 
						|
 | 
						|
// Stringer provides a canonicalized representation of node.
 | 
						|
func (client *bootstrapRESTClient) String() string {
 | 
						|
	return client.endpoint.String()
 | 
						|
}
 | 
						|
 | 
						|
// Verify - fetches system server config.
 | 
						|
func (client *bootstrapRESTClient) Verify(ctx context.Context, srcCfg ServerSystemConfig) (err error) {
 | 
						|
	if newObjectLayerFn() != nil {
 | 
						|
		return nil
 | 
						|
	}
 | 
						|
	respBody, err := client.callWithContext(ctx, bootstrapRESTMethodVerify, nil, nil, -1)
 | 
						|
	if err != nil {
 | 
						|
		return
 | 
						|
	}
 | 
						|
	defer xhttp.DrainBody(respBody)
 | 
						|
	recvCfg := ServerSystemConfig{}
 | 
						|
	if err = json.NewDecoder(respBody).Decode(&recvCfg); err != nil {
 | 
						|
		return err
 | 
						|
	}
 | 
						|
	return srcCfg.Diff(recvCfg)
 | 
						|
}
 | 
						|
 | 
						|
func verifyServerSystemConfig(ctx context.Context, endpointServerPools EndpointServerPools) error {
 | 
						|
	srcCfg := getServerSystemCfg()
 | 
						|
	clnts := newBootstrapRESTClients(endpointServerPools)
 | 
						|
	var onlineServers int
 | 
						|
	var offlineEndpoints []string
 | 
						|
	var retries int
 | 
						|
	for onlineServers < len(clnts)/2 {
 | 
						|
		for _, clnt := range clnts {
 | 
						|
			if err := clnt.Verify(ctx, srcCfg); err != nil {
 | 
						|
				if !isNetworkError(err) {
 | 
						|
					logger.Info(fmt.Errorf("%s has incorrect configuration: %w", clnt.String(), err).Error())
 | 
						|
				}
 | 
						|
				offlineEndpoints = append(offlineEndpoints, clnt.String())
 | 
						|
				continue
 | 
						|
			}
 | 
						|
			onlineServers++
 | 
						|
		}
 | 
						|
		select {
 | 
						|
		case <-ctx.Done():
 | 
						|
			return ctx.Err()
 | 
						|
		default:
 | 
						|
			// Sleep for a while - so that we don't go into
 | 
						|
			// 100% CPU when half the endpoints are offline.
 | 
						|
			time.Sleep(100 * time.Millisecond)
 | 
						|
			retries++
 | 
						|
			// after 5 retries start logging that servers are not reachable yet
 | 
						|
			if retries >= 5 {
 | 
						|
				logger.Info(fmt.Sprintf("Waiting for atleast %d remote servers to be online for bootstrap check", len(clnts)/2))
 | 
						|
				if len(offlineEndpoints) > 0 {
 | 
						|
					logger.Info(fmt.Sprintf("Following servers are currently offline or unreachable %s", offlineEndpoints))
 | 
						|
				}
 | 
						|
				retries = 0 // reset to log again after 5 retries.
 | 
						|
			}
 | 
						|
			offlineEndpoints = nil
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return nil
 | 
						|
}
 | 
						|
 | 
						|
func newBootstrapRESTClients(endpointServerPools EndpointServerPools) []*bootstrapRESTClient {
 | 
						|
	seenHosts := set.NewStringSet()
 | 
						|
	var clnts []*bootstrapRESTClient
 | 
						|
	for _, ep := range endpointServerPools {
 | 
						|
		for _, endpoint := range ep.Endpoints {
 | 
						|
			if seenHosts.Contains(endpoint.Host) {
 | 
						|
				continue
 | 
						|
			}
 | 
						|
			seenHosts.Add(endpoint.Host)
 | 
						|
 | 
						|
			// Only proceed for remote endpoints.
 | 
						|
			if !endpoint.IsLocal {
 | 
						|
				clnts = append(clnts, newBootstrapRESTClient(endpoint))
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return clnts
 | 
						|
}
 | 
						|
 | 
						|
// Returns a new bootstrap client.
 | 
						|
func newBootstrapRESTClient(endpoint Endpoint) *bootstrapRESTClient {
 | 
						|
	serverURL := &url.URL{
 | 
						|
		Scheme: endpoint.Scheme,
 | 
						|
		Host:   endpoint.Host,
 | 
						|
		Path:   bootstrapRESTPath,
 | 
						|
	}
 | 
						|
 | 
						|
	restClient := rest.NewClient(serverURL, globalInternodeTransport, newCachedAuthToken())
 | 
						|
	restClient.HealthCheckFn = nil
 | 
						|
 | 
						|
	return &bootstrapRESTClient{endpoint: endpoint, restClient: restClient}
 | 
						|
}
 |