Add SendWithContext to send an FCM request with a context.

fcm.go

@@ -2,6 +2,7 @@ package fcm
 
 import (
 	"bytes"
+	"context"
 	"encoding/json"
 	"fmt"
 	"io/ioutil"
@@ -154,8 +155,7 @@ func (this *FcmClient) apiKeyHeader() string {
 }
 
 // sendOnce send a single request to fcm
-func (this *FcmClient) sendOnce() (*FcmResponseStatus, error) {
-
+func (this *FcmClient) sendOnce(ctx context.Context) (*FcmResponseStatus, error) {
 	fcmRespStatus := new(FcmResponseStatus)
 
 	jsonByte, err := this.Message.toJsonByte()
@@ -167,6 +167,10 @@ func (this *FcmClient) sendOnce() (*FcmResponseStatus, error) {
 	request.Header.Set("Authorization", this.apiKeyHeader())
 	request.Header.Set("Content-Type", "application/json")
 
+	if ctx != nil {
+		request = request.WithContext(ctx)
+	}
+
 	client := &http.Client{}
 	response, err := client.Do(request)
 
@@ -199,8 +203,13 @@ func (this *FcmClient) sendOnce() (*FcmResponseStatus, error) {
 
 // Send to fcm
 func (this *FcmClient) Send() (*FcmResponseStatus, error) {
-	return this.sendOnce()
+	return this.sendOnce(nil)
+
+}
 
+// SendWithContext sends the request to FCM with a context
+func (this *FcmClient) SendWithContext(ctx context.Context) (*FcmResponseStatus, error) {
+	return this.sendOnce(ctx)
 }
 
 // toJsonByte converts FcmMsg to a json byte

fcm_test.go

@@ -1,6 +1,7 @@
 package fcm
 
 import (
+	"context"
 	"fmt"
 	"net/http"
 	"net/http/httptest"
@@ -159,6 +160,42 @@ func TestRegIdHandle_2(t *testing.T) {
 	}
 }
 
+func TestSendWithContext(t *testing.T) {
+
+	srv := httptest.NewServer(http.HandlerFunc(regIdHandle))
+	chgUrl(srv)
+	defer srv.Close()
+
+	c := NewFcmClient("key")
+
+	data := map[string]string{
+		"msg": "Hello World",
+		"sum": "Happy Day",
+	}
+
+	c.NewFcmMsgTo("/topics/topicName", data)
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	res, err := c.SendWithContext(ctx)
+	if err != nil {
+		t.Error("Response Error : ", err)
+	}
+	if res == nil {
+		t.Error("Res is nil")
+	}
+
+	// After the cancellation the request is expected to fail.
+	cancel()
+
+	_, err = c.SendWithContext(ctx)
+	if err == nil {
+		t.Errorf("expected context error, got %v", err)
+	}
+
+}
+
 func chgUrl(ts *httptest.Server) {
 	fcmServerUrl = ts.URL
 }