@splitsoftware/splitio-react
Version:
A React library to easily integrate and use Split JS SDK
43 lines (42 loc) • 2.93 kB
JavaScript
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 });
}