Explorar o código

pkg/plugins/client.go: don't try to encode os decode if it's nil

When user call the `Call()` method, they don't always want to sent
some args or get the return value, so they use `nil` when call `Call()`
method and this will casue an error. It's better to not trying to
encode or decode if it's nil.

Signed-off-by: Lei Jitang <leijitang@huawei.com>
Lei Jitang %!s(int64=9) %!d(string=hai) anos
pai
achega
a2d348968f
Modificáronse 2 ficheiros con 13 adicións e 5 borrados
  1. 9 5
      pkg/plugins/client.go
  2. 4 0
      pkg/plugins/client_test.go

+ 9 - 5
pkg/plugins/client.go

@@ -60,17 +60,21 @@ type Client struct {
 // It will retry for 30 seconds if a failure occurs when calling.
 func (c *Client) Call(serviceMethod string, args interface{}, ret interface{}) error {
 	var buf bytes.Buffer
-	if err := json.NewEncoder(&buf).Encode(args); err != nil {
-		return err
+	if args != nil {
+		if err := json.NewEncoder(&buf).Encode(args); err != nil {
+			return err
+		}
 	}
 	body, err := c.callWithRetry(serviceMethod, &buf, true)
 	if err != nil {
 		return err
 	}
 	defer body.Close()
-	if err := json.NewDecoder(body).Decode(&ret); err != nil {
-		logrus.Errorf("%s: error reading plugin resp: %v", serviceMethod, err)
-		return err
+	if ret != nil {
+		if err := json.NewDecoder(body).Decode(&ret); err != nil {
+			logrus.Errorf("%s: error reading plugin resp: %v", serviceMethod, err)
+			return err
+		}
 	}
 	return nil
 }

+ 4 - 0
pkg/plugins/client_test.go

@@ -63,6 +63,10 @@ func TestEchoInputOutput(t *testing.T) {
 	if !reflect.DeepEqual(output, m) {
 		t.Fatalf("Expected %v, was %v\n", m, output)
 	}
+	err = c.Call("Test.Echo", nil, nil)
+	if err != nil {
+		t.Fatal(err)
+	}
 }
 
 func TestBackoff(t *testing.T) {