Fixed type conversion interface error

gaia-pipeline · Mar 6, 2018 · b660daf · b660daf
1 parent 424cca8
commit b660daf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/plugin/grpc.go b/plugin/grpc.go
@@ -11,8 +11,8 @@ import (
 // PluginGRPC is the Gaia plugin interface used for communication
 // with the plugin.
 type PluginGRPC interface {
-	GetJobs() (*proto.Plugin_GetJobsClient, error)
-	ExecuteJob(job *proto.Job) (*proto.Empty, error)
+	GetJobs() (proto.Plugin_GetJobsClient, error)
+	ExecuteJob(job *proto.Job) (*proto.JobResult, error)
 }
 
 // GRPCClient represents gRPC client

diff --git a/plugin/plugin.go b/plugin/plugin.go
@@ -11,7 +11,7 @@ import (
 )
 
 const (
-	pluginMapKey = "plugin"
+	pluginMapKey = "Plugin"
 )
 
 var handshake = plugin.HandshakeConfig{
@@ -96,7 +96,7 @@ func (p *Plugin) GetJobs() ([]gaia.Job, error) {
 
 	// receive all jobs
 	for {
-		job, err := (*stream).Recv()
+		job, err := stream.Recv()
 
 		// Got all jobs
 		if err == io.EOF {