Написать функцию, которая принимает число и возвращает количество цифр в этом числе

int dig(int digit);

void main()
{

setlocale(LC_ALL, "Russian");

int a;
do
{

cout << "Эта программа показывает количество цифр в числе.\nПожалуйста введите число: ";
cin >> a;
cout << "\nКоличество цифр в числе - " << dig(a) << "\n";
cout << "Если хотите продолжить, нажмите 1\n";
cin >> a;

} while (a == 1);

}

int dig(int digit)
{

int res = 0;
while(digit > 0)
{

res++;
digit /= 10;

}
return res;

}