The systemd configuration file should allow to use all the available configuration options for wildfly just as the init one does. It'll probably be a good idea to base it on the one provided by wildlfy itself.