Function set

apps/sim/lib/uploads/utils/user-file-base64.server.ts:219–278

(file: UserFile, value: string, ttlSeconds: number)

      }
    },
    async set(file: UserFile, value: string, ttlSeconds: number) {
      const key = getFullCacheKey(executionId, file)
      const valueBytes = Buffer.byteLength(value, 'utf8')
      try {
        if (!executionId) {
          await redis.set(key, value, 'EX', ttlSeconds)
          return
        }

        const limits = getExecutionRedisBudgetLimits()
        if (valueBytes > limits.maxSingleWriteBytes) {
          throw new ExecutionResourceLimitError({
            resource: 'redis_key_bytes',
            attemptedBytes: valueBytes,
            limitBytes: limits.maxSingleWriteBytes,
          })
        }
        const cacheTtlSeconds = Math.max(ttlSeconds, limits.ttlSeconds)
        const budgetReservation: ExecutionRedisBudgetReservation = {
          executionId,
          userId: options.userId,
          category: 'base64_cache',
          operation: 'set_base64_cache',
          bytes: valueBytes,
          logger,
        }
        const budgetKeys = getExecutionRedisBudgetKeys(budgetReservation)
        const result = (await redis.eval(
          SET_BASE64_CACHE_SCRIPT,
          2 + budgetKeys.length,
          key,
          getBudgetIndexKey(executionId),
          ...budgetKeys,
          value,
          cacheTtlSeconds,
          getFileCacheKey(file),
          serializeBudgetEntry({ bytes: valueBytes, userId: options.userId }),
          valueBytes,
          limits.maxExecutionBytes,
          limits.maxUserBytes,
          limits.ttlSeconds
        )) as [number, string, number | string | null]
        const [allowed, resource, current] = result
        if (allowed !== 1) {
          throw new ExecutionResourceLimitError({
            resource:
              resource === 'user_redis_bytes' ? 'user_redis_bytes' : 'execution_redis_bytes',
            attemptedBytes: valueBytes,
            currentBytes: Number(current ?? 0),
            limitBytes:
              resource === 'user_redis_bytes' ? limits.maxUserBytes : limits.maxExecutionBytes,
          })
        }
      } catch (error) {
        if (isExecutionResourceLimitError(error)) {
          throw error
        }
        logger.warn(`[${options.requestId}] Redis set failed, skipping cache`, error)