githublaohu commented on code in PR #803:
URL: https://github.com/apache/rocketmq-client-go/pull/803#discussion_r851025843
##########
internal/request_response_future.go:
##########
@@ -0,0 +1,135 @@
+package internal
+
+import (
+ "context"
+ "fmt"
+ "sync"
+ "time"
+
+ "github.com/patrickmn/go-cache"
+ "github.com/pkg/errors"
+
+ "github.com/apache/rocketmq-client-go/v2/primitive"
+ "github.com/apache/rocketmq-client-go/v2/rlog"
+)
+
+var RequestResponseFutureMap = NewRequestResponseFutureMap()
+
+type requestResponseFutureCache struct {
+ cache *cache.Cache
+}
+
+func NewRequestResponseFutureMap() *requestResponseFutureCache {
+ tmpRrfCache := requestResponseFutureCache{
+ cache: cache.New(5*time.Minute, 10*time.Minute),
+ }
+
+ // OnEvicted delete the timeout RequestResponseFuture, trigger set the
failure cause.
+ tmpRrfCache.cache.OnEvicted(func(s string, i interface{}) {
+ rrf, ok := i.(*RequestResponseFuture)
+ if !ok {
+ rlog.Error("convert i to RequestResponseFuture err",
map[string]interface{}{
+ "correlationId": s,
+ })
+ return
+ }
+ if !rrf.IsTimeout() {
+ return
+ }
+
+ err := fmt.Errorf("correlationId:%s request timeout, no reply
message", s)
+ rrf.CauseErr = err
+ rrf.ExecuteRequestCallback()
+ })
+ return &tmpRrfCache
+}
+
+// SetRequestResponseFuture set rrf to map
+func (fm *requestResponseFutureCache) SetRequestResponseFuture(rrf
*RequestResponseFuture) {
+ fm.cache.Set(rrf.CorrelationId, rrf, rrf.Timeout)
+}
+
+// SetResponseToRequestResponseFuture set reply to rrf
+func (fm *requestResponseFutureCache)
SetResponseToRequestResponseFuture(correlationId string, reply
*primitive.Message) error {
+ rrf, exist := fm.RequestResponseFuture(correlationId)
+ if !exist {
+ return errors.Wrapf(nil, "correlationId:%s not exist in map",
correlationId)
+ }
+ rrf.PutResponseMessage(reply)
+ if rrf.RequestCallback != nil {
+ rrf.ExecuteRequestCallback()
+ }
+ return nil
+}
+
+// RequestResponseFuture get rrf from map by the CorrelationId
+func (fm *requestResponseFutureCache) RequestResponseFuture(correlationId
string) (*RequestResponseFuture, bool) {
+ res, exists := fm.cache.Get(correlationId)
+ if exists {
+ return res.(*RequestResponseFuture), exists
+ }
+ return nil, exists
+}
+
+// RemoveRequestResponseFuture remove the rrf from map
+func (fm *requestResponseFutureCache)
RemoveRequestResponseFuture(correlationId string) {
+ fm.cache.Delete(correlationId)
Review Comment:
删除的时候，一定要做这些行为。因为异常行为，都会调用删除方法。如果不写到这里，每个点都需要做类似的处理。
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
