chore: adds sorted set increment in incubating (#106)

incubating/simple_cache_client.go

@@ -32,8 +32,7 @@ type ScsClient interface {
 	SortedSetGetScore(ctx context.Context, request *SortedSetGetScoreRequest) (SortedSetGetScoreResponse, error)
 	SortedSetRemove(ctx context.Context, request *SortedSetRemoveRequest) error
 	SortedSetGetRank(ctx context.Context, request *SortedSetGetRankRequest) (SortedSetGetRankResponse, error)
-	// TODO need to impl sortedset increment still
-	//SortedSetIncrement(ctx context.Context, request *SortedSetIncrementRequest)
+	SortedSetIncrement(ctx context.Context, request *SortedSetIncrementRequest) (SortedSetIncrementResponse, error)
 
 	Close()
 }
@@ -448,6 +447,20 @@ func (c *DefaultScsClient) SortedSetGetRank(ctx context.Context, request *Sorted
 	}
 }
 
+func (c *DefaultScsClient) SortedSetIncrement(ctx context.Context, request *SortedSetIncrementRequest) (SortedSetIncrementResponse, error) {
+	rsp, err := c.dataClient.SortedSetIncrement(ctx, &models.SortedSetIncrementRequest{
+		CacheName:     request.CacheName,
+		SetName:       []byte(request.SetName),
+		ElementName:   request.ElementName.AsBytes(),
+		Amount:        request.Amount,
+		CollectionTTL: request.CollectionTTL,
+	})
+	if err != nil {
+		return nil, convertMomentoSvcErrorToCustomerError(err)
+	}
+	return &SortedSetIncrementResponseSuccess{Value: rsp.Value}, nil
+}
+
 // Close shutdown the client.
 func (c *DefaultScsClient) Close() {
 	defer c.internalClient.Close()

incubating/sortedset_requests.go

@@ -85,6 +85,6 @@ type SortedSetIncrementRequest struct {
 	CacheName     string
 	SetName       string
 	ElementName   momento.Bytes
-	Amount        uint64
+	Amount        float64
 	CollectionTTL utils.CollectionTTL
 }

incubating/sortedset_response.go

@@ -83,3 +83,12 @@ func (SortedSetRankHit) isSortedSetRankElement() {}
 type SortedSetRankMiss struct{}
 
 func (SortedSetRankMiss) isSortedSetRankElement() {}
+
+type SortedSetIncrementResponse interface {
+	isSortedSetIncrementResponse()
+}
+type SortedSetIncrementResponseSuccess struct {
+	Value float64
+}
+
+func (SortedSetIncrementResponseSuccess) isSortedSetIncrementResponse() {}

internal/models/sortedset_requests.go

@@ -79,6 +79,6 @@ type SortedSetIncrementRequest struct {
 	CacheName     string
 	SetName       []byte
 	ElementName   []byte
-	Amount        uint64
+	Amount        float64
 	CollectionTTL incubating.CollectionTTL
 }

internal/models/sortedset_response.go

@@ -60,3 +60,7 @@ type SortedSetGetRankHit struct {
 }
 
 func (SortedSetGetRankHit) isSortedSetGetRankResponse() {}
+
+type SortedSetIncrementResponse struct {
+	Value float64
+}

internal/services/scs_data_sorted_set_service.go

@@ -131,6 +131,27 @@ func (client *ScsDataClient) SortedSetGetRank(ctx context.Context, request *mode
 	}
 }
 
+func (client *ScsDataClient) SortedSetIncrement(ctx context.Context, request *models.SortedSetIncrementRequest) (*models.SortedSetIncrementResponse, momentoerrors.MomentoSvcErr) {
+	ctx, cancel := context.WithTimeout(ctx, client.requestTimeout)
+	defer cancel()
+	resp, err := client.grpcClient.SortedSetIncrement(
+		metadata.NewOutgoingContext(ctx, createNewMetadata(request.CacheName)),
+		&pb.XSortedSetIncrementRequest{
+			SetName:         request.SetName,
+			ElementName:     request.ElementName,
+			Amount:          request.Amount,
+			TtlMilliseconds: uint64(request.CollectionTTL.Ttl.Milliseconds()),
+			RefreshTtl:      request.CollectionTTL.RefreshTtl,
+		},
+	)
+	if err != nil {
+		return nil, momentoerrors.ConvertSvcErr(err)
+	}
+	return &models.SortedSetIncrementResponse{
+		Value: resp.Value,
+	}, nil
+}
+
 func (client *ScsDataClient) SortedSetRemove(ctx context.Context, request *models.SortedSetRemoveRequest) momentoerrors.MomentoSvcErr {
 	ctx, cancel := context.WithTimeout(ctx, client.requestTimeout)
 	defer cancel()