protocol/protocol-modbus/src/main/java/wei/yigulu/modbus/utils/ModbusCommandDataUtils.java

package wei.yigulu.modbus.utils;

import com.google.common.primitives.Bytes;
import lombok.extern.slf4j.Slf4j;
import wei.yigulu.modbus.domain.FunctionCode;
import wei.yigulu.modbus.domain.command.AbstractModbusCommand;
import wei.yigulu.modbus.domain.command.RtuModbusCommand;
import wei.yigulu.modbus.domain.command.TcpModbusCommand;
import wei.yigulu.modbus.domain.confirm.AbstractModbusConfirm;
import wei.yigulu.modbus.domain.confirm.RtuModbusConfirm;
import wei.yigulu.modbus.domain.confirm.TcpModbusConfirm;
import wei.yigulu.modbus.domain.datatype.RegisterValue;
import wei.yigulu.modbus.domain.tcpextracode.TransactionIdentifier;
import wei.yigulu.modbus.exceptiom.ModbusException;
import wei.yigulu.modbus.netty.ModbusMasterBuilderInterface;
import wei.yigulu.modbus.netty.ModbusRtuMasterBuilder;
import wei.yigulu.netty.AbstractMasterBuilder;
import wei.yigulu.netty.AbstractTcpMasterBuilder;
import wei.yigulu.netty.MasterInterface;

import java.nio.ByteBuffer;
import java.util.ArrayList;
import java.util.List;

/**
 * modbs下达数据命令的工具类
 *
 * @author: xiuwei
 * @version:
 */
@Slf4j
public class ModbusCommandDataUtils {


	/**
	 * 下达寄存器控制命令
	 *
	 * @param masterBuilder 主站对象
	 * @param address       起始地址
	 * @param values        命令值
	 */
	public static boolean commandRegister(MasterInterface masterBuilder, Integer slaveId, Integer address, List<RegisterValue> values) throws ModbusException {
		if (!(masterBuilder instanceof ModbusMasterBuilderInterface)) {
			throw new RuntimeException("请传人实现了<ModbusMasterBuilderInterface>的Master");
		}
		List<Byte> bs = new ArrayList<>();
		AbstractModbusCommand modbusCommand;
		ByteBuffer buffer;
		AbstractModbusConfirm confirm;
		try {
			if (masterBuilder instanceof ModbusRtuMasterBuilder) {
				modbusCommand = new RtuModbusCommand();
				modbusCommand.setSlaveId(slaveId).setRegisters(address, values);
				modbusCommand.encode(bs);
				masterBuilder.sendFrameToOpposite(Bytes.toArray(bs));
				buffer = ((ModbusMasterBuilderInterface) masterBuilder).getOrCreateSynchronousWaitingRoom().getData(0);
				confirm = new RtuModbusConfirm().decode(buffer);
			} else {
				modbusCommand = new TcpModbusCommand();
				((TcpModbusCommand) modbusCommand).setTransactionIdentifier(TransactionIdentifier.getInstance((AbstractTcpMasterBuilder) masterBuilder));
				modbusCommand.setSlaveId(slaveId).setRegisters(address, values);
				modbusCommand.encode(bs);
				masterBuilder.sendFrameToOpposite(Bytes.toArray(bs));
				buffer = ((ModbusMasterBuilderInterface) masterBuilder).getOrCreateSynchronousWaitingRoom().getData(((TcpModbusCommand) modbusCommand).getTcpExtraCode().getTransactionIdentifier().getSeq());
				confirm = new TcpModbusConfirm().decode(buffer);
			}
		} catch (ModbusException e) {
			log.error("控制命令执行失败:" + e.getMsg());
			return false;
		}
		if (address.equals(confirm.getStartAddress())) {
			if (confirm.getFunctionCode() == modbusCommand.getFunctionCode()) {
				if (confirm.getFunctionCode() == FunctionCode.WRITE_COIL || confirm.getFunctionCode() == FunctionCode.WRITE_REGISTER) {
					if (Bytes.indexOf(confirm.getB2(), modbusCommand.getDataBytes()) == 0) {
						return true;
					} else {
						log.warn("返回的确认值和输出值不同");
					}
				} else {
					if (confirm.getQuantity().equals(modbusCommand.getQuantity())) {
						return true;
					} else {
						log.warn("返回的值数量和输出数量不同");
					}
				}
			}
		}
		return false;
	}


	/**
	 * 下达寄存器控制命令
	 *
	 * @param masterBuilder 主站对象
	 * @param address       起始地址
	 * @param value         命令值
	 */
	public static boolean commandRegister(MasterInterface masterBuilder, Integer slaveId, Integer address, RegisterValue value) throws ModbusException {
		List<RegisterValue> list = new ArrayList<>();
		list.add(value);
		return commandRegister(masterBuilder, slaveId, address, list);
	}
}