As one of the most promising critical technologies for realizing 5G networks, mmWave communication technology has received more and more attention and research. At the same time, due to the integration and symbiosis of mmWave communication, massive MIMO, and ultra-dense network, recent organizations have also jointly proposed the rational introduction of these three technologies in mobile communication systems to improve spectrum efficiency and energy efficiency in 5G. This paper focuses on a single-cell multi-user mmWave massive MIMO system. To optimize the total data rate, we propose a beam allocation algorithm based on a one-to-one matching game under the condition that the number of RF chains is less than the number of users. The simulation results show that the proposed algorithm obtains almost the same performance as the exhaustive method on the premise that the algorithm complexity is lower than the thorough method.