mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
111 lines
3.2 KiB
Go
111 lines
3.2 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package featureflag
|
|
|
|
import (
|
|
"math"
|
|
"reflect"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/splitio/go-client/v6/splitio/client"
|
|
"github.com/splitio/go-client/v6/splitio/conf"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/hanzoteam/server/server/public/shared/mlog"
|
|
)
|
|
|
|
type SyncParams struct {
|
|
ServerID string
|
|
SplitKey string
|
|
SyncIntervalSeconds int
|
|
Log *mlog.Logger
|
|
Attributes map[string]any
|
|
}
|
|
|
|
type Synchronizer struct {
|
|
SyncParams
|
|
|
|
client *client.SplitClient
|
|
stop chan struct{}
|
|
stopped chan struct{}
|
|
}
|
|
|
|
var featureNames = getStructFields(model.FeatureFlags{})
|
|
|
|
func NewSynchronizer(params SyncParams) (*Synchronizer, error) {
|
|
cfg := conf.Default()
|
|
if params.Log != nil {
|
|
cfg.Logger = &splitLogger{wrappedLog: params.Log.With(mlog.String("service", "split"))}
|
|
} else {
|
|
cfg.LoggerConfig.LogLevel = math.MinInt32
|
|
}
|
|
factory, err := client.NewSplitFactory(params.SplitKey, cfg)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "unable to create split factory")
|
|
}
|
|
|
|
return &Synchronizer{
|
|
SyncParams: params,
|
|
client: factory.Client(),
|
|
stop: make(chan struct{}),
|
|
stopped: make(chan struct{}),
|
|
}, nil
|
|
}
|
|
|
|
// EnsureReady blocks until the synchronizer is ready to update feature flag values
|
|
func (f *Synchronizer) EnsureReady() error {
|
|
if err := f.client.BlockUntilReady(10); err != nil {
|
|
return errors.Wrap(err, "split.io client could not initialize")
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (f *Synchronizer) UpdateFeatureFlagValues(base model.FeatureFlags) model.FeatureFlags {
|
|
featuresMap := f.client.Treatments(f.ServerID, featureNames, f.Attributes)
|
|
ffm := featureFlagsFromMap(featuresMap, base)
|
|
return ffm
|
|
}
|
|
|
|
func (f *Synchronizer) Close() {
|
|
f.client.Destroy()
|
|
}
|
|
|
|
// featureFlagsFromMap sets the feature flags from a map[string]string.
|
|
// It starts with baseFeatureFlags and only sets values that are
|
|
// given by the upstream management system.
|
|
// Makes the assumption that all feature flags are strings or booleans.
|
|
// Strings are converted to booleans by considering case insensitive "on" or any value considered by strconv.ParseBool as true and any other value as false.
|
|
func featureFlagsFromMap(featuresMap map[string]string, baseFeatureFlags model.FeatureFlags) model.FeatureFlags {
|
|
refStruct := reflect.ValueOf(&baseFeatureFlags).Elem()
|
|
for fieldName, fieldValue := range featuresMap {
|
|
refField := refStruct.FieldByName(fieldName)
|
|
// "control" is returned by split.io if the treatment is not found, in this case we should use the default value.
|
|
if !refField.IsValid() || !refField.CanSet() || fieldValue == "control" {
|
|
continue
|
|
}
|
|
|
|
switch refField.Type().Kind() {
|
|
case reflect.Bool:
|
|
parsedBoolValue, _ := strconv.ParseBool(fieldValue)
|
|
refField.Set(reflect.ValueOf(strings.ToLower(fieldValue) == "on" || parsedBoolValue))
|
|
default:
|
|
refField.Set(reflect.ValueOf(fieldValue))
|
|
}
|
|
}
|
|
return baseFeatureFlags
|
|
}
|
|
|
|
func getStructFields(s any) []string {
|
|
structType := reflect.TypeOf(s)
|
|
fieldNames := make([]string, 0, structType.NumField())
|
|
for field := range structType.Fields() {
|
|
fieldNames = append(fieldNames, field.Name)
|
|
}
|
|
|
|
return fieldNames
|
|
}
|