Update Thrift dependency version (and generated bindings) (osquery#59)

- Refactor for use of `context.Context` in new thrift bindings
- Does not (yet) expose the context to users of the library

Author: zwass

Gopkg.lock

@@ -1,12 +1,7 @@
 
 [[constraint]]
-  # This version includes concurrency fixes, but excludes changes to the
-  # library APIs. Using a version beyond this would require users to build the
-  # Thrift compiler from source to get API-compatible generated code. After the
-  # next Thrift release, this should become that tagged release (and the code
-  # generated appropriately).
   name = "git.apache.org/thrift.git"
-  revision = "0dd823580c78a79ae9696eb9b3650e400fff140f"
+  version = "0.11.0"
 
 [[constraint]]
   name = "github.com/Microsoft/go-winio"

Gopkg.toml

@@ -7,7 +7,6 @@ deps:
 	dep ensure -vendor-only
 
 gen: ./osquery.thrift
-	rm -rf ./gen
 	mkdir ./gen
 	thrift --gen go:package_prefix=github.com/kolide/osquery-go/gen/ -out ./gen ./osquery.thrift
 	rm -rf gen/osquery/extension-remote gen/osquery/extension_manager-remote
@@ -36,4 +35,7 @@ example_config: examples/config/*.go
 test: all
 	go test -race -cover -v $(shell go list ./... | grep -v /vendor/)
 
+clean:
+	rm -rf ./build ./gen
+
 .PHONY: all

Makefile

@@ -1,6 +1,7 @@
 package osquery
 
 import (
+	"context"
 	"time"
 
 	"github.com/kolide/osquery-go/gen/osquery"
@@ -54,34 +55,34 @@ func (c *ExtensionManagerClient) Close() {
 
 // Ping requests metadata from the extension manager.
 func (c *ExtensionManagerClient) Ping() (*osquery.ExtensionStatus, error) {
-	return c.Client.Ping()
+	return c.Client.Ping(context.Background())
 }
 
 // Call requests a call to an extension (or core) registry plugin.
 func (c *ExtensionManagerClient) Call(registry, item string, request osquery.ExtensionPluginRequest) (*osquery.ExtensionResponse, error) {
-	return c.Client.Call(registry, item, request)
+	return c.Client.Call(context.Background(), registry, item, request)
 }
 
 // Extensions requests the list of active registered extensions.
 func (c *ExtensionManagerClient) Extensions() (osquery.InternalExtensionList, error) {
-	return c.Client.Extensions()
+	return c.Client.Extensions(context.Background())
 }
 
 // RegisterExtension registers the extension plugins with the osquery process.
 func (c *ExtensionManagerClient) RegisterExtension(info *osquery.InternalExtensionInfo, registry osquery.ExtensionRegistry) (*osquery.ExtensionStatus, error) {
-	return c.Client.RegisterExtension(info, registry)
+	return c.Client.RegisterExtension(context.Background(), info, registry)
 }
 
 // Options requests the list of bootstrap or configuration options.
 func (c *ExtensionManagerClient) Options() (osquery.InternalOptionList, error) {
-	return c.Client.Options()
+	return c.Client.Options(context.Background())
 }
 
 // Query requests a query to be run and returns the extension response.
 // Consider using the QueryRow or QueryRows helpers for a more friendly
 // interface.
 func (c *ExtensionManagerClient) Query(sql string) (*osquery.ExtensionResponse, error) {
-	return c.Client.Query(sql)
+	return c.Client.Query(context.Background(), sql)
 }
 
 // QueryRows is a helper that executes the requested query and returns the
@@ -117,5 +118,5 @@ func (c *ExtensionManagerClient) QueryRow(sql string) (map[string]string, error)
 
 // GetQueryColumns requests the columns returned by the parsed query.
 func (c *ExtensionManagerClient) GetQueryColumns(sql string) (*osquery.ExtensionResponse, error) {
-	return c.Client.GetQueryColumns(sql)
+	return c.Client.GetQueryColumns(context.Background(), sql)
 }

client.go

@@ -1,11 +1,12 @@
 package osquery
 
 import (
+	"context"
 	"errors"
 	"testing"
 
 	"github.com/kolide/osquery-go/gen/osquery"
-	"github.com/kolide/osquery-go/gen/osquery/mock"
+	"github.com/kolide/osquery-go/mock"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -14,7 +15,7 @@ func TestQueryRows(t *testing.T) {
 	client := &ExtensionManagerClient{Client: mock}
 
 	// Transport related error
-	mock.QueryFunc = func(sql string) (*osquery.ExtensionResponse, error) {
+	mock.QueryFunc = func(ctx context.Context, sql string) (*osquery.ExtensionResponse, error) {
 		return nil, errors.New("boom!")
 	}
 	rows, err := client.QueryRows("select 1")
@@ -23,7 +24,7 @@ func TestQueryRows(t *testing.T) {
 	assert.NotNil(t, err)
 
 	// Nil status
-	mock.QueryFunc = func(sql string) (*osquery.ExtensionResponse, error) {
+	mock.QueryFunc = func(ctx context.Context, sql string) (*osquery.ExtensionResponse, error) {
 		return &osquery.ExtensionResponse{}, nil
 	}
 	rows, err = client.QueryRows("select 1")
@@ -32,7 +33,7 @@ func TestQueryRows(t *testing.T) {
 	assert.NotNil(t, err)
 
 	// Query error
-	mock.QueryFunc = func(sql string) (*osquery.ExtensionResponse, error) {
+	mock.QueryFunc = func(ctx context.Context, sql string) (*osquery.ExtensionResponse, error) {
 		return &osquery.ExtensionResponse{
 			Status: &osquery.ExtensionStatus{Code: 1, Message: "bad query"},
 		}, nil
@@ -46,7 +47,7 @@ func TestQueryRows(t *testing.T) {
 	expectedRows := []map[string]string{
 		{"1": "1"},
 	}
-	mock.QueryFunc = func(sql string) (*osquery.ExtensionResponse, error) {
+	mock.QueryFunc = func(ctx context.Context, sql string) (*osquery.ExtensionResponse, error) {
 		return &osquery.ExtensionResponse{
 			Status:   &osquery.ExtensionStatus{Code: 0, Message: "OK"},
 			Response: expectedRows,
@@ -64,7 +65,7 @@ func TestQueryRows(t *testing.T) {
 		{"1": "1"},
 		{"1": "2"},
 	}
-	mock.QueryFunc = func(sql string) (*osquery.ExtensionResponse, error) {
+	mock.QueryFunc = func(ctx context.Context, sql string) (*osquery.ExtensionResponse, error) {
 		return &osquery.ExtensionResponse{
 			Status:   &osquery.ExtensionStatus{Code: 0, Message: "OK"},
 			Response: expectedRows,