StreamingKit保存在线音频流到本地

StreamingKit是一个强大的IOS音频播放工具，链接在此。具体用法和功能不再赘述，网上资料很多。本文主要记录笔者在项目中遇到的问题－－保存在线音频流到本地。

具体需求如下：在线播放音频流到时候，需要将播放完的音频流保存到本地，这样下次播放的时候就不需要从网络加载。

首先看下StreamingKit源码，在STKAudioPlayer.m中可以找到+(STKDataSource*) dataSourceFromURL:(NSURL*)url方法。

+(STKDataSource*) dataSourceFromURL:(NSURL*)url

{

  STKDataSource* retval = nil;

  if ([url.scheme isEqualToString:@"file"])
  {
    retval = [[STKLocalFileDataSource alloc] initWithFilePath:url.path];
  }
  else if ([url.scheme caseInsensitiveCompare:@"http"] == NSOrderedSame || [url.scheme caseInsensitiveCompare:@"https"] == NSOrderedSame)
  {
    retval = [[STKAutoRecoveringHTTPDataSource alloc] initWithHTTPDataSource:[[STKHTTPDataSource alloc] initWithURL:url];
  }
  return retval;

}

可见在线音频会通过STKHTTPDataSource对象封装。查看STKHTTPDataSource.m源码，可以看到有一个-(int) privateReadIntoBuffer:(UInt8*)buffer withSize:(int)size方法。OK，既然能读就能写，就在这儿下手。

首先给STKHTTPDataSource添加两个变量

NSString * audioCacheFilePath;
AudioFileID audioFile;

修改下init方法：

-(instancetype) initWithURL:(NSURL*)url andCacheFilePath : (NSString *) filePath;

-(instancetype) initWithURL:(NSURL*)url httpRequestHeaders:(NSDictionary*)httpRequestHeaders andCacheFilePath : (NSString *) filePath;

-(instancetype) initWithURLProvider:(STKURLProvider)urlProvider andCacheFilePath : (NSString *) filePath;

-(instancetype) initWithAsyncURLProvider:(STKAsyncURLProvider)asyncUrlProvider andCacheFilePath : (NSString *) filePath;

我们在-(instancetype) initWithAsyncURLProvider:(STKAsyncURLProvider)asyncUrlProviderIn andCacheFilePath : (NSString *) filePath方法里，对audioFile就行初始化。

if(filePath){

  audioCacheFilePath = filePath;

  AudioStreamBasicDescription asbd = (AudioStreamBasicDescription)

  {

    .mSampleRate = 44100.00,

    .mFormatID = kAudioFormatMPEGLayer3,

    .mFramesPerPacket = 1,

    .mChannelsPerFrame = 2,

    .mBytesPerFrame = sizeof(SInt16) * 2 /*channelsPerFrame*/,

    .mBitsPerChannel = 8 * sizeof(SInt16),

    .mBytesPerPacket = (sizeof(SInt16) * 2)

  };

  OSStatus audioErr = noErr;

  audioErr = AudioFileCreateWithURL((__bridge CFURLRef)[NSURL fileURLWithPath:[filePath stringByAppendingString:@"_tmp"]], kAudioFileMP3Type, &asbd, kAudioFileFlags_EraseFile, &audioFile);
}

注意：这里我们的audioFile并不是指向filePath，而是一个临时文件。这是因为如果在写入文件的过程中出现网络问题，或者应用挂掉，那么我们的缓存文件就是一个不完整的音频文件，所以先将它保存在一个临时文件中。此处笔者保存文件格式是mp3，读者可以根据自己的需要自行调整文件格式，但是要保证AudioFileCreateWithURL方法和AudioFileCreateWithURL中的格式和参数相符，不然就会出现音频文件无法播放的问题。

然后修改-(int) privateReadIntoBuffer:(UInt8*)buffer withSize:(int)size方法：

-(int) privateReadIntoBuffer:(UInt8*)buffer withSize:(int)size
{
  if (size == 0){
    return 0;
  }

  if (prefixBytes != nil){

    int count = MIN(size, (int)prefixBytes.length - prefixBytesRead);

    [prefixBytes getBytes:buffer length:count];

    prefixBytesRead += count;

    if (prefixBytesRead >= prefixBytes.length){

      prefixBytes = nil;

    }

    return count;

  }

  int read = [super readIntoBuffer:buffer withSize:size];

  if(audioFile && audioCacheFilePath && read >= 0){

    UInt32 bytesToWrite = read;

    AudioFileWriteBytes(audioFile, false, relativePosition,     &bytesToWrite, buffer);

  }

  if (read < 0){

    return read;

  }

  relativePosition += read;

  return read;

}

最后，我们需要在eof的时候将临时文件改成目标文件，添加方法：

在STKAutoRecoveringHTTPDataSource.m的-(void) dataSourceEof:(STKDataSource*)dataSource方法添加如下逻辑：

if([dataSource isKindOfClass:[STKHTTPDataSource class]]){

  [(STKHTTPDataSource*)dataSource closeCacheFile];

}

当然，你还的给+(STKDataSource*) dataSourceFromURL:(NSURL*)url方法添加参数，以使用我们新的init方法。

第一次使用StreamingKit，如有纰漏还望见。欢迎讨论！

源码&安装
github: git@github.com:liyaozhong/StreamingKit.git
cocoapods: pod 'AdvancedStreamingKit'