快捷導(dǎo)航

使用Go初始化Struct的方法詳解

更新時(shí)間：2024年01月02日 11:00:06 作者：程序員讀書(shū)

面向?qū)ο缶幊陶Z(yǔ)言最基礎(chǔ)的概念就是類(lèi)(class),不過(guò)Go語(yǔ)言并沒(méi)有類(lèi)的概念,所以使用Go語(yǔ)言開(kāi)發(fā)時(shí),我們一般會(huì)用struct(結(jié)構(gòu)體)來(lái)模擬面向?qū)ο笾械念?lèi),下面我們來(lái)介紹幾種創(chuàng)建struct類(lèi)型變量的方法,需要的朋友可以參考下

引言

面向?qū)ο缶幊陶Z(yǔ)言最基礎(chǔ)的概念就是類(lèi)(class)，不過(guò)Go語(yǔ)言并沒(méi)有類(lèi)的概念，所以使用Go語(yǔ)言開(kāi)發(fā)時(shí)，我們一般會(huì)用struct(結(jié)構(gòu)體)來(lái)模擬面向?qū)ο笾械念?lèi)。

類(lèi)一般是通過(guò)構(gòu)造方法(constructors)來(lái)初始化類(lèi)的實(shí)例(對(duì)象)中的屬性，不過(guò)Go的struct并沒(méi)有構(gòu)造方法這種機(jī)制，那要怎么樣初始化struct類(lèi)型的實(shí)例呢？

下面我們來(lái)介紹幾種創(chuàng)建struct類(lèi)型變量的方法。

字面量

創(chuàng)建并初始化一個(gè)struct變量最簡(jiǎn)單直接的方式就是使用struct字面量：

//app/school.go
package school

type Student struct {
	ID    int
	Name  string
	Score int
	Grade string
}

//main.go
package main
func main() {
	s := &Student{
		ID:    1,
		Name:  "小明",
		Score: 90,
		Grade: "高中二班",
	}
}

不過(guò)有時(shí)候我們只是需要一個(gè)臨時(shí)struct類(lèi)型的話，可以使用匿名struct：

func main() {

	s := struct {
		ID    int
		Name  string
		Score int
		Grade string
	}{
		ID: 1, 
        Name: "小明",
        Score: 90,
        Grade: "高中二年級(jí)",
	}
}

使用內(nèi)置new函數(shù)

除了字面量外，使用Go內(nèi)置的new函數(shù)也可以創(chuàng)建一個(gè)struct變量，不過(guò)這種方式需要自己為struct的每個(gè)字段賦初始值：

func main(){
  s := new(Student)
  s.ID = 1
  s.Name = "小明"
  s.Score = 90
  s.Grade = "高中二班"
}

構(gòu)造函數(shù)

前面我們說(shuō)過(guò)Go語(yǔ)言的struct沒(méi)有構(gòu)造函數(shù)，但是我們可以自定義函數(shù)來(lái)初始化struct，自定義函數(shù)的好處在于：

可以達(dá)到復(fù)用的目的。
可以起到封裝的效果。
可以校驗(yàn)初始化值是否在允許的范圍內(nèi)。

一般我們自定義的構(gòu)造函數(shù)都以New為前綴，比如：

package school

type student struct {
	ID    int
	Name  string
	Score int
	Grade string
}

func NewStudent(ID int, Name string,Score int Grade string) *Student {
    if ID < 0{
        panic("ID不能小于0")
    }
    
    if Score < 0 || Score > 100{
        panic("Score必須在0~100之間")
    }
    
    s := new(Student)
    s.ID = 1
    s.Name = "小明"
    s.Score = 90
    s.Grade = "高中二班"
    return s
}

package main 

import "app/school"

func main(){
    s := school.NewStudent(1,"小明",90,"高中二班")
}

上面的例子中，我們自定義了NewStudent()函數(shù)，之后可以在其他包復(fù)用該函數(shù)來(lái)初始化struct，將Student改為student，這樣其他包無(wú)法直接訪問(wèn)student結(jié)構(gòu)體，達(dá)到了封裝的效果，而在NewStudent()函數(shù)中，我們也可以驗(yàn)證參數(shù)是否合理。

