You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
114 lines
3.2 KiB
114 lines
3.2 KiB
/*
|
|
* Minio Cloud Storage, (C) 2016 Minio, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"net/rpc"
|
|
"path"
|
|
"strings"
|
|
|
|
router "github.com/gorilla/mux"
|
|
"github.com/minio/minio-go/pkg/set"
|
|
)
|
|
|
|
// Routes paths for "minio control" commands.
|
|
const (
|
|
controlPath = "/control"
|
|
)
|
|
|
|
// Find local node through the command line arguments.
|
|
func getLocalAddress(srvCmdConfig serverCmdConfig) string {
|
|
if !srvCmdConfig.isDistXL {
|
|
return fmt.Sprintf(":%d", globalMinioPort)
|
|
}
|
|
for _, export := range srvCmdConfig.disks {
|
|
// Validates if remote disk is local.
|
|
if isLocalStorage(export) {
|
|
var host string
|
|
if idx := strings.LastIndex(export, ":"); idx != -1 {
|
|
host = export[:idx]
|
|
}
|
|
return fmt.Sprintf("%s:%d", host, globalMinioPort)
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
// Initializes remote control clients for making remote requests.
|
|
func initRemoteControlClients(srvCmdConfig serverCmdConfig) []*AuthRPCClient {
|
|
if !srvCmdConfig.isDistXL {
|
|
return nil
|
|
}
|
|
var newExports []string
|
|
// Initialize auth rpc clients.
|
|
exports := srvCmdConfig.disks
|
|
remoteHosts := set.NewStringSet()
|
|
|
|
var remoteControlClnts []*AuthRPCClient
|
|
for _, export := range exports {
|
|
// Validates if remote disk is local.
|
|
if isLocalStorage(export) {
|
|
continue
|
|
}
|
|
newExports = append(newExports, export)
|
|
}
|
|
for _, export := range newExports {
|
|
var host string
|
|
if idx := strings.LastIndex(export, ":"); idx != -1 {
|
|
host = export[:idx]
|
|
}
|
|
remoteHosts.Add(fmt.Sprintf("%s:%d", host, globalMinioPort))
|
|
}
|
|
for host := range remoteHosts {
|
|
remoteControlClnts = append(remoteControlClnts, newAuthClient(&authConfig{
|
|
accessKey: serverConfig.GetCredential().AccessKeyID,
|
|
secretKey: serverConfig.GetCredential().SecretAccessKey,
|
|
secureConn: isSSL(),
|
|
address: host,
|
|
path: path.Join(reservedBucket, controlPath),
|
|
loginMethod: "Control.LoginHandler",
|
|
}))
|
|
}
|
|
return remoteControlClnts
|
|
}
|
|
|
|
// Represents control object which provides handlers for control
|
|
// operations on server.
|
|
type controlAPIHandlers struct {
|
|
ObjectAPI func() ObjectLayer
|
|
StorageDisks []StorageAPI
|
|
RemoteControls []*AuthRPCClient
|
|
LocalNode string
|
|
}
|
|
|
|
// Register control RPC handlers.
|
|
func registerControlRPCRouter(mux *router.Router, srvCmdConfig serverCmdConfig) {
|
|
// Initialize Control.
|
|
ctrlHandlers := &controlAPIHandlers{
|
|
ObjectAPI: newObjectLayerFn,
|
|
RemoteControls: initRemoteControlClients(srvCmdConfig),
|
|
LocalNode: getLocalAddress(srvCmdConfig),
|
|
StorageDisks: srvCmdConfig.storageDisks,
|
|
}
|
|
|
|
ctrlRPCServer := rpc.NewServer()
|
|
ctrlRPCServer.RegisterName("Control", ctrlHandlers)
|
|
|
|
ctrlRouter := mux.NewRoute().PathPrefix(reservedBucket).Subrouter()
|
|
ctrlRouter.Path(controlPath).Handler(ctrlRPCServer)
|
|
}
|
|
|