OSS 上传基类修改

galaxy/galaxy/Classes/Base/C/YHBaseViewModel.swift

@@ -79,7 +79,7 @@ class YHBaseViewModel {
             return
         }
         YHOSSManager.share.updateToken {
-            YHOSSManager.share.getPublic(url: url, isPublic, callBackBlock: { [weak self] success, error in
+            YHOSSManager.share.getPublic(url: url, callBackBlock: { [weak self] success, error in
                 guard let _ = self else { return }
                 DispatchQueue.main.async {
                     // 在这里执行需要在主线程上完成的任务

galaxy/galaxy/Classes/Tools/Upload/YHOSSManager.swift

@@ -25,6 +25,8 @@ class YHOSSManager: NSObject, URLSessionDelegate {
     var publicModel: YHOSSModel = YHOSSModel()
     var mProvider: OSSStsTokenCredentialProvider!
     var mClient: OSSClient!
+    var publicProvider: OSSStsTokenCredentialProvider!
+    var publicClient: OSSClient!
     var fileName: [String] = []
     
     override init() {
@@ -60,6 +62,8 @@ class YHOSSManager: NSObject, URLSessionDelegate {
                 return
             }
             self.model = resultModel
+            self.mProvider = OSSStsTokenCredentialProvider(accessKeyId: self.model.accessKeyId, secretKeyId: self.model.accessKeySecret, securityToken: self.model.securityToken)
+            self.mClient = OSSClient(endpoint: self.model.endpoint, credentialProvider: self.mProvider)
             if let block = completionHandler {
                 block()
             }
@@ -74,6 +78,8 @@ class YHOSSManager: NSObject, URLSessionDelegate {
                 return
             }
             self.publicModel = resultModel
+            self.publicProvider = OSSStsTokenCredentialProvider(accessKeyId: self.publicModel.accessKeyId, secretKeyId: self.publicModel.accessKeySecret, securityToken: self.publicModel.securityToken)
+            self.publicClient = OSSClient(endpoint: self.publicModel.endpoint, credentialProvider: self.publicProvider)
         } failBlock: { err in
         }
     }
@@ -89,8 +95,8 @@ class YHOSSManager: NSObject, URLSessionDelegate {
             self.mProvider = OSSStsTokenCredentialProvider(accessKeyId: self.publicModel.accessKeyId, secretKeyId: self.publicModel.accessKeySecret, securityToken: self.publicModel.securityToken)
             self.mClient = OSSClient(endpoint: self.publicModel.endpoint, credentialProvider: self.mProvider)
         } else {
-            self.mProvider = OSSStsTokenCredentialProvider(accessKeyId: self.model.accessKeyId, secretKeyId: self.model.accessKeySecret, securityToken: self.model.securityToken)
-            self.mClient = OSSClient(endpoint: self.model.endpoint, credentialProvider: self.mProvider)
+//            self.mProvider = OSSStsTokenCredentialProvider(accessKeyId: self.model.accessKeyId, secretKeyId: self.model.accessKeySecret, securityToken: self.model.securityToken)
+//            self.mClient = OSSClient(endpoint: self.model.endpoint, credentialProvider: self.mProvider)
         }
         DispatchQueue.global().async {
             
@@ -117,8 +123,12 @@ class YHOSSManager: NSObject, URLSessionDelegate {
             request.uploadProgress = { (bytesSent: Int64, totalBytesSent: Int64, totalBytesExpectedToSend: Int64) -> Void in
                 printLog("bytesSent:\(bytesSent),totalBytesSent:\(totalBytesSent),totalBytesExpectedToSend:\(totalBytesExpectedToSend)")
             }
-            
-            let task = self.mClient.putObject(request)
+            var task = OSSTask<AnyObject>()
+            if isPublic {
+                task = self.publicClient.putObject(request)
+            } else {
+                task = self.mClient.putObject(request)
+            }
             
             task.continue({ (t) -> Any? in
                 let result = t
@@ -139,7 +149,7 @@ class YHOSSManager: NSObject, URLSessionDelegate {
                     self.fileName.removeFirst()
                 }
                 return
-            })
+            }).waitUntilFinished()
         }
     }
     
@@ -149,13 +159,6 @@ class YHOSSManager: NSObject, URLSessionDelegate {
             callBackBlock(nil, YHErrorModel())
             return
         }
-        if isPublic {
-            self.mProvider = OSSStsTokenCredentialProvider(accessKeyId: self.publicModel.accessKeyId, secretKeyId: self.publicModel.accessKeySecret, securityToken: self.publicModel.securityToken)
-            self.mClient = OSSClient(endpoint: self.publicModel.endpoint, credentialProvider: self.mProvider)
-        } else {
-            self.mProvider = OSSStsTokenCredentialProvider(accessKeyId: self.model.accessKeyId, secretKeyId: self.model.accessKeySecret, securityToken: self.model.securityToken)
-            self.mClient = OSSClient(endpoint: self.model.endpoint, credentialProvider: self.mProvider)
-        }
         DispatchQueue.global().async {
             let fileExtension = URL(fileURLWithPath: file).pathExtension.lowercased()
             request.uploadingData = fileData
@@ -180,8 +183,12 @@ class YHOSSManager: NSObject, URLSessionDelegate {
             request.uploadProgress = { (bytesSent: Int64, totalBytesSent: Int64, totalBytesExpectedToSend: Int64) -> Void in
                 printLog("bytesSent:\(bytesSent),totalBytesSent:\(totalBytesSent),totalBytesExpectedToSend:\(totalBytesExpectedToSend)")
             }
-            
-            let task = self.mClient.putObject(request)
+            var task = OSSTask<AnyObject>()
+            if isPublic {
+                task = self.publicClient.putObject(request)
+            } else {
+                task = self.mClient.putObject(request)
+            }
             
             task.continue({ (t) -> Any? in
                 let result = t
@@ -204,18 +211,12 @@ class YHOSSManager: NSObject, URLSessionDelegate {
                     self.fileName.removeFirst()
                 }
                 return
-            })
+            }).waitUntilFinished()
         }
     }
     
-    func getPublic(url: String, _ isPublic: Bool = false, callBackBlock:@escaping (_ success: String?, _ error:YHErrorModel?)->()) {
-        if isPublic {
-            self.mProvider = OSSStsTokenCredentialProvider(accessKeyId: self.publicModel.accessKeyId, secretKeyId: self.publicModel.accessKeySecret, securityToken: self.publicModel.securityToken)
-            self.mClient = OSSClient(endpoint: self.publicModel.endpoint, credentialProvider: self.mProvider)
-        } else {
-            self.mProvider = OSSStsTokenCredentialProvider(accessKeyId: self.model.accessKeyId, secretKeyId: self.model.accessKeySecret, securityToken: self.model.securityToken)
-            self.mClient = OSSClient(endpoint: self.model.endpoint, credentialProvider: self.mProvider)
-        }
+    func getPublic(url: String, callBackBlock:@escaping (_ success: String?, _ error:YHErrorModel?)->()) {
+    
         DispatchQueue.global().async {
             
             let bucketName = self.getBucket(url)