ICode9

精准搜索请尝试: 精确搜索
首页 > 数据库> 文章详细

Golang示例续期锁:Redis+Channel+sync.Mutex

2022-06-21 16:35:49  阅读:177  来源: 互联网

标签:return func err 示例 redis Redis sync key time


package main

import (
	"context"
	"fmt"
	"github.com/go-redis/redis"
	"golang.org/x/sync/errgroup"
	"log"
	"sync"
	"time"
)

func main() {
	NewRedis()
	fu1()

	time.Sleep(100 * time.Second)
}

var rdb *redis.Client
var ctx = context.Background()
var mutex sync.Mutex

func NewRedis() {
	rdb = redis.NewClient(&redis.Options{
		Addr:     "127.0.0.1:6379",
		Password: "", // no password set
		DB:       0,  // use default DB
	})
}
func Lock(key string) error {
	mutex.Lock()
	defer mutex.Unlock()
	_, err := rdb.SetNX(key, 1, 1*time.Second).Result()
	if err != nil {
		log.Println(err.Error())
	}
	return err
}
func UnLock(key string) error {
	_, err := rdb.Del(key).Result()
	if err != nil {
		log.Println(err.Error())
		return err
	}
	return err
}
func Expire(key string) error {
	_, err := rdb.Expire(key, 1*time.Second).Result()
	if err != nil {
		log.Println(err.Error())
		return err
	}
	return err
}
func fu1() error {
	ch := make(chan bool)
	// 加锁
	err := Lock("lock_key")
	if err != nil {
		return err
	}
	//解锁
	defer func() {
		err = UnLock("lock_key")
		if err != nil {
			fmt.Println(err.Error())
			return
		}
		fmt.Println("release redis lock success")
	}()

	g, _ := errgroup.WithContext(context.Background())
	//...主业务代码
	g.Go(func() error {
		time.Sleep(15 * time.Second)
		ch <- true
		return nil
	})

	//锁续期
	g.Go(func() error {
		ticker := time.NewTicker(time.Second * 1)
		for {
			select {
			// 任务还没执行完 每秒续期
			case <-ticker.C:
				Expire("lock_key")
				fmt.Println(time.Now())

				//收到执行完的 channel 就关闭time定时任务
			case <-ch:
				ticker.Stop()
				return nil
			}

		}
	})
	//等待信号量
	if err = g.Wait(); err != nil {
		return err
	}
	close(ch)
	return nil
}

  

标签:return,func,err,示例,redis,Redis,sync,key,time
来源: https://www.cnblogs.com/Biuget-Golang/p/16397260.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有