[Golang] 实用小示例

2340

本文章会持续更新更多golang常用小技巧；

1. 单例模式

  • 多用于只实例化一次，如配置模块，日志模块等；
package main

import (
	"fmt"
	"sync"
)

var cc *WebConfig
var once sync.Once

type WebConfig struct {
	Port int `json:"port"`
}

func GetConfig() *WebConfig  {
	once.Do(func() {
		cc = &WebConfig{Port: 8080}
	})
	return cc
}

func main()  {
	cfg1 := GetConfig()
	cfg2 := GetConfig()
	fmt.Println(cfg1 == cfg2)
}

复制代码

2. 函数执行超时控制


package main

import (
	"fmt"
	"time"
)
//1 .业务过程放到协程
// 2、把业务结果塞入channel
func job() chan string{
	ret:=make(chan string)
	go func() {
		time.Sleep(time.Second*2)
		ret<- "success"
	}()
	return ret
}
func run() (interface{},  error) {
	c:=job()
	select {
	case r:=<-c:
		return r,nil
	case <-time.After(time.Second*3):
		return nil,fmt.Errorf("time out")
	}

}
func main()  {
	fmt.Println(run())
}

复制代码

3. 反射使用示例

  • 主要理清楚 reflect.Kind 和 reflect.Value 的用法
package main

import (
	"fmt"
	"reflect"
)

type User struct {
	UserId int `name:"uid" bcd:"3456"`
	UserName string
}
func Map2Struct(m map[string]interface{},u interface{})  {
	v:=reflect.ValueOf(u)
	if v.Kind()==reflect.Ptr{
		v=v.Elem()
		if v.Kind()!=reflect.Struct{
			panic("must struct")
		}
		findFromMap:= func(key string,nameTag string ) interface {}{
			for k,v:=range m{
				if k==key || k==nameTag {
					return v
				}
			}
			return nil
		}
		for i:=0;i<v.NumField();i++{
			get_value:=findFromMap(v.Type().Field(i).Name,v.Type().Field(i).Tag.Get("name"))
			if get_value!=nil && reflect.ValueOf(get_value).Kind()==v.Field(i).Kind(){
				v.Field(i).Set(reflect.ValueOf(get_value))
			}
		}
	}else{
		panic("must ptr")
	}
}

func main()  {
	u:=&User{}
	m:=map[string]interface{}{
		"id":123,
		"uid":101,
		"UserName":"shenyi",
		"age":19,
	}
	Map2Struct(m,u)
	fmt.Printf("%+v", u)
}
复制代码

4. 构造函数技巧

  • 作用是提高代码可维护性和阅读性，但是这种写法相对的会损失一些性能
package main

import (
	"fmt"
)

type UserAttrFunc func(*User) //设置User属性的 函数类型
type UserAttrFuncs []UserAttrFunc
func(this UserAttrFuncs) apply(u *User)  {
	for _,f:=range this{
		f(u)
	}
}
func WithUserID(id int) UserAttrFunc  {
	return func(u *User) {
		u.Id=id
	}
}
func WithUserName(name string) UserAttrFunc  {
	return func(u *User) {
		u.Name=name
	}
}

type User struct {
	Id int
	Name string
	Sex byte
}
// 有选择性的对ID 进行赋值
func NewUser(fs ...UserAttrFunc) *User  {
	u:= new(User)
	UserAttrFuncs(fs).apply(u)
	return u
}

func main()  {
	u:=NewUser(
		WithUserName("shenyi"),
		WithUserID(105),
	)
	fmt.Println(u)

}
复制代码

5. 简单工厂模式

package main

import "fmt"

type UserType int 

type User interface {
	GetRole() string
}
type Member struct {}
func(this *Member) GetRole() string {
	return   "会员用户"
}
type Admin struct {}
func(this *Admin) GetRole() string {
	return  "后台管理用户"
}

// 枚举类型
const (
	Mem UserType = iota
	Adm
)
func CreateUser(t UserType) User{
	switch t {
	case Mem:
		return new(Member)
	case Adm:
		return new(Admin)
	default:
		return new(Member)
	}
}

func main()  {
	fmt.Println(CreateUser(Adm).GetRole())
}
复制代码

6. 抽象工厂模式

  • 抽象工厂模式，主要也是在多态情况下的一种高维护性和扩展性的设计模式，符合高层代码不应以来具体模块的实现而应依赖于抽象这么一个设计思想；
