Loading...
墨滴

CoderStar

2021/12/06  阅读:41  主题:橙心

从 SIL 角度看 Swift 中的值类型与引用类型

前言

Hi Coder,我是 CoderStar!

在 Swift 开发过程中,你很可能至少问过自己一次structclass之间的区别,即使你自己没问过,你的面试官应该也问过。对这个问题的答案中,可能最大的区别就是一个是值类型,而另一个是引用类型,今天我们就来具体聊聊这个区别。

那在介绍值类型与引用类型之前,我们还是先来回顾一下structclass之间的区别这个问题。

class & struct

在 Swift 中,其实classstruct之间的核心区别不是很多,有很多区别是值类型与引用类型这个区别隐形带来的天然的区别。

  • class 可以继承,struct 不能继承(当然struct可以利用protocol来实现类似继承的效果。);受此影响的区别有:
    • struct中方法的派发方式全都是直接派发,而class中根据实际情况有多种派发方式,详情可看Swift派发机制
  • class 需要自己定义构造函数,struct 默认生成;

    struct 默认生成的构造函数必须包括所有成员参数,只有当所有参数都为可选型时,可直接不用传入参数直接简单构造,class 中的属性必须都有默认值,否则编译错误, 可以通过声明时赋值或者构造函数赋值两种方式给属性设置默认值。

  • class 是引用类型,struct 是值类型;受此影响的区别有:
    • struct 改变其属性受修饰符 let 影响,不可改变,class 不受影响;
    • struct 方法中需要修改自身属性时 (非 init 方法),方法需要前缀修饰符 mutating
    • struct 因为是值类型的原因,所以自动线程安全,而且也不存在循环引用导致内存泄漏的风险;
    • ...
    • 更多看下一章节
  • ...

在 Swift 中,很多基础类型,如StringInt等等,都是使用Struct来定义。对于如何选择两者这个问题上,Apple 在一些官方文档中也给出了它们之间的区别以及官方建议。

来自《choosing_between_structures_and_classes》

在向 app 中添加新数据类型时,您不妨考虑以下建议来帮助自己做出合理的选择。

  • 默认使用结构。
  • 在需要 Objective-C 互操作性时使用类。
  • 在需要控制建模数据的恒等性时使用类。
  • 将结构与协议搭配,通过共享实现来采用行为。

值类型 & 引用类型

那在 Swift 中,值类型与引用类型之间的区别有哪些呢?

  • 存储方式及位置:大部分值类型存储在栈上,大部分引用类型存储在堆上;
  • 内存:值类型没有引用计数,也不会存在循环引用以及内存泄漏等问题;
  • 线程安全:值类型天然线程安全,而引用类型需要开发者通过加锁等方式来保证;
  • 拷贝方式:值类型拷贝的是内容,而引用类型拷贝的是指针,从一定意义上讲就是所谓的深拷贝及浅拷贝;

在 Swift 中,值类型除了struct之外还有enumtuple,引用类型除了class之外还有closure/func

存储方式及位置

上文说的'堆'和'栈'是程序运行中的不同内存空间。

关于堆、栈存储原理,美团的这篇【基本功】深入剖析Swift性能优化给出了细节说明,这里就不再赘述了,大概说下结论。

值类型默认存储在栈区,栈区内存是连续的,通过出栈入栈进行分配和销毁,速度很快,而且每个线程都有自己的栈空间,所以不需要考虑线程安全问题;访问存储内容时一次就可以拿到值。

引用类型,只在栈区存储了对象的指针,指针指向的对象的内存是分配在堆区的。堆在分配和释放时都要调用函数(MALLOC,FREE) 动态申请 / 释放内存,这些都会花费一些时间,而且因为堆空间被所有线程共享,所以在使用时要考虑线程安全。访问存储内容时,需要两次访问内存,第一次得取得指针,第二次才是真正的数据。

其中在 64 位系统上,iOS 加入了Tagged Pointer优化方式,即直接在指针中存储值,比如NSNumber以及NSString结构。

从描述来看,我们得到的最重要的结论是使用值类型比使用引用类型更快,具体技术指标可查看why-choose-struct-over-class,还有一个测试项目StructVsClassPerformance

通过上面的描述,我们可以有一个问题,就是所有的class都存储在堆上,所有的struct都存储在栈上吗?这也是本篇文章的重点。其实对于绝大多数情况而言,这种说法都是没问题的,但是总会有些特殊情况。

在阅读下文之前,我们先看一下,如何判断对象是在栈分配还是在堆分配。对于这个问题我们可以在SIL.rst中找到答案。Swift编译生成的SIL文件中,会包含派发指令,与内存分配相关的命令中,有alloc-stackalloc-box命令可以来帮助我们解决这个问题,简单来说前者就是来栈上分类内存的指令,而后者就是在堆上分配任务的指令。

栈上的引用类型

堆栈上的分配和释放成本远低于堆上的分配和释放,因此有时编译器可能会提升引用类型也存储在堆栈上,这个过程实际发生在 SIL 优化阶段,官方术语叫做Memory promotion。关于这一说法,我们可以在Guaranteed Optimization and Diagnostic Passes找到支撑。

