Fixed wrong import.

Signed-off-by: Bartek Plotka <[email protected]>

CHANGELOG.md

@@ -4,6 +4,7 @@
 Kedge service:
 * [x] added stripping out proxy auth header after using it.
 * [x] fixed error handling causing in particular cases.
+* [x] added graceful shutdown 
 
 Winch (kedge client):
 * [x] better error handling (adding response headers to indicate what error happen)

Dockerfile_loadtest

@@ -11,7 +11,7 @@ ENV GOBIN=/go/bin
 
 RUN mkdir /etc/corp-auth
 
-RUN apt-get update && apt-get install -qq -y --no-install-recommends git wget curl ca-certificates openssh-client
+RUN apt-get update && apt-get install -qq -y --no-install-recommends git vim wget curl ca-certificates openssh-client
 
 RUN curl -fsSL "${GOLANG_DOWNLOAD_URL}" -o golang.tar.gz \
       && tar -C /usr/local -xzf golang.tar.gz \

server/main.go

@@ -13,7 +13,6 @@ import (
 	"syscall"
 	"time"
 
-	"github.com/Sirupsen/logrus"
 	"github.com/grpc-ecosystem/go-grpc-middleware"
 	"github.com/grpc-ecosystem/go-grpc-middleware/logging/logrus"
 	"github.com/grpc-ecosystem/go-grpc-middleware/tags"
@@ -42,6 +41,7 @@ import (
 	"golang.org/x/net/trace"
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
+	"github.com/Sirupsen/logrus"
 )
 
 var (
@@ -256,7 +256,7 @@ func main() {
 }
 
 func shutdown(ctx context.Context, bouncer *http.Server, grpcDirector *grpc.Server, debugServer *http.Server) {
-	logrus.Info("Shutting down servers gracefully.")
+	log.Info("Shutting down servers gracefully.")
 
 	innerCtx, cancel := context.WithTimeout(ctx, 20*time.Second)
 	defer cancel()