當(dāng)然自定義構(gòu)造函數(shù)也是必須以New為前綴的，比如標(biāo)準(zhǔn)庫(kù)os包中，初始化文件句柄時(shí)，可以使用Open()，Create()和OpenFile()等函數(shù)來(lái)初始化os.File：

//os包的file.go文件的代碼
func Open(name string) (*File, error) {
	return OpenFile(name, O_RDONLY, 0)
}

func Create(name string) (*File, error) {
	return OpenFile(name, O_RDWR|O_CREATE|O_TRUNC, 0666)
}

func OpenFile(name string, flag int, perm FileMode) (*File, error) {
	testlog.Open(name)
	f, err := openFileNolog(name, flag, perm)
	if err != nil {
		return nil, err
	}
	f.appendMode = flag&O_APPEND != 0

	return f, nil
}

支持可變參數(shù)的構(gòu)造函數(shù)

在上面的例子中，我們可以通過(guò)NewStudent()函數(shù)創(chuàng)建一個(gè)student類(lèi)型的變量，但是這個(gè)構(gòu)造函數(shù)的參數(shù)的順序與個(gè)數(shù)是固定的，如果有多個(gè)地方調(diào)用NewStudent()函數(shù)，此時(shí)在該函數(shù)新增一個(gè)參數(shù)時(shí)，那么所有調(diào)用到該函數(shù)的代碼都必須調(diào)整。

我們可以再優(yōu)化一下我們的構(gòu)造函數(shù)，使用其參數(shù)的個(gè)數(shù)與順序是可變的：

package school

type student struct {
	ID    int
	Name  string
	Score int
	Grade string
}

type StudentOptionFunc func(*student)

func WithID(id int) StudentOptionFunc {
	return func(s *student) {
		s.ID = id
	}
}

func WithName(name string) StudentOptionFunc {
	return func(s *student) {
		s.Name = name
	}
}

func WithScore(score int) StudentOptionFunc {
	return func(s *student) {
		s.Score = score
	}
}

func WithGrade(grade string) StudentOptionFunc {
	return func(s *student) {
		s.Grade = grade
	}
}

func NewStudent(opts ...StudentOptionFunc) *student {
	s := &student{}
	for _, opt := range opts {
		opt(s)
	}
	return s
}

上面的例子中，構(gòu)造函數(shù)NewStudent()的參數(shù)是不定長(zhǎng)StudentOptionFunc類(lèi)型的函數(shù)，可以看作是一個(gè)裝了多個(gè)函數(shù)的切片，在NewStudent()函數(shù)內(nèi)部遍歷這個(gè)切片，通過(guò)切片中的每個(gè)函數(shù)來(lái)初始化自己的字段。

接下來(lái)在調(diào)用中，我們就可以不受參數(shù)個(gè)數(shù)與順序的影響來(lái)調(diào)用NewStudent()函數(shù)了：

package main

import (
	"app/school"
	"fmt"
)

func main() {
	s1 := school.NewStudent(
		school.WithName("小明"), 
        school.WithScore(90),
        school.WithID(1),
        school.WithGrade(""),
	)	
    
    s2 := school.NewStudent(
		school.WithName("小花"), 
        school.WithScore(90),
	)	
}

折中方案

上面演示了兩種自定義構(gòu)造函數(shù)，一種初始化參數(shù)個(gè)數(shù)與順序完全固定，這種方式太死板了，而一種則是可以自由傳入?yún)?shù)的順序與個(gè)數(shù)，這種方式又太自由了，因?yàn)槲覀兛梢韵胍粋€(gè)折中的方案，即把兩種方式結(jié)合起來(lái)：

func NewStudent(id int, Name string, opts ...StudentOptionFunc) *student {
	s := &student{ID: id, Name: Name}
	for _, opt := range opts {
		opt(s)
	}
	return s
}