mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-08-28 10:54:59 +00:00
e3d6f96657
A configurable limit has been added to restrict the number of mutating calls in a `pql.Query`. This is to prevents requests from timing out from large queries. The default is set to 5000 writes per request and is configurable through the configuration file and the command line flags.
111 lines
4.4 KiB
Go
111 lines
4.4 KiB
Go
// Copyright 2017 Pilosa Corp.
|
|
//
|
|
// 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"
|
|
"io"
|
|
"os"
|
|
"os/signal"
|
|
"runtime/pprof"
|
|
"time"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/pilosa/pilosa/server"
|
|
)
|
|
|
|
// Server is global so that tests can control and verify it.
|
|
var Server *server.Command
|
|
|
|
func NewServeCmd(stdin io.Reader, stdout, stderr io.Writer) *cobra.Command {
|
|
Server = server.NewCommand(stdin, stdout, stderr)
|
|
serveCmd := &cobra.Command{
|
|
Use: "server",
|
|
Short: "Run Pilosa.",
|
|
Long: `pilosa server runs Pilosa.
|
|
|
|
It will load existing data from the configured
|
|
directory, and start listening client connections
|
|
on the configured port.`,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
Server.Server.Handler.Version = Version
|
|
fmt.Fprintf(Server.Stderr, "Pilosa %s, build time %s\n", Version, BuildTime)
|
|
|
|
// Start CPU profiling.
|
|
if Server.CPUProfile != "" {
|
|
f, err := os.Create(Server.CPUProfile)
|
|
if err != nil {
|
|
return fmt.Errorf("create cpu profile: %v", err)
|
|
}
|
|
defer f.Close()
|
|
|
|
fmt.Fprintln(Server.Stderr, "Starting cpu profile")
|
|
pprof.StartCPUProfile(f)
|
|
time.AfterFunc(Server.CPUTime, func() {
|
|
fmt.Fprintln(Server.Stderr, "Stopping cpu profile")
|
|
pprof.StopCPUProfile()
|
|
f.Close()
|
|
})
|
|
}
|
|
|
|
// Execute the program.
|
|
if err := Server.Run(); err != nil {
|
|
return fmt.Errorf("error running server: %v", err)
|
|
}
|
|
|
|
// First SIGKILL causes server to shut down gracefully.
|
|
c := make(chan os.Signal, 2)
|
|
signal.Notify(c, os.Interrupt)
|
|
select {
|
|
case sig := <-c:
|
|
fmt.Fprintf(Server.Stderr, "Received %s; gracefully shutting down...\n", sig.String())
|
|
|
|
// Second signal causes a hard shutdown.
|
|
go func() { <-c; os.Exit(1) }()
|
|
|
|
if err := Server.Close(); err != nil {
|
|
return err
|
|
}
|
|
case <-Server.Done:
|
|
fmt.Fprintf(Server.Stderr, "Server closed externally")
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
flags := serveCmd.Flags()
|
|
|
|
flags.StringVarP(&Server.Config.DataDir, "data-dir", "d", "~/.pilosa", "Directory to store pilosa data files.")
|
|
flags.StringVarP(&Server.Config.Host, "bind", "b", ":10101", "Default URI on which pilosa should listen.")
|
|
flags.IntVarP(&Server.Config.MaxWritesPerRequest, "max-writes-per-request", "", Server.Config.MaxWritesPerRequest, "Number of write commands per request.")
|
|
flags.IntVarP(&Server.Config.Cluster.ReplicaN, "cluster.replicas", "", 1, "Number of hosts each piece of data should be stored on.")
|
|
flags.StringSliceVarP(&Server.Config.Cluster.Hosts, "cluster.hosts", "", []string{}, "Comma separated list of hosts in cluster.")
|
|
flags.StringSliceVarP(&Server.Config.Cluster.InternalHosts, "cluster.internal-hosts", "", []string{}, "Comma separated list of hosts in cluster used for internal communication.")
|
|
flags.DurationVarP((*time.Duration)(&Server.Config.Cluster.PollingInterval), "cluster.poll-interval", "", time.Minute, "Polling interval for cluster.") // TODO what actually is this?
|
|
flags.StringVarP(&Server.Config.Plugins.Path, "plugins.path", "", "", "Path to plugin directory.")
|
|
flags.StringVar(&Server.Config.LogPath, "log-path", "", "Log path")
|
|
flags.DurationVarP((*time.Duration)(&Server.Config.AntiEntropy.Interval), "anti-entropy.interval", "", time.Minute*10, "Interval at which to run anti-entropy routine.")
|
|
flags.StringVarP(&Server.CPUProfile, "profile.cpu", "", "", "Where to store CPU profile.")
|
|
flags.DurationVarP(&Server.CPUTime, "profile.cpu-time", "", 30*time.Second, "CPU profile duration.")
|
|
flags.StringVarP(&Server.Config.Cluster.Type, "cluster.type", "", "static", "Determine how the cluster handles membership and state sharing. Choose from [static, http, gossip]")
|
|
flags.StringVarP(&Server.Config.Cluster.GossipSeed, "cluster.gossip-seed", "", "", "Host with which to seed the gossip membership.")
|
|
flags.StringVarP(&Server.Config.Cluster.InternalPort, "cluster.internal-port", "", "", "Port to which pilosa should bind for internal state sharing.")
|
|
|
|
return serveCmd
|
|
}
|
|
|
|
func init() {
|
|
subcommandFns["server"] = NewServeCmd
|
|
}
|