UNPKG

@splitsoftware/splitio-react

Version:

A React library to easily integrate and use Split JS SDK

43 lines (42 loc) 2.93 kB
import { __assign } from "tslib"; import * as React from 'react'; import memoizeOne from 'memoize-one'; import { argsAreEqual, getTreatments } from './utils'; import { useSplitClient } from './useSplitClient'; function evaluateFeatureFlags(client, _lastUpdate, names, attributes, _clientAttributes, flagSets, options, factory) { return client && client.getStatus().isOperational && (names || flagSets) ? names ? client.getTreatments(names, attributes, options) : client.getTreatmentsByFlagSets(flagSets, attributes, options) : names ? getTreatments(names, false, factory) : {}; // empty object when evaluating with flag sets and client is not ready } export function memoizeGetTreatments() { return memoizeOne(evaluateFeatureFlags, argsAreEqual); } /** * `useTreatments` is a hook that returns an Split Context object extended with a `treatments` property object that contains feature flag evaluations. * It uses the `useSplitClient` hook to access the client, and invokes the `client.getTreatments()` method if the `names` option is provided, * or the `client.getTreatmentsByFlagSets()` method if the `flagSets` option is provided. * * @param options - An options object with a list of feature flag names or flag sets to evaluate, and an optional `attributes` and `splitKey` values to configure the client. * @returns A Split Context object extended with a Treatments instance, that might contain control treatments if the client is not available or ready, or if feature flag names do not exist. * * @example * ```js * const { treatments: { feature_1, feature_2 }, isReady, isReadyFromCache, hasTimedout, lastUpdate, ... } = useTreatments({ names: ['feature_1', 'feature_2']}); * ``` * * @see {@link https://developer.harness.io/docs/feature-management-experimentation/sdks-and-infrastructure/client-side-sdks/javascript-sdk/#multiple-evaluations-at-once} */ export function useTreatments(options) { var context = useSplitClient(__assign(__assign({}, options), { attributes: undefined })); var factory = context.factory, client = context.client, lastUpdate = context.lastUpdate; var names = options.names, flagSets = options.flagSets, attributes = options.attributes, properties = options.properties; var getTreatments = React.useMemo(memoizeGetTreatments, []); // Shallow copy `client.getAttributes` result for memoization, as it returns the same reference unless `client.clearAttributes` is invoked. // Note: the same issue occurs with the `names` and `attributes` arguments if they are mutated directly by the user instead of providing a new object. var treatments = getTreatments(client, lastUpdate, names, attributes, client ? __assign({}, client.getAttributes()) : {}, flagSets, properties && { properties: properties }, factory); return __assign(__assign({}, context), { treatments: treatments }); }