seaweedfs/weed/command/master.go

140 lines
4.6 KiB
Go
Raw Normal View History

package command
import (
2012-09-04 02:18:02 +00:00
"net/http"
"os"
"runtime"
2012-09-04 02:18:02 +00:00
"strconv"
"strings"
2012-09-28 16:13:17 +00:00
"time"
"github.com/chrislusf/seaweedfs/weed/glog"
2018-05-10 06:11:54 +00:00
"github.com/chrislusf/seaweedfs/weed/pb/master_pb"
"github.com/chrislusf/seaweedfs/weed/server"
"github.com/chrislusf/seaweedfs/weed/util"
"github.com/gorilla/mux"
2017-01-10 09:01:12 +00:00
"github.com/soheilhy/cmux"
"google.golang.org/grpc/reflection"
)
func init() {
2012-09-04 02:18:02 +00:00
cmdMaster.Run = runMaster // break init cycle
}
var cmdMaster = &Command{
2012-09-04 02:18:02 +00:00
UsageLine: "master -port=9333",
Short: "start a master server",
Long: `start a master server to provide volume=>location mapping service
and sequence number of file ids
`,
}
var (
mport = cmdMaster.Flag.Int("port", 9333, "http listen port")
2015-02-12 05:04:43 +00:00
masterIp = cmdMaster.Flag.String("ip", "localhost", "master <ip>|<server> address")
2014-10-21 08:28:17 +00:00
masterBindIp = cmdMaster.Flag.String("ip.bind", "0.0.0.0", "ip address to bind to")
metaFolder = cmdMaster.Flag.String("mdir", os.TempDir(), "data directory to store meta data")
2018-08-12 21:25:31 +00:00
masterPeers = cmdMaster.Flag.String("peers", "", "all master nodes in comma separated ip:port list, example: 127.0.0.1:9093,127.0.0.1:9094")
volumeSizeLimitMB = cmdMaster.Flag.Uint("volumeSizeLimitMB", 30*1000, "Master stops directing writes to oversized volumes.")
2017-01-08 19:01:46 +00:00
volumePreallocate = cmdMaster.Flag.Bool("volumePreallocate", false, "Preallocate disk space for volumes.")
mpulse = cmdMaster.Flag.Int("pulseSeconds", 5, "number of seconds between heartbeats")
defaultReplicaPlacement = cmdMaster.Flag.String("defaultReplication", "000", "Default replication type if not specified.")
// mTimeout = cmdMaster.Flag.Int("idleTimeout", 30, "connection idle seconds")
mMaxCpu = cmdMaster.Flag.Int("maxCpu", 0, "maximum number of CPUs. 0 means all available CPUs")
garbageThreshold = cmdMaster.Flag.String("garbageThreshold", "0.3", "threshold to vacuum and reclaim spaces")
masterWhiteListOption = cmdMaster.Flag.String("whiteList", "", "comma separated Ip addresses having write permission. No limit if empty.")
masterSecureKey = cmdMaster.Flag.String("secure.secret", "", "secret to encrypt Json Web Token(JWT)")
masterCpuProfile = cmdMaster.Flag.String("cpuprofile", "", "cpu profile output file")
2017-06-22 08:33:58 +00:00
masterMemProfile = cmdMaster.Flag.String("memprofile", "", "memory profile output file")
masterWhiteList []string
)
func runMaster(cmd *Command, args []string) bool {
if *mMaxCpu < 1 {
*mMaxCpu = runtime.NumCPU()
}
runtime.GOMAXPROCS(*mMaxCpu)
2017-06-22 08:33:58 +00:00
util.SetupProfiling(*masterCpuProfile, *masterMemProfile)
if err := util.TestFolderWritable(*metaFolder); err != nil {
glog.Fatalf("Check Meta Folder (-mdir) Writable %s : %s", *metaFolder, err)
}
if *masterWhiteListOption != "" {
masterWhiteList = strings.Split(*masterWhiteListOption, ",")
}
if *volumeSizeLimitMB > 30*1000 {
glog.Fatalf("volumeSizeLimitMB should be smaller than 30000")
}
2013-12-01 22:41:47 +00:00
r := mux.NewRouter()
ms := weed_server.NewMasterServer(r, *mport, *metaFolder,
2017-01-08 19:01:46 +00:00
*volumeSizeLimitMB, *volumePreallocate,
2017-02-13 05:58:44 +00:00
*mpulse, *defaultReplicaPlacement, *garbageThreshold,
2015-01-05 22:20:04 +00:00
masterWhiteList, *masterSecureKey,
)
2012-09-04 04:31:13 +00:00
2014-10-21 08:28:17 +00:00
listeningAddress := *masterBindIp + ":" + strconv.Itoa(*mport)
glog.V(0).Infoln("Start Seaweed Master", util.VERSION, "at", listeningAddress)
listener, e := util.NewListener(listeningAddress, 0)
if e != nil {
2015-01-14 01:04:41 +00:00
glog.Fatalf("Master startup error: %v", e)
2012-09-28 17:21:06 +00:00
}
go func() {
time.Sleep(100 * time.Millisecond)
2018-08-12 21:25:31 +00:00
myMasterAddress, peers := checkPeers(*masterIp, *mport, *masterPeers)
raftServer := weed_server.NewRaftServer(r, peers, myMasterAddress, *metaFolder, ms.Topo, *mpulse)
ms.SetRaftServer(raftServer)
}()
2017-01-10 09:01:12 +00:00
// start grpc and http server
m := cmux.New(listener)
grpcL := m.Match(cmux.HTTP2HeaderField("content-type", "application/grpc"))
httpL := m.Match(cmux.Any())
// Create your protocol servers.
2018-07-04 02:07:55 +00:00
grpcS := util.NewGrpcServer()
2018-05-10 06:11:54 +00:00
master_pb.RegisterSeaweedServer(grpcS, ms)
2017-01-10 09:01:12 +00:00
reflection.Register(grpcS)
httpS := &http.Server{Handler: r}
go grpcS.Serve(grpcL)
go httpS.Serve(httpL)
if err := m.Serve(); err != nil {
glog.Fatalf("master server failed to serve: %v", err)
2012-09-04 02:18:02 +00:00
}
2017-01-10 09:01:12 +00:00
2012-09-04 02:18:02 +00:00
return true
}
2018-08-12 21:25:31 +00:00
func checkPeers(masterIp string, masterPort int, peers string) (masterAddress string, cleanedPeers []string) {
masterAddress = masterIp + ":" + strconv.Itoa(masterPort)
if peers != "" {
cleanedPeers = strings.Split(peers, ",")
}
hasSelf := false
for _, peer := range cleanedPeers {
if peer == masterAddress {
hasSelf = true
break
}
}
peerCount := len(cleanedPeers)
if !hasSelf {
peerCount += 1
}
if peerCount%2 == 0 {
2018-08-12 21:25:31 +00:00
glog.Fatalf("Only odd number of masters are supported!")
}
return
}