feat: add callback remove funcion

callback/callback.go

@@ -83,8 +83,9 @@ func (cb *Callback) HandleFunc(w http.ResponseWriter, r *http.Request) {
 	ctx = context.WithValue(ctx, internal.CallbackRetryCounterKey, retryCounter)
 
 	var (
-		code int
-		date time.Time
+		code   int
+		date   time.Time
+		remove bool
 	)
 
 	retryAfter := func(c int, d time.Time) {
@@ -93,13 +94,24 @@ func (cb *Callback) HandleFunc(w http.ResponseWriter, r *http.Request) {
 	}
 	ctx = context.WithValue(ctx, internal.CallbackRetryAfterKey, retryAfter)
 
+	removeFunc := func() {
+		remove = true
+	}
+	ctx = context.WithValue(ctx, internal.CallbackRemove, removeFunc)
+
 	if err := cb.Handler(ctx, e); err != nil {
 		cb.logf("callback: %v", err)
 		http.Error(w, "Bad Request", http.StatusBadRequest)
 
 		return
 	}
 
+	if remove {
+		_, _ = w.Write([]byte("remove"))
+
+		return
+	}
+
 	if code != 0 {
 		w.Header().Set("Retry-After", date.Format(http.TimeFormat)) // RFC 7231, 7.1.3
 		http.Error(w, http.StatusText(code), code)

callback/context.go

@@ -30,3 +30,8 @@ func RetryAfter(ctx context.Context, code int, date time.Time) {
 		date,
 	)
 }
+
+// Remove VK Callback server.
+func Remove(ctx context.Context) {
+	ctx.Value(internal.CallbackRemove).(func())()
+}

callback/context_test.go

@@ -57,3 +57,26 @@ func TestRetryAfter(t *testing.T) {
 	assert.Equal(t, code, rr.Code)
 	assert.Equal(t, date.Format(http.TimeFormat), rr.Header().Get("Retry-After"))
 }
+
+func TestRemove(t *testing.T) {
+	t.Parallel()
+
+	cb := callback.NewCallback()
+	cb.MessageNew(func(ctx context.Context, obj events.MessageNewObject) {
+		callback.Remove(ctx)
+	})
+
+	jsonStr := []byte(`{"type": "message_new","object": {}}`)
+
+	req, err := http.NewRequest("POST", "/callback", bytes.NewBuffer(jsonStr))
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	rr := httptest.NewRecorder()
+	handler := http.HandlerFunc(cb.HandleFunc)
+
+	handler.ServeHTTP(rr, req)
+
+	assert.Equal(t, "remove", rr.Body.String())
+}

internal/transport.go

@@ -27,6 +27,7 @@ const (
 	LongPollTsKey
 	CallbackRetryCounterKey
 	CallbackRetryAfterKey
+	CallbackRemove
 )
 
 // ContextClient return *http.Client.