Go学习:struct和方法

定义struct

type treeNode struct {
	value       int
	left, right *treeNode
}

• go语言的对象仅支持封装,不支持继承和多态
• go语言只有struct没有class
• 继承和多态由接口来做

创建对象

package main

import (
	"fmt"
)

type treeNode struct {
	value       int
	left, right *treeNode
}
/**
	自定义工厂函数
 */
func createNode(value int) *treeNode {
	return &treeNode{value: value}
}

func main() {

	var root treeNode
	// {0 <nil> <nil>}
	fmt.Println(root)

	root = treeNode{value: 3}
	root.left = &treeNode{}
	root.right = &treeNode{5,nil,nil}
	// 内建函数
	root.right.left = new(treeNode)
	root.left.right = createNode(2)
	
	nodes := []treeNode {
		{value: 3},
		{},
		{6,nil,&root},
	}
	// [{3 <nil> <nil>} {0 <nil> <nil>} {6 <nil> 0xc000004078}]
	fmt.Println(nodes)
}

• 使用自定义工厂函数
• 返回了局部变量的地址!

结构的创建是创建再堆还是栈上?

• 不一定,是由go语言的编译器和运行环境来决定的

  • 如果局部变量没有取地址没有返回出去给其它函数引用,编译器就会认为变量没有给别人用,就会再栈上分配
  • 如果局部变量取了地址并且返回去给其他函数引用,编译器就会认为变量给别人引用,就会再堆上分配,该变量就会参与垃圾回收,当指针没有在引用的时候就会被回收掉
  • go语言不一定退出函数变量就被销毁了

为struct定义方法

package main

import (
	"fmt"
)

type treeNode struct {
	value       int
	left, right *treeNode
}

/**
自定义工厂函数
*/
func createNode(value int) *treeNode {
	return &treeNode{value: value}
}

/**
在函数名前加了一个括号
node就是方法接收者,代表这个方法是给treeNode使用的
相当于java的this
*/
func (node treeNode) print() {
	fmt.Print(node.value, " ")
}

/**
调用者改成指针,保证调用能把值修改成功
*/
func (node *treeNode) setValue(value int) {
	if node == nil{
		// 可以调用但是没办法设置值,不return就会报错
		fmt.Println("node is nil")
		return
	}
	node.value = value
}

/**
遍历树,从最左->最右
 */
func (node *treeNode)traverse(){
	if node == nil {
		return
	}
	node.left.traverse()
	node.print()
	node.right.traverse()
}

func main() {

	var root treeNode
	// {0 <nil> <nil>}
	fmt.Println(root)

	root = treeNode{value: 3}
	root.left = &treeNode{}
	root.right = &treeNode{5, nil, nil}
	// 内建函数
	root.right.left = new(treeNode)
	root.left.right = createNode(2)
	root.right.left.setValue(4)
	// 0 2 3 4 5
	root.traverse()
	fmt.Println()

	// 4
	root.right.left.print()
	fmt.Println()

	// 3
	root.print()
	fmt.Println()
	root.setValue(100)

	pRoot := &root
	// 100
	pRoot.print()
	fmt.Println()
	pRoot.setValue(200)
	// 200
	pRoot.print()
	fmt.Println()

	var nilNode *treeNode
	// node is nil
	nilNode.setValue(1)

}

• 使用指针才可以改变结构内容
• nil指针也可以调用方法
• 值/指针接收者均可接收值/指针

值接收者VS指针接收者

• 值接收者是go语言独有
• 要改变内容必须使用指针接收者;因为值接收者是一个拷贝,再函数内的修改不会影响到其他地方
• 结构过大也要考虑使用指针接收者;因为值接收者是一个拷贝,结构过大考虑性能也建议使用指针接收者
• 一致性: 如有指针接收者,最好都是指针接收者(代码简单,统一)
• 不管使用值接收者还是指针接收者,调用的时候都是该怎么调用就怎么调用,不用做处理