package main

import "fmt"


type User interface {
	GetRole() string
}
type Member struct {}
func(this *Member) GetRole() string {
	return   "会员用户"
}
type Admin struct {}
func(this *Admin) GetRole() string {
	return  "后台管理用户"
}
const (
	Mem=iota
	Adm
)
func CreateUser(t int) User{
	switch t {
	case Mem:
		return new(Member)
	case Adm:
		return new(Admin)
	default:
		return new(Member)
	}
}

type AbstractFactory interface {
	CreateUser() User
}
type MemberFactory struct {}
func(this *MemberFactory) CreateUser() User{
	return &Member{}
}
type AdminFactory struct {}
func(this *AdminFactory) CreateUser() User{
	return &Admin{}
}

func main()  {
	var fact AbstractFactory=new(AdminFactory)
	fmt.Println(fact.CreateUser().GetRole())
}
复制代码

7. 装饰器模式

  • 装饰器是体现AOP思想的一种设计模式；在无需改动原有方法的前提下添加新功能；
package main

import "net/http"


func CheckLogin(f http.HandlerFunc) http.HandlerFunc{
	return func(writer http.ResponseWriter, request *http.Request) {
		 if request.URL.Query().Get("token")==""{
			 writer.Write([]byte("token error"))
		 }else{
			 f(writer,request)
		 }

	}
}
func index(writer http.ResponseWriter, request *http.Request) {
	writer.Write([]byte("index"))
}
func main()  {

	 http.HandleFunc("/",CheckLogin(index) )
	 http.ListenAndServe(":8080",nil)

}
复制代码

8. 简易前缀树

  • 多用于在路由框架，进行路由匹配
package main

import "fmt"

type Node struct {
    isend bool  //是否最后一个
    Children map[string]*Node  //这一步 用map就省的遍历了
}
func NewNode() *Node {
    return &Node{Children:make(map[string]*Node)}
}

type Trie struct {
    root *Node
}
func NewTrie() *Trie {
    return &Trie{root:NewNode()}
}

func(this *Trie) Insert(str string ){
    // 第一个肯定是空节点
    current:=this.root
    for _,item:=range ([]rune)(str){
        if _,ok:=current.Children[string(item)];!ok{
            current.Children[string(item)]=NewNode()
        }
        current=current.Children[string(item)]
    }
    current.isend=true   //最后一个
}

func(this *Trie) Search(str string ) bool{
    current:=this.root
    for _,item:=range ([]rune)(str){
        if _,ok:=current.Children[string(item)];!ok{
            return  false
        }
        current=current.Children[string(item)]
    }
    return current.isend    //最后一个
}


func test()  {
    strs:=[]string{"go","gin","golang","goapp","guest"}
    tree:=NewTrie()
    for _,s:=range strs{
        tree.Insert(s)
    }

    // 最终一个都匹配不到
    strs=append(strs,"gi","gogo","gia")
    for _,s:=range strs{
        fmt.Println(tree.Search(s))
    }

}
func main()  {

    test()
}

复制代码

9. 行读取文件或字符串

package main

import (
	"bufio"
	"bytes"
	"fmt"
	"log"
	"strings"
)

func dropCR(data []byte) []byte {
	if len(data) > 0 && data[len(data)-1] == ':' {
		return data[0 : len(data)-1]
	}
	return data
}

func split(data []byte, atEOF bool) (advance int, token []byte, err error) {
	if atEOF && len(data) == 0 {
		return 0, nil, nil
	}
	if i := bytes.IndexByte(data, ':'); i >= 0 {
		// We have a full newline-terminated line.
		return i + 1, dropCR(data[0:i]), nil
	}
	// If we're at EOF, we have a final, non-terminated line. Return it.
	if atEOF {
		return len(data), dropCR(data), nil
	}
	// Request more data.
	return 0, nil, nil
}

func main() {
	/*
	Scanner 方法
	NewScanner 创建 Scanner
	Scanner.Split 设置处理函数
	Scanner.Scan 获取当前token, 扫描下一token
	Scanner.Bytes 将token以 []byte 的形式返回
	Scanner.Text 将token以 string 的形式返回
	Scanner.Err 获取处理方法返回的错误
	*/

	reader := strings.NewReader("aaa:bbb:ccc:ddd:eee:fff:ggg")
	scanner := bufio.NewScanner(reader)

	/*
	ScanBytes 将token处理为单一字节
	ScanRunes 将token处理为utf-8编码的unicode码
	ScanWords 以空格分隔
	tokenScanLines 以换行符分分割token
	*/
	// 替换原有默认以'\n'的形式切割，上面两端代码是模仿源码的分割函数
	scanner.Split(split)
	count := 0
	for scanner.Scan() {
		log.Println(scanner.Text())
		count++
	}
	fmt.Println("一共有", count, "行")

}

