如何封装直播网站源码中存储功能的教程
日常开发中，存储对于短视频、直播、1v1、约玩等都是不可获取的功能，此次我们来介绍一下云豹直播网站源码中存储功能的代码封装，以七牛、腾讯为例，代码如下：
一、获取存储类型
#pragma mark - 第一步:获取存储类型(tx 、qiniu)
-(void)getCOSType:(YBStorageTypeBlock)complete {
NSDate *oldDate = [StorageConfig getResTime] ? [StorageConfig getResTime] : [NSDate date];
NSDate *curDate = [NSDate date];
NSTimeInterval timeInterval = [curDate timeIntervalSinceDate:oldDate];
[MBProgressHUD showMessage:@""];
YBWeakSelf;
//这里过期时间默认 10 分钟，不建议改特别大的值(七牛存储token有过期时间,这个时间一定要小于七牛的过期时间)
if (![StorageConfig getResTime] || timeInterval > 600) {
//说明没有存储过或者超过10分钟、获取存储类型(如果是七牛顺带获取token)
[YBToolClass postNetworkWithUrl:@"Video.getCosInfo" andParameter:nil success:^(int code, id _Nonnull info, NSString * _Nonnull msg) {
[MBProgressHUD hideHUD];
if (code == 0) {
NSDictionary *infoDic = [info firstObject];
//保留存储时间、存储类型
[StorageConfig saveResTime:[NSDate date]];
[StorageConfig saveStorageType:minstr([infoDic valueForKey:@"cloudtype"])];
if ([[infoDic valueForKey:@"qiniuInfo"] isKindOfClass:[NSDictionary class]]) {
NSDictionary *qnInfo = [infoDic valueForKey:@"qiniuInfo"];
[StorageConfig saveQNToken:minstr([qnInfo valueForKey:@"qiniuToken"])];
[StorageConfig saveQNDomain:minstr([qnInfo valueForKey:@"qiniu_domain"])];
[StorageConfig saveQNZone:minstr([qnInfo valueForKey:@"qiniu_zone"])];
}
if ([[infoDic valueForKey:@"txCloudInfo"] isKindOfClass:[NSDictionary class]]) {
NSDictionary *txInfo = [infoDic valueForKey:@"txCloudInfo"];
[StorageConfig saveTXRegion:minstr([txInfo valueForKey:@"region"])];
[StorageConfig saveTXBucket:minstr([txInfo valueForKey:@"bucket"])];
[StorageConfig saveTXAppID:minstr([txInfo valueForKey:@"appid"])];
}
}else {
[MBProgressHUD showError:msg];
}
} fail:^{
[MBProgressHUD hideHUD];
}];
}else {
//10分钟内不请求接口,使用本地存贮的值
[MBProgressHUD hideHUD];
}
}
二、开始上传
#pragma mark - 第二步:开始上传(tx 、qiniu)
//图片
-(void)yb_storageImg:(UIImage *)img andName:(NSString *)imgName complete:(YBStorageFinishaBlock)complete{
NSString *cloudType = minstr([StorageConfig getStorageType]);
NSData *thumbData = UIImagePNGRepresentation(img);
if ([cloudType isEqual:@"tx"]) {
//腾讯
QCloudCOSXMLUploadObjectRequest* put = [QCloudCOSXMLUploadObjectRequest new];
put.object = imgName;//[NSString stringWithFormat:@"dspdemo/%@",imgName];
put.bucket = minstr([StorageConfig getTXBucket]);
put.body = thumbData;
[put setSendProcessBlock:^(int64_t bytesSent, int64_t totalBytesSent, int64_t totalBytesExpectedToSend) {
NSLog(@"rk;;upload %lld totalSend %lld aim %lld", bytesSent, totalBytesSent, totalBytesExpectedToSend);
}];
[put setFinishBlock:^(id outputObject, NSError* error) {
QCloudUploadObjectResult *rst = outputObject;
NSLog(@"rk;;111111:\nlocation:%@\n%@",rst.location,rst.key);
if (error) {
//失败
}else{
//成功
}
}];
[[QCloudCOSTransferMangerService defaultCOSTransferManager] UploadObject:put];
}else {
//七牛
QNConfiguration *config = [QNConfiguration build:^(QNConfigurationBuilder *builder) {
builder.zone = [self getQNZone];
}];
QNUploadOption *option = [[QNUploadOption alloc]initWithMime:nil progressHandler:nil params:nil checkCrc:NO cancellationSignal:nil];
QNUploadManager *upManager = [[QNUploadManager alloc] initWithConfiguration:config];
[upManager putData:thumbData key:imgName token:minstr([StorageConfig getQNToken]) complete:^(QNResponseInfo *info, NSString *key, NSDictionary *resp) {
NSLog(@"info ===== %@ \n resp ===== %@", info,resp);
if (info.ok) {
//成功
}else {
//失败
}
} option:option];
}
}
//音、视频
-(void)yb_storageVideoOrVoice:(NSString *)filePath andName:(NSString *)fileName complete:(YBStorageFinishaBlock)complete{
NSString *cloudType = minstr([StorageConfig getStorageType]);
if ([cloudType isEqual:@"tx"]) {
//腾讯
QCloudCOSXMLUploadObjectRequest* put = [QCloudCOSXMLUploadObjectRequest new];
NSURL* pathUrl = [NSURL fileURLWithPath:filePath];
put.object = fileName;
put.bucket = minstr([StorageConfig getTXBucket]);
put.body = pathUrl;
[put setSendProcessBlock:^(int64_t bytesSent, int64_t totalBytesSent, int64_t totalBytesExpectedToSend) {
NSLog(@"rk;;upload %lld totalSend %lld aim %lld", bytesSent, totalBytesSent, totalBytesExpectedToSend);
}];
[put setFinishBlock:^(id outputObject, NSError* error) {
QCloudUploadObjectResult *rst = outputObject;
NSLog(@"rk;;111111:\nlocation:%@\n%@",rst.location,rst.key);
if (error) {
//失败
}else{
//成功
}
}];
[[QCloudCOSTransferMangerService defaultCOSTransferManager] UploadObject:put];
}else {
//七牛
QNConfiguration *config = [QNConfiguration build:^(QNConfigurationBuilder *builder) {
builder.zone = [self getQNZone];
}];
QNUploadOption *option = [[QNUploadOption alloc]initWithMime:nil progressHandler:nil params:nil checkCrc:NO cancellationSignal:nil];
QNUploadManager *upManager = [[QNUploadManager alloc] initWithConfiguration:config];
[upManager putFile:filePath key:fileName token:minstr([StorageConfig getQNToken]) complete:^(QNResponseInfo *info, NSString *key, NSDictionary *resp) {
NSLog(@"info ===== %@ \n resp ===== %@", info,resp);
if (info.ok) {
//成功
}else {
//失败
}
} option:option];
}
}
#pragma mark - 腾讯初始化、验证开始
-(void)txInit {
if (![[StorageConfig getStorageType] isEqual:@"tx"]) {
return;
}
if ([PublicObj checkNull:[StorageConfig getTXAppID]] || [PublicObj checkNull:[StorageConfig getTXBucket]] ||[PublicObj checkNull:[StorageConfig getTXRegion]]) {
dispatch_async(dispatch_get_main_queue(), ^{
[MBProgressHUD showError:@"腾讯云存储信息不完整"];
});
return;
}
QCloudServiceConfiguration* configuration = [QCloudServiceConfiguration new];
configuration.appID = minstr([StorageConfig getTXAppID]);//@"1258210369";
configuration.signatureProvider = self;
QCloudCOSXMLEndPoint* endpoint = [[QCloudCOSXMLEndPoint alloc] init];
endpoint.regionName = minstr([StorageConfig getTXRegion]);//@"ap-shanghai";
configuration.endpoint = endpoint;
[QCloudCOSXMLService registerDefaultCOSXMLWithConfiguration:configuration];
[QCloudCOSTransferMangerService registerDefaultCOSTransferMangerWithConfiguration:configuration];
}
//腾讯验签
- (void)signatureWithFields:(QCloudSignatureFields*)fileds request:(QCloudBizHTTPRequest*)request urlRequest:(NSMutableURLRequest*)urlRequst compelete:(QCloudHTTPAuthentationContinueBlock)continueBlock {
//此接口代替原来的 8088 端口请求
[YBToolClass postNetworkWithUrl:@"Video.getTxCosFederationToken" andParameter:nil success:^(int code, id _Nonnull info, NSString * _Nonnull msg) {
if (code == 0) {
NSDictionary *infoDic = [info firstObject];
QCloudCredential* credential = [QCloudCredential new];
credential.secretID = [NSString stringWithFormat:@"%@",[infoDic valueForKey:@"tmpSecretId"]];
credential.secretKey = [NSString stringWithFormat:@"%@",[infoDic valueForKey:@"tmpSecretKey"]];
credential.token = [NSString stringWithFormat:@"%@",[infoDic valueForKey:@"sessionToken"]];
QCloudAuthentationV5Creator* creator = [[QCloudAuthentationV5Creator alloc] initWithCredential:credential];
QCloudSignature* signature = [creator signatureForData:urlRequst];
continueBlock(signature, nil);
}else {
dispatch_async(dispatch_get_main_queue(), ^{
[MBProgressHUD showError:msg];
});
}
} fail:^{
}];
}
至此封装完毕，以后二次开发时可由此替换其他类型的存储，只需要在这个功能类中进行单独配置就好了，而外部只需要调用即可，具有高度解耦的属性，便于直播网站源码的二次开发。
声明：以上内容为作者本人原创，未经作者本人同意，禁止转载，否则将追究相关法律责任