Skip to content

Commit

Permalink
Reduce CPU usage when --timeout=0
Browse files Browse the repository at this point in the history
* Add second go routine for when a Timer is not needed.
* goimports updated some project files

Fixes containers#5531

Signed-off-by: Jhon Honce <jhonce@redhat.com>
  • Loading branch information
jwhonce authored and snj33v committed May 31, 2020
1 parent b953a56 commit dedcb39
Show file tree
Hide file tree
Showing 5 changed files with 65 additions and 41 deletions.
90 changes: 56 additions & 34 deletions pkg/api/server/server.go
Original file line number Diff line number Diff line change
Expand Up @@ -83,10 +83,6 @@ func newServer(runtime *libpod.Runtime, duration time.Duration, listener *net.Li
ConnectionCh: make(chan int),
}

server.Timer = time.AfterFunc(server.Duration, func() {
server.ConnectionCh <- NOOPHandler
})

router.NotFoundHandler = http.HandlerFunc(
func(w http.ResponseWriter, r *http.Request) {
// We can track user errors...
Expand Down Expand Up @@ -140,36 +136,15 @@ func newServer(runtime *libpod.Runtime, duration time.Duration, listener *net.Li

// Serve starts responding to HTTP requests
func (s *APIServer) Serve() error {
// stalker to count the connections. Should the timer expire it will shutdown the service.
go func() {
for delta := range s.ConnectionCh {
switch delta {
case EnterHandler:
s.Timer.Stop()
s.ActiveConnections += 1
s.TotalConnections += 1
case ExitHandler:
s.Timer.Stop()
s.ActiveConnections -= 1
if s.ActiveConnections == 0 {
// Server will be shutdown iff the timer expires before being reset or stopped
s.Timer = time.AfterFunc(s.Duration, func() {
if err := s.Shutdown(); err != nil {
logrus.Errorf("Failed to shutdown APIServer: %v", err)
os.Exit(1)
}
})
} else {
s.Timer.Reset(s.Duration)
}
case NOOPHandler:
// push the check out another duration...
s.Timer.Reset(s.Duration)
default:
logrus.Errorf("ConnectionCh received unsupported input %d", delta)
}
}
}()
// This is initialized here as Timer is not needed until Serve'ing
if s.Duration > 0 {
s.Timer = time.AfterFunc(s.Duration, func() {
s.ConnectionCh <- NOOPHandler
})
go s.ReadChannelWithTimeout()
} else {
go s.ReadChannelNoTimeout()
}

sigChan := make(chan os.Signal, 1)
signal.Notify(sigChan, syscall.SIGINT, syscall.SIGTERM)
Expand All @@ -194,6 +169,53 @@ func (s *APIServer) Serve() error {
return nil
}

func (s *APIServer) ReadChannelWithTimeout() {
// stalker to count the connections. Should the timer expire it will shutdown the service.
for delta := range s.ConnectionCh {
switch delta {
case EnterHandler:
s.Timer.Stop()
s.ActiveConnections += 1
s.TotalConnections += 1
case ExitHandler:
s.Timer.Stop()
s.ActiveConnections -= 1
if s.ActiveConnections == 0 {
// Server will be shutdown iff the timer expires before being reset or stopped
s.Timer = time.AfterFunc(s.Duration, func() {
if err := s.Shutdown(); err != nil {
logrus.Errorf("Failed to shutdown APIServer: %v", err)
os.Exit(1)
}
})
} else {
s.Timer.Reset(s.Duration)
}
case NOOPHandler:
// push the check out another duration...
s.Timer.Reset(s.Duration)
default:
logrus.Warnf("ConnectionCh received unsupported input %d", delta)
}
}
}

func (s *APIServer) ReadChannelNoTimeout() {
// stalker to count the connections.
for delta := range s.ConnectionCh {
switch delta {
case EnterHandler:
s.ActiveConnections += 1
s.TotalConnections += 1
case ExitHandler:
s.ActiveConnections -= 1
case NOOPHandler:
default:
logrus.Warnf("ConnectionCh received unsupported input %d", delta)
}
}
}

// Shutdown is a clean shutdown waiting on existing clients
func (s *APIServer) Shutdown() error {
// Duration == 0 flags no auto-shutdown of the server
Expand Down
2 changes: 1 addition & 1 deletion pkg/bindings/test/common_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -152,7 +152,7 @@ func (b *bindingTest) startAPIService() *gexec.Session {
var (
cmd []string
)
cmd = append(cmd, "--log-level=debug", "system", "service", "--timeout=999999", b.sock)
cmd = append(cmd, "--log-level=debug", "system", "service", "--timeout=0", b.sock)
return b.runPodman(cmd)
}

Expand Down
5 changes: 3 additions & 2 deletions pkg/bindings/test/volumes_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -3,11 +3,12 @@ package test_bindings
import (
"context"
"fmt"
"net/http"
"time"

"github.com/containers/libpod/pkg/api/handlers"
"github.com/containers/libpod/pkg/bindings/containers"
"github.com/containers/libpod/pkg/bindings/volumes"
"net/http"
"time"

"github.com/containers/libpod/pkg/bindings"
. "github.com/onsi/ginkgo"
Expand Down
2 changes: 1 addition & 1 deletion pkg/varlinkapi/system.go
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ import (
"time"

"github.com/containers/image/v5/pkg/sysregistriesv2"
"github.com/containers/libpod/cmd/podman/varlink"
iopodman "github.com/containers/libpod/cmd/podman/varlink"
"github.com/containers/libpod/libpod/define"
"github.com/sirupsen/logrus"
)
Expand Down
7 changes: 4 additions & 3 deletions test/e2e/search_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -5,13 +5,14 @@ package integration
import (
"bytes"
"fmt"
. "github.com/containers/libpod/test/utils"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
"io/ioutil"
"os"
"strconv"
"text/template"

. "github.com/containers/libpod/test/utils"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)

type endpoint struct {
Expand Down

0 comments on commit dedcb39

Please sign in to comment.