只有小米来源得用户，才走亿米

application/api/controller/User.php

@@ -131,8 +131,10 @@ class User extends Api
             ];
             $ret = $this->auth->register($mobile, Random::alnum(), '', $mobile, $extend);
             //亿米
-            $ua = input('ua','','trim');
-            $this->yimi_advert($ua);
+            if(input('register_from','') == 'xiaomi'){
+                $ua = input('ua','','trim');
+                $this->yimi_advert($ua);
+            }
         }
         if ($ret) {
             Sms::flush($mobile, 'mobilelogin');
@@ -221,8 +223,10 @@ class User extends Api
             ];
             $ret = $this->auth->openid_register($wechat_openid,$reg_data);
             //亿米
-            $ua = input('ua','','trim');
-            $this->yimi_advert($ua);
+            if(input('register_from','') == 'xiaomi'){
+                $ua = input('ua','','trim');
+                $this->yimi_advert($ua);
+            }
         }
         if ($ret) {
             $data = $this->userInfo('return');
@@ -293,8 +297,10 @@ class User extends Api
                     $ret = $this->auth->register($mobile, Random::alnum(), '', $mobile, $extend);
                     $is_register = 1;
                     //亿米
-                    $ua = input('ua','','trim');
-                    $this->yimi_advert($ua);
+                    if(input('register_from','') == 'xiaomi'){
+                        $ua = input('ua','','trim');
+                        $this->yimi_advert($ua);
+                    }
                 }
 
                 //结果