2015-02-08 11:54:21 +01:00
|
|
|
/*
|
2015-03-19 22:35:50 +01:00
|
|
|
* Minimalist Object Storage, (C) 2015 Minio, Inc.
|
2015-02-08 11:54:21 +01:00
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
2015-01-14 20:29:04 +01:00
|
|
|
package main
|
|
|
|
|
2015-01-26 02:15:18 +01:00
|
|
|
import (
|
2015-04-02 21:48:22 +02:00
|
|
|
"fmt"
|
2015-04-28 04:23:37 +02:00
|
|
|
"io/ioutil"
|
2015-01-29 04:52:30 +01:00
|
|
|
"os"
|
2015-04-02 21:48:22 +02:00
|
|
|
"os/user"
|
2015-04-23 11:33:32 +02:00
|
|
|
"path"
|
2015-04-02 21:48:22 +02:00
|
|
|
"runtime"
|
|
|
|
"strconv"
|
|
|
|
"strings"
|
2015-03-25 19:49:04 +01:00
|
|
|
"time"
|
2015-01-29 01:07:53 +01:00
|
|
|
|
2015-04-23 11:33:32 +02:00
|
|
|
"github.com/dustin/go-humanize"
|
2015-03-25 04:33:36 +01:00
|
|
|
"github.com/minio-io/cli"
|
2015-04-09 03:02:03 +02:00
|
|
|
"github.com/minio-io/minio/pkg/iodine"
|
2015-04-24 01:13:14 +02:00
|
|
|
"github.com/minio-io/minio/pkg/server"
|
2015-04-23 06:31:29 +02:00
|
|
|
"github.com/minio-io/minio/pkg/server/httpserver"
|
2015-03-24 02:12:20 +01:00
|
|
|
"github.com/minio-io/minio/pkg/utils/log"
|
2015-01-26 02:15:18 +01:00
|
|
|
)
|
|
|
|
|
2015-04-02 21:48:22 +02:00
|
|
|
var globalDebugFlag = false
|
|
|
|
|
2015-04-23 06:31:29 +02:00
|
|
|
var commands = []cli.Command{
|
|
|
|
modeCmd,
|
|
|
|
}
|
|
|
|
|
|
|
|
var modeCommands = []cli.Command{
|
|
|
|
memoryCmd,
|
|
|
|
donutCmd,
|
|
|
|
}
|
|
|
|
|
|
|
|
var modeCmd = cli.Command{
|
|
|
|
Name: "mode",
|
|
|
|
Subcommands: modeCommands,
|
2015-04-23 11:33:32 +02:00
|
|
|
Description: "Mode of execution",
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
var memoryCmd = cli.Command{
|
2015-04-23 11:33:32 +02:00
|
|
|
Name: "memory",
|
|
|
|
Description: "Limit maximum memory usage to SIZE in [B, KB, MB, GB]",
|
|
|
|
Action: runMemory,
|
|
|
|
CustomHelpTemplate: `NAME:
|
2015-04-23 12:34:16 +02:00
|
|
|
minio mode {{.Name}} - {{.Description}}
|
2015-04-23 11:33:32 +02:00
|
|
|
|
|
|
|
USAGE:
|
2015-04-23 12:34:16 +02:00
|
|
|
minio mode {{.Name}} SIZE
|
2015-04-23 11:33:32 +02:00
|
|
|
|
|
|
|
EXAMPLES:
|
|
|
|
1. Limit maximum memory usage to 64MB
|
2015-04-23 12:34:16 +02:00
|
|
|
$ minio mode {{.Name}} 64MB
|
2015-04-23 11:33:32 +02:00
|
|
|
|
|
|
|
2. Limit maximum memory usage to 4GB
|
2015-04-23 12:34:16 +02:00
|
|
|
$ minio mode {{.Name}} 4GB
|
2015-04-23 11:33:32 +02:00
|
|
|
`,
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
var donutCmd = cli.Command{
|
2015-04-23 11:33:32 +02:00
|
|
|
Name: "donut",
|
|
|
|
Description: "Specify a path to instantiate donut",
|
|
|
|
Action: runDonut,
|
|
|
|
CustomHelpTemplate: `NAME:
|
2015-04-23 12:34:16 +02:00
|
|
|
minio mode {{.Name}} - {{.Description}}
|
2015-04-23 11:33:32 +02:00
|
|
|
|
|
|
|
USAGE:
|
2015-04-23 12:34:16 +02:00
|
|
|
minio mode {{.Name}} PATH
|
2015-04-23 11:33:32 +02:00
|
|
|
|
|
|
|
EXAMPLES:
|
2015-04-23 12:34:16 +02:00
|
|
|
1. Create a donut volume under "/mnt/backup"
|
|
|
|
$ minio mode {{.Name}} /mnt/backup
|
|
|
|
|
|
|
|
2. Create a temporary donut volume under "/tmp"
|
|
|
|
$ minio mode {{.Name}} /tmp
|
2015-04-23 11:33:32 +02:00
|
|
|
|
2015-04-23 20:26:47 +02:00
|
|
|
3. Create a donut volume under collection of paths
|
|
|
|
$ minio mode {{.Name}} /mnt/backup2014feb /mnt/backup2014feb
|
|
|
|
|
2015-04-23 11:33:32 +02:00
|
|
|
`,
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
|
|
|
|
2015-03-25 04:33:36 +01:00
|
|
|
var flags = []cli.Flag{
|
|
|
|
cli.StringFlag{
|
2015-04-28 04:30:32 +02:00
|
|
|
Name: "domain",
|
2015-03-25 04:33:36 +01:00
|
|
|
Value: "",
|
|
|
|
Usage: "domain used for routing incoming API requests",
|
|
|
|
},
|
|
|
|
cli.StringFlag{
|
2015-04-28 04:30:32 +02:00
|
|
|
Name: "port-api",
|
2015-03-25 04:33:36 +01:00
|
|
|
Value: ":9000",
|
|
|
|
Usage: "address for incoming API requests",
|
|
|
|
},
|
|
|
|
cli.StringFlag{
|
2015-04-28 04:30:32 +02:00
|
|
|
Name: "port-mgmt",
|
2015-03-25 04:33:36 +01:00
|
|
|
Value: ":9001",
|
|
|
|
Usage: "address for incoming Management UI requests",
|
|
|
|
},
|
|
|
|
cli.StringFlag{
|
2015-04-28 04:30:32 +02:00
|
|
|
Name: "cert",
|
2015-03-25 04:33:36 +01:00
|
|
|
Hide: true,
|
|
|
|
Value: "",
|
|
|
|
Usage: "cert.pem",
|
|
|
|
},
|
|
|
|
cli.StringFlag{
|
2015-04-28 04:30:32 +02:00
|
|
|
Name: "key",
|
2015-03-25 04:33:36 +01:00
|
|
|
Hide: true,
|
|
|
|
Value: "",
|
|
|
|
Usage: "key.pem",
|
|
|
|
},
|
2015-04-02 21:48:22 +02:00
|
|
|
cli.BoolFlag{
|
|
|
|
Name: "debug",
|
|
|
|
Usage: "print debug information",
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
func init() {
|
|
|
|
// Check for the environment early on and gracefuly report.
|
|
|
|
_, err := user.Current()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("minio: Unable to obtain user's home directory. \nError: %s\n", err)
|
|
|
|
}
|
2015-03-25 04:33:36 +01:00
|
|
|
}
|
|
|
|
|
2015-04-23 06:31:29 +02:00
|
|
|
func runMemory(c *cli.Context) {
|
2015-04-23 11:33:32 +02:00
|
|
|
if len(c.Args()) < 1 {
|
|
|
|
cli.ShowCommandHelpAndExit(c, "memory", 1) // last argument is exit code
|
|
|
|
}
|
2015-04-23 06:31:29 +02:00
|
|
|
apiServerConfig := getAPIServerConfig(c)
|
2015-04-23 11:33:32 +02:00
|
|
|
maxMemory, err := humanize.ParseBytes(c.Args().First())
|
2015-04-23 06:31:29 +02:00
|
|
|
if err != nil {
|
2015-04-23 11:33:32 +02:00
|
|
|
log.Fatalf("MaxMemory not a numeric value with reason: %s", err)
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
2015-04-24 01:13:14 +02:00
|
|
|
memoryDriver := server.MemoryFactory{
|
2015-04-23 06:31:29 +02:00
|
|
|
Config: apiServerConfig,
|
2015-04-24 01:13:14 +02:00
|
|
|
MaxMemory: maxMemory,
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
2015-04-24 01:13:14 +02:00
|
|
|
apiServer := memoryDriver.GetStartServerFunc()
|
2015-04-23 06:31:29 +02:00
|
|
|
webServer := getWebServerConfigFunc(c)
|
2015-04-24 01:13:14 +02:00
|
|
|
servers := []server.StartServerFunc{apiServer, webServer}
|
|
|
|
server.StartMinio(servers)
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func runDonut(c *cli.Context) {
|
2015-04-23 11:33:32 +02:00
|
|
|
u, err := user.Current()
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalln(err)
|
|
|
|
}
|
|
|
|
if len(c.Args()) < 1 {
|
|
|
|
cli.ShowCommandHelpAndExit(c, "donut", 1) // last argument is exit code
|
|
|
|
}
|
2015-04-23 20:26:47 +02:00
|
|
|
|
|
|
|
// supporting multiple paths
|
|
|
|
var paths []string
|
|
|
|
if strings.TrimSpace(c.Args().First()) == "" {
|
|
|
|
p := path.Join(u.HomeDir, "minio-storage", "donut")
|
|
|
|
paths = append(paths, p)
|
|
|
|
} else {
|
|
|
|
for _, arg := range c.Args() {
|
|
|
|
paths = append(paths, strings.TrimSpace(arg))
|
|
|
|
}
|
2015-04-23 11:33:32 +02:00
|
|
|
}
|
2015-04-23 06:31:29 +02:00
|
|
|
apiServerConfig := getAPIServerConfig(c)
|
2015-04-24 01:13:14 +02:00
|
|
|
donutDriver := server.DonutFactory{
|
2015-04-23 06:31:29 +02:00
|
|
|
Config: apiServerConfig,
|
2015-04-24 01:13:14 +02:00
|
|
|
Paths: paths,
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
2015-04-24 01:13:14 +02:00
|
|
|
apiServer := donutDriver.GetStartServerFunc()
|
2015-04-23 06:31:29 +02:00
|
|
|
webServer := getWebServerConfigFunc(c)
|
2015-04-24 01:13:14 +02:00
|
|
|
servers := []server.StartServerFunc{apiServer, webServer}
|
|
|
|
server.StartMinio(servers)
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
|
|
|
|
2015-04-23 21:43:27 +02:00
|
|
|
func getAPIServerConfig(c *cli.Context) httpserver.Config {
|
2015-02-01 11:35:00 +01:00
|
|
|
certFile := c.String("cert")
|
|
|
|
keyFile := c.String("key")
|
|
|
|
if (certFile != "" && keyFile == "") || (certFile == "" && keyFile != "") {
|
2015-04-23 11:33:32 +02:00
|
|
|
log.Fatalln("Both certificate and key must be provided to enable https")
|
2015-02-01 11:35:00 +01:00
|
|
|
}
|
|
|
|
tls := (certFile != "" && keyFile != "")
|
2015-04-23 21:43:27 +02:00
|
|
|
return httpserver.Config{
|
2015-04-23 11:33:32 +02:00
|
|
|
Domain: c.GlobalString("domain"),
|
|
|
|
Address: c.GlobalString("api-address"),
|
2015-03-06 06:09:16 +01:00
|
|
|
TLS: tls,
|
2015-02-01 11:35:00 +01:00
|
|
|
CertFile: certFile,
|
|
|
|
KeyFile: keyFile,
|
|
|
|
}
|
2015-04-23 06:31:29 +02:00
|
|
|
}
|
|
|
|
|
2015-04-24 01:13:14 +02:00
|
|
|
func getWebServerConfigFunc(c *cli.Context) server.StartServerFunc {
|
2015-04-23 21:43:27 +02:00
|
|
|
config := httpserver.Config{
|
2015-04-23 11:33:32 +02:00
|
|
|
Domain: c.GlobalString("domain"),
|
|
|
|
Address: c.GlobalString("web-address"),
|
2015-03-06 06:09:16 +01:00
|
|
|
TLS: false,
|
2015-02-01 11:35:00 +01:00
|
|
|
CertFile: "",
|
|
|
|
KeyFile: "",
|
|
|
|
}
|
2015-04-24 01:13:14 +02:00
|
|
|
webDrivers := server.WebFactory{
|
2015-04-23 06:31:29 +02:00
|
|
|
Config: config,
|
|
|
|
}
|
2015-04-24 01:13:14 +02:00
|
|
|
return webDrivers.GetStartServerFunc()
|
2015-02-01 11:35:00 +01:00
|
|
|
}
|
|
|
|
|
2015-04-02 21:48:22 +02:00
|
|
|
// Tries to get os/arch/platform specific information
|
|
|
|
// Returns a map of current os/arch/platform/memstats
|
|
|
|
func getSystemData() map[string]string {
|
|
|
|
host, err := os.Hostname()
|
|
|
|
if err != nil {
|
|
|
|
host = ""
|
|
|
|
}
|
|
|
|
memstats := &runtime.MemStats{}
|
|
|
|
runtime.ReadMemStats(memstats)
|
|
|
|
mem := fmt.Sprintf("Used: %s | Allocated: %s | Used-Heap: %s | Allocated-Heap: %s",
|
2015-04-23 11:33:32 +02:00
|
|
|
humanize.Bytes(memstats.Alloc),
|
|
|
|
humanize.Bytes(memstats.TotalAlloc),
|
|
|
|
humanize.Bytes(memstats.HeapAlloc),
|
|
|
|
humanize.Bytes(memstats.HeapSys))
|
2015-04-02 21:48:22 +02:00
|
|
|
platform := fmt.Sprintf("Host: %s | OS: %s | Arch: %s",
|
|
|
|
host,
|
|
|
|
runtime.GOOS,
|
|
|
|
runtime.GOARCH)
|
|
|
|
goruntime := fmt.Sprintf("Version: %s | CPUs: %s", runtime.Version(), strconv.Itoa(runtime.NumCPU()))
|
|
|
|
return map[string]string{
|
|
|
|
"PLATFORM": platform,
|
|
|
|
"RUNTIME": goruntime,
|
|
|
|
"MEM": mem,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-04-25 06:49:07 +02:00
|
|
|
// Version is based on MD5SUM of its binary
|
|
|
|
var Version = mustHashBinarySelf()
|
|
|
|
|
2015-04-28 01:08:11 +02:00
|
|
|
// BuildDate - build time
|
|
|
|
var BuildDate string
|
2015-04-27 22:08:51 +02:00
|
|
|
|
2015-01-14 20:29:04 +01:00
|
|
|
func main() {
|
2015-03-25 19:44:43 +01:00
|
|
|
// set up iodine
|
2015-04-25 06:49:07 +02:00
|
|
|
iodine.SetGlobalState("minio.version", Version)
|
2015-03-25 19:44:43 +01:00
|
|
|
iodine.SetGlobalState("minio.starttime", time.Now().Format(time.RFC3339))
|
|
|
|
|
|
|
|
// set up app
|
2015-01-29 04:52:30 +01:00
|
|
|
app := cli.NewApp()
|
|
|
|
app.Name = "minio"
|
2015-04-25 06:49:07 +02:00
|
|
|
app.Version = Version
|
2015-04-28 01:08:11 +02:00
|
|
|
app.Compiled, _ = time.Parse(time.RFC3339Nano, BuildDate)
|
2015-03-25 04:33:36 +01:00
|
|
|
app.Author = "Minio.io"
|
|
|
|
app.Usage = "Minimalist Object Storage"
|
|
|
|
app.Flags = flags
|
2015-04-23 06:31:29 +02:00
|
|
|
app.Commands = commands
|
2015-04-02 21:48:22 +02:00
|
|
|
app.Before = func(c *cli.Context) error {
|
|
|
|
globalDebugFlag = c.GlobalBool("debug")
|
|
|
|
if globalDebugFlag {
|
|
|
|
app.ExtraInfo = getSystemData()
|
2015-04-28 04:23:37 +02:00
|
|
|
} else {
|
|
|
|
log.Debug = log.New(ioutil.Discard, "", 0)
|
2015-04-02 21:48:22 +02:00
|
|
|
}
|
|
|
|
return nil
|
|
|
|
}
|
2015-04-06 00:52:31 +02:00
|
|
|
app.RunAndExitOnError()
|
2015-01-14 20:29:04 +01:00
|
|
|
}
|