Memory promotion is implemented as two optimization phases, the first of which performs capture analysis to promote alloc_box instructions to alloc_stack, and the second of which promotes non-address-exposed alloc_stack instructions to SSA registers.

大致意思是就是 SIL 阶段会尽量进行内存提升,将原来堆内存提升为栈内存,栈内存提升为 SSA 寄存器内存。

具体优化部分代码我们可以在AllocBoxToStack.cpp中看到。

堆上的值类型

在《Swift 进阶》书中有过这么一段话,(在 3.0 版本中出现,5.0 版本删除掉了):

Swift 的结构体一般被存储在栈上,而非堆上。不过这其实是一种优化: 默认情况下结构体是存储在堆上的,但是在绝大多数时候,这个优化会生效,并将结构体存储到栈上。当结构体变量被一个函数闭合的时候,优化将不再生效,此时这个结构体将存储在堆上。

看到这句话有些同学会有点摸不着头脑,为什么默认情况结构体会存在堆上,然后经过优化时候才存储到栈上。下面我们来看struct编译生成的相关 SIL 文件。

struct Test {}

这是一个非常简单的struct结构体,简单到连属性都没了,我们使用swiftc命令生成 SIL 文件,命令如下:

swiftc Test.swift -emit-silgen | xcrun swift-demangle > TestSILGen.sil

其含义就是生成Raw SIL,也就是原生SIL文件,没有经过任何优化和处理。更多命令可以看之前输出的一篇文章iOS编译简析

生成的 SIL 文件内容如下:

sil_stage raw

import Builtin
import Swift
import SwiftShims

struct Test {
  init()
}

// main
sil [ossa] @main : $@convention(c) (Int32UnsafeMutablePointer<Optional<UnsafeMutablePointer<Int8>>>) -> Int32 {
bb0(%0 : $Int32, %1 : $UnsafeMutablePointer<Optional<UnsafeMutablePointer<Int8>>>):
  %2 = integer_literal $Builtin.Int320          // user: %3
  %3 = struct $Int32 (%2 : $Builtin.Int32)        // user: %4
  return %3 : $Int32                              // id: %4
} // end sil function 'main'

