Fea, 添加StringView::SplitAndTakeFirst

Author: mingkuang-Chuyu

include/YY/Base/Strings/StringView.h

@@ -228,7 +228,7 @@ namespace YY
                     }
                 }
 
-                size_t __YYAPI Find(char_t _ch, size_t _uIndex = 0) noexcept
+                size_t __YYAPI Find(char_t _ch, size_t _uIndex = 0) const noexcept
                 {
                     for (; _uIndex < GetSize(); ++_uIndex)
                     {
@@ -241,6 +241,30 @@ namespace YY
                     return -1;
                 }
 
+                size_t __YYAPI Find(StringView _sStr, size_t _uIndex = 0) const noexcept
+                {
+                    if (_sStr.IsEmpty())
+                        return -1;
+
+                    if (_uIndex + _sStr.GetLength() < GetSize())
+                    {
+                        return size_t(-1);
+                    }
+
+                    auto _sStart = sString + _uIndex;
+                    const auto _sEnd = sString + GetSize() - _sStr.GetLength();
+                    const auto _cbCmp = _sStr.GetSize() * sizeof(_sStr[0]);
+                    for (; _sStart <= _sEnd; ++_sStart)
+                    {
+                        if (memcmp(_sStart, _sStr.GetConstString(), _cbCmp) == 0)
+                        {
+                            return _uIndex;
+                        }
+                    }
+
+                    return -1;
+                }
+
                 StringView& __YYAPI Slice(size_t _uRemoveStart, size_t _uRemoveEnd = 0u) noexcept
                 {
                     if (_uRemoveStart + _uRemoveEnd >= cchString)
@@ -324,6 +348,56 @@ namespace YY
                     TrimEnd(_sTrimChars);
                     return *this;
                 }
+
+                StringView __YYAPI SplitAndTakeFirst(_In_ char_t _chSplit, _Out_opt_ StringView* _psRemaining = nullptr) const
+                {
+                    auto _uIndex = Find(_chSplit);
+                    if (_uIndex == size_t(-1))
+                    {
+                        StringView _sResult = *this;
+                        if (_psRemaining)
+                        {
+                            *_psRemaining = StringView();
+                        }
+
+                        return _sResult;
+                    }
+                    else
+                    {
+                        StringView _sResult = Substring(0, _uIndex);;
+                        if (_psRemaining)
+                        {
+                            *_psRemaining = Substring(_uIndex + 1);
+                        }
+
+                        return _sResult;
+                    }
+                }
+
+                StringView __YYAPI SplitAndTakeFirst(_In_ StringView _sSplit, _Out_opt_ StringView* _psRemaining = nullptr) const
+                {
+                    auto _uIndex = Find(_sSplit);
+                    if (_uIndex == size_t(-1))
+                    {
+                        StringView _sResult = *this;
+                        if (_psRemaining)
+                        {
+                            *_psRemaining = StringView();
+                        }
+
+                        return _sResult;
+                    }
+                    else
+                    {
+                        StringView _sResult = Substring(0, _uIndex);;
+                        if (_psRemaining)
+                        {
+                            *_psRemaining = Substring(_uIndex + _sSplit.GetLength());
+                        }
+
+                        return _sResult;
+                    }
+                }
             };
 
             template<>
@@ -529,7 +603,7 @@ namespace YY
                     }
                 }
 
-                size_t __YYAPI Find(char_t _ch, size_t _uIndex = 0) noexcept
+                size_t __YYAPI Find(char_t _ch, size_t _uIndex = 0) const noexcept
                 {
                     for (; _uIndex < GetSize(); ++_uIndex)
                     {
@@ -542,6 +616,30 @@ namespace YY
                     return -1;
                 }
 
+                size_t __YYAPI Find(StringView _sStr, size_t _uIndex = 0) const noexcept
+                {
+                    if (_sStr.IsEmpty())
+                        return -1;
+
+                    if (_uIndex + _sStr.GetLength() < GetSize())
+                    {
+                        return size_t(-1);
+                    }
+
+                    auto _sStart = sString + _uIndex;
+                    const auto _sEnd = sString + GetSize() - _sStr.GetLength();
+                    const auto _cbCmp = _sStr.GetSize() * sizeof(_sStr[0]);
+                    for (; _sStart <= _sEnd; ++_sStart)
+                    {
+                        if (memcmp(_sStart, _sStr.GetConstString(), _cbCmp) == 0)
+                        {
+                            return _uIndex;
+                        }
+                    }
+
+                    return -1;
+                }
+
                 StringView& __YYAPI Slice(size_t _uRemoveStart, size_t _uRemoveEnd = 0u) noexcept
                 {
                     if (_uRemoveStart + _uRemoveEnd >= cchString)
@@ -622,6 +720,56 @@ namespace YY
                     TrimEnd(_sTrimChars);
                     return *this;
                 }
+
+                StringView __YYAPI SplitAndTakeFirst(_In_ char_t _chSplit, _Out_opt_ StringView* _psRemaining = nullptr) const
+                {
+                    auto _uIndex = Find(_chSplit);
+                    if (_uIndex == size_t(-1))
+                    {
+                        StringView _sResult = *this;
+                        if (_psRemaining)
+                        {
+                            *_psRemaining = StringView();
+                        }
+
+                        return _sResult;
+                    }
+                    else
+                    {
+                        StringView _sResult = Substring(0, _uIndex);;
+                        if (_psRemaining)
+                        {
+                            *_psRemaining = Substring(_uIndex + 1);
+                        }
+
+                        return _sResult;
+                    }
+                }
+
+                StringView __YYAPI SplitAndTakeFirst(_In_ StringView _sSplit, _Out_opt_ StringView* _psRemaining = nullptr) const
+                {
+                    auto _uIndex = Find(_sSplit);
+                    if (_uIndex == size_t(-1))
+                    {
+                        StringView _sResult = *this;
+                        if (_psRemaining)
+                        {
+                            *_psRemaining = StringView();
+                        }
+
+                        return _sResult;
+                    }
+                    else
+                    {
+                        StringView _sResult = Substring(0, _uIndex);;
+                        if (_psRemaining)
+                        {
+                            *_psRemaining = Substring(_uIndex + _sSplit.GetLength());
+                        }
+
+                        return _sResult;
+                    }
+                }
             };
 
             typedef StringView<YY::Base::achar_t, Encoding::ANSI> aStringView;