java使用talib计算股票的rsi，kdj等指标

写了一个talib工具类，因为自己只用到了这三个指标，所以只开发了这三个，给有需要的小伙伴一个参考。

maven：

<dependency>
         <groupId>com.tictactec</groupId>
         <artifactId>ta-lib</artifactId>
         <version>0.4.0</version>
</dependency>

实体类：

package cn.flashylife.vo;

import lombok.Getter;
import lombok.Setter;
import lombok.ToString;

import java.math.BigDecimal;

/**
 * 股票行情信息实体类（用来查询固定的字段）
 */
@Setter
@Getter
@ToString
public class StockBasicInfoVO {
    /**
     * 交易日期
     */
    private String sbiTradeDate;
    /**
     * 股票代码
     */
    private String sbiCode;
    /**
     * 最新价(收盘价)
     */
    private BigDecimal sbiClose;
    /**
     * 最高价
     */
    private BigDecimal sbiHigh;
    /**
     * 最低价
     */
    private BigDecimal sbiLow;
}

工具类：
因为自己调用前已经对非空，非null做过判断，使用时需注意。

package cn.flashylife.util;

import cn.flashylife.vo.StockBasicInfoVO;
import com.tictactec.ta.lib.Core;
import com.tictactec.ta.lib.MAType;
import com.tictactec.ta.lib.MInteger;
import com.tictactec.ta.lib.RetCode;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * @date 2022年3月29日13:50:24
 * @Description
 *      股票指标工具类
 *      用来计算各种指标(如：ma，rsi，kdj)
 * @author flashylife
 */
public class StockIndexUtils {

    /**
     * 禁止初始化
     */
    private StockIndexUtils() {}

    /**
     * 获取ma
     *
     * @param value ma的参数(5，10,15,20等)
     * @return
     */
    public static double[] ma(int value, List<StockBasicInfoVO> list) {
        if (value <= 0) {
            throw new RuntimeException("天数不能小于0");
        }
        // 收盘价
        double[] inClose = new double[list.size()];
        // 返回的数据最后value + 1没有值
        double[] output = new double[list.size() - value + 1];

        for (int i = 0; i < list.size(); i++) {
            inClose[i] = list.get(i).getSbiClose().doubleValue();
        }

        Core core = new Core();

        // ma 需要四舍五入
        RetCode code = core.sma(0, inClose.length - 1, inClose, value, new MInteger(), new MInteger(), output);

        if (code == RetCode.Success) {
            output = Arrays.stream(output)
                    .map(e -> new BigDecimal(e)
                            .setScale(2, BigDecimal.ROUND_HALF_UP).doubleValue())
                    .toArray();
        }
        return output;
    }

    /**
     * 获取rsi
     * @param value rsi的参数(6,12,24等)
     * @return
     */
    public static double[] rsi(int value, List<StockBasicInfoVO> list) {
        // 收盘价
        double[] inClose = new double[list.size()];
        // 返回的数据最后value没有值
        double[] output = new double[list.size() - value];

        for (int i = 0; i < list.size(); i++) {
            inClose[i] = list.get(i).getSbiClose().doubleValue();
        }

        Core core = new Core();

        // rsi 直接截取
        RetCode code = core.rsi(0, inClose.length - 1, inClose, value, new MInteger(), new MInteger(), output);

        if (code == RetCode.Success) {
            output = Arrays.stream(output)
                    .map(e -> new BigDecimal(e)
                            .setScale(2, BigDecimal.ROUND_DOWN).doubleValue())
                    .toArray();
        }
        return output;
    }

    /**
     * 获取kdj(使用默认的9,3,3)
     *
     * @return
     */
    public static List kdj(List<StockBasicInfoVO> list) {
        // 最高
        double[] inHigh = new double[list.size()];
        // 收盘
        double[] inClose = new double[list.size()];
        // 最低
        double[] inLow = new double[list.size()];

        // 输出的k（最后16位没有值）
        double[] k = new double[list.size() - 16];
        // 输出的d（最后16位没有值）
        double[] d = new double[list.size() - 16];
        // 手动计算j值(3*k - 2*d)
        double[] j = new double[list.size() - 16];

        for (int i = 0; i < list.size(); i++) {
            inHigh[i] = list.get(i).getSbiHigh().doubleValue();
            inClose[i] = list.get(i).getSbiClose().doubleValue();
            inLow[i] = list.get(i).getSbiLow().doubleValue();
        }

        List kdjList = null;

        Core core = new Core();

        // kd 直接截取
        RetCode code = core.stoch(0, inHigh.length - 1, inHigh, inLow, inClose, 9, 5, MAType.Ema, 5, MAType.Ema, new MInteger(), new MInteger(), k, d);

        if (code == RetCode.Success) {
            // 计算j值(保留2位)
            for (int i = 0; i < d.length; i++) {
                BigDecimal b1 = new BigDecimal(k[i]).multiply(new BigDecimal(3));
                BigDecimal b2 = new BigDecimal(d[i]).multiply(new BigDecimal(2));
                j[i] = b1.subtract(b2).setScale(2, BigDecimal.ROUND_DOWN).doubleValue();
            }

            // 计算k值(保留2位)
            k = Arrays.stream(k)
                    .map(e -> new BigDecimal(e)
                            .setScale(2, BigDecimal.ROUND_DOWN).doubleValue())
                    .toArray();

            // 计算d值(保留2位)
            d = Arrays.stream(d)
                    .map(e -> new BigDecimal(e)
                            .setScale(2, BigDecimal.ROUND_DOWN).doubleValue())
                    .toArray();

            kdjList = new ArrayList();
            kdjList.add(k);
            kdjList.add(d);
            kdjList.add(j);
        }
        return kdjList == null ? Collections.emptyList() : kdjList;
    }
}