diff options
Diffstat (limited to 'commands/limit_darwin.go')
-rw-r--r-- | commands/limit_darwin.go | 84 |
1 files changed, 84 insertions, 0 deletions
diff --git a/commands/limit_darwin.go b/commands/limit_darwin.go new file mode 100644 index 000000000..6799f37b1 --- /dev/null +++ b/commands/limit_darwin.go @@ -0,0 +1,84 @@ +// Copyright 2018 The Hugo Authors. All rights reserved. +// +// 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 commands + +import ( + "syscall" + + "github.com/spf13/cobra" + jww "github.com/spf13/jwalterweatherman" +) + +var _ cmder = (*limitCmd)(nil) + +type limitCmd struct { + *baseCmd +} + +func newLimitCmd() *limitCmd { + ccmd := &cobra.Command{ + Use: "ulimit", + Short: "Check system ulimit settings", + Long: `Hugo will inspect the current ulimit settings on the system. +This is primarily to ensure that Hugo can watch enough files on some OSs`, + RunE: func(cmd *cobra.Command, args []string) error { + var rLimit syscall.Rlimit + err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit) + if err != nil { + return newSystemError("Error Getting rlimit ", err) + } + + jww.FEEDBACK.Println("Current rLimit:", rLimit) + + if rLimit.Cur >= newRlimit { + return nil + } + + jww.FEEDBACK.Println("Attempting to increase limit") + rLimit.Cur = newRlimit + err = syscall.Setrlimit(syscall.RLIMIT_NOFILE, &rLimit) + if err != nil { + return newSystemError("Error Setting rLimit ", err) + } + err = syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit) + if err != nil { + return newSystemError("Error Getting rLimit ", err) + } + jww.FEEDBACK.Println("rLimit after change:", rLimit) + + return nil + }, + } + + return &limitCmd{baseCmd: newBaseCmd(ccmd)} +} + +const newRlimit = 10240 + +func tweakLimit() { + var rLimit syscall.Rlimit + err := syscall.Getrlimit(syscall.RLIMIT_NOFILE, &rLimit) + if err != nil { + jww.WARN.Println("Unable to get rlimit:", err) + return + } + if rLimit.Cur < newRlimit { + rLimit.Cur = newRlimit + err = syscall.Setrlimit(syscall.RLIMIT_NOFILE, &rLimit) + if err != nil { + // This may not succeed, see https://github.com/golang/go/issues/30401 + jww.INFO.Println("Unable to increase number of open files limit:", err) + } + } +} |