Method IncrementCounter

internal/storage/redis/cache.go:62–101

(keyId string, timeUnit key.TimeUnit, incr int64)

}

func (c *Cache) IncrementCounter(keyId string, timeUnit key.TimeUnit, incr int64) error {
	ctxTimeout, cancel := context.WithTimeout(context.Background(), c.wt)
	defer cancel()

	ts, err := getCounterTimeStamp(timeUnit)
	if err != nil {
		return err
	}

	err = c.client.HIncrBy(ctxTimeout, keyId, strconv.FormatInt(ts, 10), incr).Err()
	if err != nil {
		return err
	}

	ctxTimeout, cancel = context.WithTimeout(context.Background(), c.rt)
	defer cancel()
	dur := c.client.TTL(ctxTimeout, keyId)
	err = dur.Err()
	if err != nil {
		return err
	}

	val := dur.Val()
	if val < 0 {
		ttl, err := getCounterTtl(timeUnit)
		if err != nil {
			return err
		}

		ctxTimeout, cancel = context.WithTimeout(context.Background(), c.wt)
		defer cancel()
		err = c.client.ExpireAt(ctxTimeout, keyId, ttl).Err()
		if err != nil {
			return err
		}

	}

	return nil
}

func getCounterTtl(rateLimitUnit key.TimeUnit) (time.Time, error) {
	now := time.Now().UTC()