I’d hope what's happening now is already enough for everyone to reevaluate those risks!
In any case, threatening to shut Apple down wouldn’t be a message that you can’t do business in China, just that everybody, no matter how big, needs to toe the line. Do that, and you won’t have any problems.