import (
	"errors"
	"fmt"
	"github.com/BlueStorm001/codec"
	"log"
	"net"
	"sync"
	"time"
)

type Bootstrap struct {
	conn          net.Conn
	Addr          string
	codec         *codec.Packer
	Receive       func(bootstrap *Bootstrap, body []byte)
	MaxIdleTime   time.Time
	mu            sync.Mutex
	ConnectStatus bool
}

func New(addr string) *Bootstrap {
    return &Bootstrap{
        Addr: addr, 
        codec: codec.NewPacketFieldLength(4)，// netty 解包
    }
}

func (bootstrap *Bootstrap) Connect() (err error) {
	if bootstrap.Receive == nil {
		return errors.New("没有回调方法")
	}
	tcpAddr, _ := net.ResolveTCPAddr("tcp", bootstrap.Addr)
	if bootstrap.conn, err = net.Dial("tcp", tcpAddr.String()); err != nil {
		return err
	}
	bootstrap.MaxIdleTime = time.Now()
	go bootstrap.Read()
	return
}

// InspectStatus 状态检查
// 主要是利用服务器向客户端发起心跳包或数据记载时间
func (bootstrap *Bootstrap) InspectStatus() bool {
	if time.Now().Sub(bootstrap.MaxIdleTime).Seconds() > 30 {
		return false
	}
	return true
}
func (bootstrap *Bootstrap) Read() {
	for {
		var data = make([]byte, 1024*10)
		n, err := bootstrap.conn.Read(data)
		if err != nil {
			fmt.Println(err)
			return
		}
		if bootstrap.codec.Receiver == nil {
			bootstrap.codec.Receiver = bootstrap.receive
		}
		bootstrap.MaxIdleTime = time.Now()
		bootstrap.codec.PacketFieldLengthDecode(data[:n])
	}
}

func (bootstrap *Bootstrap) receive(data []byte) {
	go bootstrap.Receive(bootstrap, data)
}

func (bootstrap *Bootstrap) Send(data []byte) (err error) {
	if bootstrap.conn == nil {
		return errors.New("链接失败")
	}
	bootstrap.mu.Lock()
	//netty 组包
	if _, err = bootstrap.conn.Write(bootstrap.codec.PacketFieldLengthEncode(data)); err != nil {
		bootstrap.ConnectStatus = false
	}
	bootstrap.mu.Unlock()
	return
}

func (bootstrap *Bootstrap) Close() {
	bootstrap.MaxIdleTime = time.Now().Add(time.Hour * -1)
	if bootstrap.conn == nil {
		return
	}
	bootstrap.conn.Close()
}



func main() {
    //以下是客户端链接服务器的测试
    bootstrap := New("127.0.0.1:8088")
    if err := bootstrap.Connect(); err == nil {
        bootstrap.Receive = func(bootstrap *Bootstrap, body []byte) {
            fmt.Println(body)
            if err = bootstrap.Send(body); err != nil {
                fmt.Println(err)
            }
        }
        bootstrap.Send([]byte("ok"))
    }
}