复制代码
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
后端
喜欢就支持一下吧
点赞0 分享
AI-robot的头像-一一网
asgbookphp ‘index.php’ 跨站脚本攻击漏洞-一一网
asgbookphp ‘index.php’ 跨站脚本攻击漏洞
asgbookphp ‘index.php’ 跨站脚本攻击漏洞
14年前 2.6W+
机器学习之随机森林回归篇（RandomForestRegressor）-一一网
机器学习之随机森林回归篇（RandomForestRegressor）
机器学习之随机森林回归篇（RandomForestRegressor）
4年前 4202
详解数仓中的数据分层：ODS、DWD、DWM、DWS、ADS-一一网
详解数仓中的数据分层：ODS、DWD、DWM、DWS、ADS
详解数仓中的数据分层：ODS、DWD、DWM、DWS、ADS
4年前 3895
DZCP Clanportal ＇Index.PHP＇任意文件上载漏洞-一一网
DZCP Clanportal ＇Index.PHP＇任意文件上载漏洞
DZCP Clanportal ＇Index.PHP＇任意文件上载漏洞
19年前 3842
Wireshark之抓包文件保存-一一网
Wireshark之抓包文件保存
Wireshark之抓包文件保存
4年前 3406
支付宝生活号H5接入支付宝支付流程-一一网
支付宝生活号H5接入支付宝支付流程
支付宝生活号H5接入支付宝支付流程
3年前 3135
相关推荐
Wireshark之抓包文件保存-一一网
Wireshark之抓包文件保存
Wireshark之抓包文件保存
4年前 3406
Integer的最大值是什么？｜ Java Debug 笔记-一一网
Integer的最大值是什么？｜ Java Debug 笔记
Integer的最大值是什么？｜ Java Debug 笔记
4年前 2767
使用arthas trace命令查看代码各方法响应时间-一一网
使用arthas trace命令查看代码各方法响应时间
使用arthas trace命令查看代码各方法响应时间
4年前 2722
Spring Boot集成Druid异常discard long time none received connection.-一一网
Spring Boot集成Druid异常discard long time none received connection.
Spring Boot集成Druid异常discard long time none received connection.
4年前 2137
pandas系列之导出为.csv文件-一一网
pandas系列之导出为.csv文件
pandas系列之导出为.csv文件
4年前 2109
使用XXL-JOB时，如何避免多台服务器重复调度任务？｜工作踩坑系列-一一网
使用XXL-JOB时，如何避免多台服务器重复调度任务？｜工作踩坑系列
使用XXL-JOB时，如何避免多台服务器重复调度任务？｜工作踩坑系列
4年前 1681
搜索
开启精彩搜索
世界，您好！-一一网
3.7W+人已阅读
世界，您好！
TOP1
番茄社区无限观看+付费视频破解教程-一一网
番茄社区无限观看+付费视频破解教程
5年前3.2W+人已阅读
TOP2
香蕉视频v3.8.3_VIP会员破解教程-一一网
香蕉视频v3.8.3_VIP会员破解教程
5年前3.1W+人已阅读
TOP3
[桜井宁宁]COS和泉纱雾超可爱写真福利集-一一网
密码保护：[桜井宁宁]COS和泉纱雾超可爱写真福利集
5年前2.8W+人已阅读
TOP4
[桜井宁宁] 爆乳奶牛少女cos写真-一一网
密码保护：[桜井宁宁] 爆乳奶牛少女cos写真
5年前2.8W+人已阅读
TOP5
快猫v1.1.7会员破解版_最新快猫vip破解教程大全-一一网
快猫v1.1.7会员破解版_最新快猫vip破解教程大全
5年前2.7W+人已阅读
TOP6
标签云
默认路由黑裙黑群晖麦克风车机鸿蒙系统鸿蒙系统鸿蒙OS鸿蒙驱动软件驱动题库音频剪辑音乐剪辑面板面具静态雷石集群随机阿里云