Remove remaining uses of golang.org/x/net/context (#1464)

cli/prepare.go

@@ -2,6 +2,7 @@ package cli
 
 import (
 	"bufio"
+	"context"
 	"fmt"
 	"os"
 
@@ -13,7 +14,6 @@ import (
 	"github.com/kolide/kolide/server/pubsub"
 	"github.com/kolide/kolide/server/service"
 	"github.com/spf13/cobra"
-	"golang.org/x/net/context"
 )
 
 func createPrepareCmd(configManager config.Manager) *cobra.Command {

cli/serve.go

@@ -1,6 +1,7 @@
 package cli
 
 import (
+	"context"
 	"crypto/tls"
 	"fmt"
 	"net/http"
@@ -25,7 +26,6 @@ import (
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/spf13/cobra"
-	"golang.org/x/net/context"
 )
 
 type initializer interface {
@@ -223,10 +223,11 @@ the way that the kolide server works.
 				}
 			}()
 			go func() {
-				sig := make(chan os.Signal)
+				sig := make(chan os.Signal, 1)
 				signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
 				<-sig //block on signal
-				ctx, _ := context.WithTimeout(context.Background(), 30*time.Second)
+				ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
+				defer cancel()
 				errs <- srv.Shutdown(ctx)
 			}()
 

server/kolide/decorators.go

@@ -1,10 +1,9 @@
 package kolide
 
 import (
+	"context"
 	"errors"
 	"strings"
-
-	"golang.org/x/net/context"
 )
 
 // DecoratorStore methods to manipulate decorator queries.

server/service/logging_decorators.go

@@ -1,10 +1,10 @@
 package service
 
 import (
+	"context"
 	"time"
 
 	"github.com/kolide/kolide/server/kolide"
-	"golang.org/x/net/context"
 )
 
 func (mw loggingMiddleware) ListDecorators(ctx context.Context) ([]*kolide.Decorator, error) {

server/service/metrics_decorators.go

@@ -1,11 +1,11 @@
 package service
 
 import (
+	"context"
 	"fmt"
 	"time"
 
 	"github.com/kolide/kolide/server/kolide"
-	"golang.org/x/net/context"
 )
 
 func (mw metricsMiddleware) ListDecorators(ctx context.Context) ([]*kolide.Decorator, error) {

server/service/service_decorators.go

@@ -1,8 +1,9 @@
 package service
 
 import (
+	"context"
+
 	"github.com/kolide/kolide/server/kolide"
-	"golang.org/x/net/context"
 )
 
 func (svc service) ListDecorators(ctx context.Context) ([]*kolide.Decorator, error) {

server/service/validation_decorators.go

@@ -1,8 +1,9 @@
 package service
 
 import (
+	"context"
+
 	"github.com/kolide/kolide/server/kolide"
-	"golang.org/x/net/context"
 )
 
 func validateNewDecoratorType(payload kolide.DecoratorPayload, invalid *invalidArgumentError) {

server/service/validation_decorators_test.go

@@ -1,13 +1,13 @@
 package service
 
 import (
+	"context"
 	"testing"
 
 	"github.com/kolide/kolide/server/kolide"
 	"github.com/kolide/kolide/server/mock"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
-	"golang.org/x/net/context"
 )
 
 var dtPtr = func(t kolide.DecoratorType) *kolide.DecoratorType { return &t }