// Test.init()
sil hidden [ossa] @$s4main4TestVACycfC : $@convention(method) (@thin Test.Type) -> Test 
{
// %0 "$metatype"
bb0(%0 : $@thin Test.Type):
  %1 = alloc_box ${ var Test }, let, name "self"  // user: %2
  %2 = mark_uninitialized [rootself] %1 : ${ var Test } // users: %5, %3
  %3 = project_box %2 : ${ var Test }, 0          // user: %4
  %4 = load [trivial] %3 : $*Test                 // user: %6
  destroy_value %2 : ${ var Test }                // id: %5
  return %4 : $Test                               // id: %6
// end sil function '$s4main4TestVACycfC'

我们可以很明显的看到alloc_box字眼。

然后我们再使用生成优化后 SIL 文件的命令,如下:

swiftc Test.swift -emit-sil | xcrun swift-demangle > TestSIL.sil

sil_stage canonical

import Builtin
import Swift
import SwiftShims

struct Test {
  init()
}

// main
sil @main : $@convention(c) (Int32UnsafeMutablePointer<Optional<UnsafeMutablePointer<Int8>>>) -> Int32 {
bb0(%0 : $Int32, %1 : $UnsafeMutablePointer<Optional<UnsafeMutablePointer<Int8>>>):
  %2 = integer_literal $Builtin.Int320          // user: %3
  %3 = struct $Int32 (%2 : $Builtin.Int32)        // user: %4
  return %3 : $Int32                              // id: %4
} // end sil function 'main'

// Test.init()
sil hidden @$s4main4TestVACycfC : $@convention(method) (@thin Test.Type) -> Test 
{
// %0 "$metatype"
bb0(%0 : $@thin Test.Type):
  %1 = alloc_stack $Testlet, name "self"        // user: %3
  %2 = struct $Test ()                            // user: %4
  dealloc_stack %1 : $*Test                       // id: %3
  return %2 : $Test                               // id: %4
} // end sil function '$s4main4TestVACycfC'

我们很明显看到alloc_stack字眼。

相信大家已经明白发生了什么,struct 在生成原始的 SIL 文件中实际上会使用堆指令,然后在 SIL 优化阶段会根据代码上下文环境判断是否可以优化到栈上继而对指令进行修改。那大部分情况下是都可以优化到栈上的。这个过程就有上述AllocBoxToStack.cpp文件的参与。

当然,那肯定还有另外的少部分情况。比如说:

func uniqueIntegerProvider() -> () -> Int {
    // i是Int类型,本质也是一个结构体
    var i = 0
    return {
        i+=1
        return i
    }
}

对此代码生成的两份 SIL 文件,核心部分如下:

优化前:

// uniqueIntegerProvider()
sil hidden [ossa] @main.uniqueIntegerProvider() -> () -> Swift.Int : $@convention(thin) () -> @owned @callee_guaranteed () -> Int {
bb0:
  %0 = alloc_box ${ var Int }, var, name "i"      // users: %11, %8, %1
  %1 = project_box %0 : ${ var Int }, 0           // users: %9, %6
  %2 = integer_literal $Builtin.IntLiteral0     // user: %5
  %3 = metatype $@thin Int.Type                   // user: %5
  // function_ref Int.init(_builtinIntegerLiteral:)
  %4 = function_ref @Swift.Int.init(_builtinIntegerLiteral: Builtin.IntLiteral) -> Swift.Int : $@convention(method) (Builtin.IntLiteral, @thin Int.Type) -> Int // user: %5
  %5 = apply %4(%2, %3) : $@convention(method) (Builtin.IntLiteral, @thin Int.Type) -> Int // user: %6
  store %5 to [trivial] %1 : $*Int                // id: %6
  // function_ref closure #1 in uniqueIntegerProvider()
  %7 = function_ref @closure #1 () -> Swift.Int in main.uniqueIntegerProvider() -> () -> Swift.Int : $@convention(thin) (@guaranteed { var Int }) -> Int // user: %10
  %8 = copy_value %0 : ${ var Int }               // user: %10
  mark_function_escape %1 : $*Int                 // id: %9
  %10 = partial_apply [callee_guaranteed] %7(%8) : $@convention(thin) (@guaranteed { var Int }) -> Int // user: %12
  destroy_value %0 : ${ var Int }                 // id: %11
  return %10 : $@callee_guaranteed () -> Int      // id: %12
// end sil function 'main.uniqueIntegerProvider() -> () -> Swift.Int'

优化后:

// uniqueIntegerProvider()
sil hidden @main.uniqueIntegerProvider() -> () -> Swift.Int : $@convention(thin) () -> @owned @callee_guaranteed () -> Int {
bb0:
  %0 = alloc_box ${ var Int }, var, name "i"      // users: %8, %7, %6, %1
  %1 = project_box %0 : ${ var Int }, 0           // user: %4
  %2 = integer_literal $Builtin.Int640          // user: %3
  %3 = struct $Int (%2 : $Builtin.Int64)          // user: %4
  store %3 to %1 : $*Int                          // id: %4
  // function_ref closure #1 in uniqueIntegerProvider()
  %5 = function_ref @closure #1 () -> Swift.Int in main.uniqueIntegerProvider() -> () -> Swift.Int : $@convention(thin) (@guaranteed 
var Int }) -> Int // user: %7
  strong_retain %0 : ${ var Int }                 // id: %6
  %7 = partial_apply [callee_guaranteed] %5(%0) : $@convention(thin) (@guaranteed { var Int }) -> Int // user: %9
  strong_release %0 : ${ var Int }                // id: %8
  return %7 : $@callee_guaranteed () -> Int       // id: %9
// end sil function 'main.uniqueIntegerProvider() -> () -> Swift.Int'

可以很明显的看出,无论是优化前还是优化后,使用的都是alloc_box指令,也就是说此时的变量i是存储在堆上的。其实原因也很好理解,其实就是变量 i 被函数闭合了,即使在退出作用域的情况下,仍然得保持 i 的存在。当然这只是一种情况,还会有其他的情况。

总结:所以说在 Swift 中所有的class都存储在堆上,所有的struct都存储在栈上这种说法是有问题的,只能说大部分情况是如此的,总有些情况会跟你淘气,具体存储位置还得结合结构所在上下文以及 SIL 优化手段等等因素综合分析。

拷贝方式

引用类型,在拷贝时,实际上拷贝的只是栈区存储的对象的指针;值类型拷贝的是实际的值。

对于值类型拷贝,Swift 有一套 写时复制 COW(Copy-On-Write) 优化机制,即只有赋值后值类型发生改变的时候才会进行真正的拷贝,当没有改变时,两者共享同一个内存地址。

Apple在 OptimizationTips 中,给出了一个示例,代码很简单,相信大家一下就能明白。

该文档中还有一些 Apple 给出的另外的优化方式,比如减少动态派发的方式等等,建议 enjoy。

final class Ref<T{
  var val: T
  init(_ v: T) {val = v}
}

struct Box<T{
    var ref: Ref<T>
    init(_ x: T) { ref = Ref(x) }

    var value: T {
        get { return ref.val }
        set {
          /// 判断当前对象是否只有一个引用,如果不是才进行拷贝
          if !isKnownUniquelyReferenced(&ref) {
            ref = Ref(newValue)
            return
          }
          ref.val = newValue
        }
    }
}

Swift 标准库中,StringArrayDictionarySet 等默认实现了COW,对于自定义对象,我们需要自己实现。

最后

在编写本地文章过程中,查看了Swift开源仓库 docs 目录下的一些文档,学到了很多,也建议各位读者同学enjoy!

要更加努力呀!

Let's be CoderStar!

更多资料

CoderStar

2021/12/06  阅读:41  主题:橙心

作者介绍

CoderStar