From 5fdb1d89cebbb2f4947a1d2b356b83f90b951239 Mon Sep 17 00:00:00 2001 From: Chris Lu Date: Tue, 3 Dec 2013 23:22:26 -0800 Subject: [PATCH] adding "server" command to start both volume server and an embedded master server --- go/weed/master.go | 4 +- go/weed/server.go | 131 ++++++++++++++++++++++++++++++++++++++++++++++ go/weed/weed.go | 1 + 3 files changed, 134 insertions(+), 2 deletions(-) create mode 100644 go/weed/server.go diff --git a/go/weed/master.go b/go/weed/master.go index f4c2ed087..a1dffe5ac 100644 --- a/go/weed/master.go +++ b/go/weed/master.go @@ -28,7 +28,7 @@ var cmdMaster = &Command{ var ( mport = cmdMaster.Flag.Int("port", 9333, "http listen port") mip = cmdMaster.Flag.String("ip", "localhost", "http listen port") - metaFolder = cmdMaster.Flag.String("mdir", os.TempDir(), "data directory to store mappings") + metaFolder = cmdMaster.Flag.String("mdir", os.TempDir(), "data directory to store meta data") volumeSizeLimitMB = cmdMaster.Flag.Uint("volumeSizeLimitMB", 32*1024, "Default Volume Size in MegaBytes") mpulse = cmdMaster.Flag.Int("pulseSeconds", 5, "number of seconds between heartbeats") confFile = cmdMaster.Flag.String("conf", "/etc/weedfs/weedfs.conf", "xml configuration file") @@ -58,7 +58,7 @@ func runMaster(cmd *Command, args []string) bool { glog.V(0).Infoln("Start Weed Master", VERSION, "at port", *mip+":"+strconv.Itoa(*mport)) srv := &http.Server{ - Addr: *mip+":" + strconv.Itoa(*mport), + Addr: *mip + ":" + strconv.Itoa(*mport), Handler: r, ReadTimeout: time.Duration(*mReadTimeout) * time.Second, } diff --git a/go/weed/server.go b/go/weed/server.go new file mode 100644 index 000000000..b242f0592 --- /dev/null +++ b/go/weed/server.go @@ -0,0 +1,131 @@ +package main + +import ( + "code.google.com/p/weed-fs/go/glog" + "code.google.com/p/weed-fs/go/weed/weed_server" + "github.com/gorilla/mux" + "net/http" + "os" + "runtime" + "strconv" + "strings" + "time" +) + +func init() { + cmdServer.Run = runServer // break init cycle +} + +var cmdServer = &Command{ + UsageLine: "server -port=8080 -dir=/tmp -max=5 -ip=server_name", + Short: "start a server, including volume server, and automatically elect a master server", + Long: `start both a volume server to provide storage spaces + and a master server to provide volume=>location mapping service and sequence number of file ids + + This is provided as a convenient way to start both volume server and master server. + The servers are the same as starting them separately. + So other volume servers can use this embedded master server also. + + However, this may change very soon. + The target is to start both volume server and embedded master server on all instances, + and use a leader election process to auto choose a master server. + + `, +} + +var ( + serverIp = cmdServer.Flag.String("ip", "localhost", "ip or server name") + serverMaxCpu = cmdServer.Flag.Int("maxCpu", 0, "maximum number of CPUs. 0 means all available CPUs") + serverReadTimeout = cmdServer.Flag.Int("readTimeout", 3, "connection read timeout in seconds. Increase this if uploading large files.") + serverDataCenter = cmdServer.Flag.String("dataCenter", "", "current volume server's data center name") + serverRack = cmdServer.Flag.String("rack", "", "current volume server's rack name") + serverWhiteListOption = cmdServer.Flag.String("whiteList", "", "comma separated Ip addresses having write permission. No limit if empty.") + masterPort = cmdServer.Flag.Int("masterPort", 9333, "master server http listen port") + masterMetaFolder = cmdServer.Flag.String("mdir", os.TempDir(), "data directory to store meta data") + masterVolumeSizeLimitMB = cmdServer.Flag.Uint("volumeSizeLimitMB", 32*1024, "Default Volume Size in MegaBytes") + masterConfFile = cmdServer.Flag.String("conf", "/etc/weedfs/weedfs.conf", "xml configuration file") + masterDefaultRepType = cmdServer.Flag.String("defaultReplicationType", "000", "Default replication type if not specified.") + volumePort = cmdServer.Flag.Int("port", 8080, "volume server http listen port") + volumePublicUrl = cmdServer.Flag.String("publicUrl", "", "Publicly accessible :") + volumeDataFolders = cmdServer.Flag.String("dir", os.TempDir(), "directories to store data files. dir[,dir]...") + volumeMaxDataVolumeCounts = cmdServer.Flag.String("max", "7", "maximum numbers of volumes, count[,count]...") + volumePulse = cmdServer.Flag.Int("pulseSeconds", 5, "number of seconds between heartbeats, must be smaller than the master's setting") + + serverWhiteList []string +) + +func runServer(cmd *Command, args []string) bool { + if *serverMaxCpu < 1 { + *serverMaxCpu = runtime.NumCPU() + } + runtime.GOMAXPROCS(*serverMaxCpu) + + folders := strings.Split(*volumeDataFolders, ",") + maxCountStrings := strings.Split(*volumeMaxDataVolumeCounts, ",") + maxCounts := make([]int, 0) + for _, maxString := range maxCountStrings { + if max, e := strconv.Atoi(maxString); e == nil { + maxCounts = append(maxCounts, max) + } else { + glog.Fatalf("The max specified in -max not a valid number %s", max) + } + } + if len(folders) != len(maxCounts) { + glog.Fatalf("%d directories by -dir, but only %d max is set by -max", len(folders), len(maxCounts)) + } + for _, folder := range folders { + fileInfo, err := os.Stat(folder) + if err != nil { + glog.Fatalf("No Existing Folder:%s", folder) + } + if !fileInfo.IsDir() { + glog.Fatalf("Volume Folder should not be a file:%s", folder) + } + perm := fileInfo.Mode().Perm() + glog.V(0).Infoln("Volume Folder", folder) + glog.V(0).Infoln("Permission:", perm) + } + + if *volumePublicUrl == "" { + *volumePublicUrl = *serverIp + ":" + strconv.Itoa(*volumePort) + } + if *serverWhiteListOption != "" { + serverWhiteList = strings.Split(*serverWhiteListOption, ",") + } + + go func() { + r := mux.NewRouter() + weed_server.NewMasterServer(r, VERSION, *masterPort, *masterMetaFolder, + *masterVolumeSizeLimitMB, *volumePulse, *masterConfFile, *masterDefaultRepType, *garbageThreshold, serverWhiteList, + ) + + glog.V(0).Infoln("Start Weed Master", VERSION, "at port", *serverIp+":"+strconv.Itoa(*masterPort)) + masterServer := &http.Server{ + Addr: *serverIp + ":" + strconv.Itoa(*masterPort), + Handler: r, + ReadTimeout: time.Duration(*serverReadTimeout) * time.Second, + } + e := masterServer.ListenAndServe() + if e != nil { + glog.Fatalf("Fail to start master:%s", e) + } + }() + + r := mux.NewRouter() + weed_server.NewVolumeServer(r, VERSION, *serverIp, *volumePort, *volumePublicUrl, folders, maxCounts, + *serverIp+":"+strconv.Itoa(*masterPort), *volumePulse, *serverDataCenter, *serverRack, serverWhiteList, + ) + + glog.V(0).Infoln("Start Weed volume server", VERSION, "at http://"+*serverIp+":"+strconv.Itoa(*volumePort)) + volumeServer := &http.Server{ + Addr: *serverIp + ":" + strconv.Itoa(*volumePort), + Handler: r, + ReadTimeout: (time.Duration(*serverReadTimeout) * time.Second), + } + e := volumeServer.ListenAndServe() + if e != nil { + glog.Fatalf("Fail to start volume:%s", e.Error()) + } + + return true +} diff --git a/go/weed/weed.go b/go/weed/weed.go index 896b8e237..b02b1f6ff 100644 --- a/go/weed/weed.go +++ b/go/weed/weed.go @@ -21,6 +21,7 @@ var server *string var commands = []*Command{ cmdCompact, cmdFix, + cmdServer, cmdMaster, cmdUpload, cmdDownload,