Golang实现Redis(1):Tcp层的应用服务器

main函数为入口,ListenAndServeWithSignal接收address和handle接口,server.MakeHandler()实现了handle接口

func main()  {

	config.SetUpConfig("redis.conf")

	tcp.ListenAndServeWithSignal(
		"0.0.0.0:8989",
		server.MakeHandler())
}
package tcp

import (
	"context"
	"gRedis/interface/tcp"
	"log"
	"net"
	"os"
	"os/signal"
	"sync"
	"syscall"
)

func ListenAndServeWithSignal(address string, handler tcp.Handler)  {
	
	closeChan := make(chan struct{})
	
	// 这里是golang的信号,在这里接收系统的信号,比如ctrl+c就会往一个无缓冲的管道里写入数据
	// 然后通知服务器关闭
	sigCh := make(chan os.Signal)
	signal.Notify(sigCh, syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT)
		go func() {
			sig := <-sigCh
			switch sig {
			case syscall.SIGHUP, syscall.SIGQUIT, syscall.SIGTERM, syscall.SIGINT:
				closeChan <- struct{}{}
			}
		}()
	
	
	listener, err := net.Listen("tcp", address)
	if err !=nil {
		log.Println("出错了")
	}
	
	ListenAndServe(listener,handler,closeChan)
	
}

func ListenAndServe(listener net.Listener, handler tcp.Handler, closeChan <-chan struct{})  {

	//这个接收信号  根据信号来进行关闭
	go func() {
		<-closeChan
		_ = listener.Close()
		_ = handler.Close()
	}()

	defer func() {
		_ = listener.Close()
		_ = handler.Close()
	}()

	ctx := context.Background()
	var waitDone sync.WaitGroup
	for  {
		conn, err := listener.Accept()
		if err != nil {
			return
		}
		log.Println("accept link")
		waitDone.Add(1)
		go func() {
			defer func() {
				waitDone.Done()
			}()
			//最外层for循环就是循环接收数据,
			//这里就是每次循环开启一个协程,然后处理完数据利用defer结束
			handler.Handler(ctx,conn)
		}()
	}
	waitDone.Wait()

}
package server

import (
	"bufio"
	"context"
	"gRedis/interface/db"
	"io"
	"log"
	"net"
)

type Handle struct {
	db db.DB
}

func MakeHandler() *Handle {
	return &Handle{
	}
}

func (this  *Handle) Handler(ctx context.Context,conn net.Conn)  {
	reader := bufio.NewReader(conn)
	for  {
		readString, err := reader.ReadString('\n')
		if err != nil {
			if err  == io.EOF {
				log.Println("connection close")
			} else {
				log.Println(err)
			}
			return
		}
		b := []byte(readString)
		conn.Write(b)
	}
}

func (this *Handle) Close() error  {
	return nil
}
package db

import "gRedis/interface/redis"

type DB interface {
	Exec(client redis.Connection, args [][]byte) redis.Reply
}

package redis

type Reply interface {
	ToBytes() []byte
}

package tcp

import (
	"context"
	"net"
)

type Handler interface {
	Handler(ctx context.Context,conn net.Conn)
	Close() error
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值