推荐算法单层AB实验的实现

message RecommendParam{
    optional bool disablePersonalRecFeature = 1;
    optional bool disablePersonalRuleFeature = 2; 
    optional bool disableFeedbackFeature = 3;
    optional int32 sourceTypeDiversifyTunerWebLimit = 4 [default = 2]; 
    optional int32 sourceTypeDiversifyTunerCycle = 5 [default = 10]; 
    optional bool enableSimilarTuner = 6 [default = true]; 
    optional double similarTunerThreshold = 7 [default = 1.0];
    optional int32 similarTunerCycle = 8 [default = 10]; 


}


message BucketGroup {
  optional string bucket = 1;
  optional int32 startSlot = 2;  //起始slot
  optional int32 endBuSlot = 3;  //终止slot
  optional RecommendParam param = 4; //实验参数
}


message Experiment {
  optional string name = 1;    //实验名称
  optional string algorithm = 2;  //算法标识
  optional bool enabled = 3;   //是否启用
  repeated BucketGroup bucketGroup = 4;
}


message Experiments {
  repeated Experiment experiments = 1; // 实验列表
}

public class ExperimentService {


    private final static int BUCKET_MAX_COUNT = 1000;
    /**
     * 桶实验参数
     */
    private Experiments experiments;


    @PostConstruct
    void postConsruct() {
        loadExperimentFile("experiments.conf");
    }


    /**
     * 解析实验配置文件
     *
     * @param filePath
     */
    public void loadExperimentFile(String filePath) {


        try {
            // 读取文件
            String content = FileUtil.readFileFromClasspath(filePath);
            Experiments.Builder builder = Experiments.newBuilder();
            JsonFormat.parser().ignoringUnknownFields().merge(content, builder);
            this.experiments = builder.build();
            log.info("success to update experiment file ");
        } catch (Exception e) {
            log.error("experiments get error: " + e.getMessage());
        }
    }


    /**
     * 获取slot编号
     *
     * @param auid size
     * @return
     */
    protected int getBucket(String auid, int size) {
        int hashCode = auid.hashCode();
        return Math.abs(hashCode) % size;
    }


    /**
     * 根据算法获取命中实验
     *
     * @param action
     * @return
     */
    protected Experiment getExperiment(String algorithm) {
        Experiment result = Experiment.getDefaultInstance();
        if (experiments != null) {
            for (Experiment experiment : experiments.getExperimentsList()) {
                if (experiment.getAlgorithm().equals(algorithm)) {
                    result = experiment;
                }
            }
        }
        return result;
    }


    /**
     * 获取命中实验组
     *
     * @param request
     * @return
     */
    public BucketGroup getBucketGroup(String algorithm, String auid) {
        Experiment experiment = getExperiment(algorithm);
        if (experiment == null || !experiment.getEnabled()) {
            return BucketGroup.getDefaultInstance();
        }


        int bucket = getBucket(auid, BUCKET_MAX_COUNT);
        for (BucketGroup bucketGroup : experiment.getBucketGroupList()) {
            if (bucketGroup.getStartBucket() <= bucket && bucket <= bucketGroup.getEndBucket()) {
                return bucketGroup;
            }
        }


        return BucketGroup.getDefaultInstance();
    }
}