@Entity
@Data
@EqualsAndHashCode(callSuper=false)
public class CountryEntity  extends BaseEntity{

    private String countryCode;
    private String country;

    @OneToMany(mappedBy = "country", orphanRemoval = true, cascade = CascadeType.ALL)
    private Set<StateEntity> stateEntities;
}

@Entity
@Data
@EqualsAndHashCode(callSuper=false)
public class StateEntity extends BaseEntity {

    private String stateCode;
    private String state;

    @ManyToOne
    private Country country;
}

@MappedSuperclass
@Data
public abstract class BaseEntity {

    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private Long id;

    private Date createdDateTime;
    private Date updatedDateTime;
    private Boolean isActive;
    private Boolean isDelete;

    @PrePersist
    private void setCreatedDateTime() {
        this.createdDateTime = new Date();
        this.isActive = true;
        this.isDelete = false;
    }

    @PreUpdate
    private void setUpdatedDateTime() {
        this.updatedDateTime = new Date();
    }
}