rts-sim-module/repository/button.go

package repository

import "joylink.club/rtsssimulation/repository/model/proto"

type Button struct {
	Identity

	buttonType proto.Button_ButtonType
}

func NewButton(id string, buttonType proto.Button_ButtonType) *Button {
	return &Button{
		Identity:   identity{id, proto.DeviceType_DeviceType_Button},
		buttonType: buttonType,
	}
}

func (btn *Button) GetBtnType() proto.Button_ButtonType {
	return btn.buttonType
}