コード例 #1
0
ファイル: main.go プロジェクト: johnvilsack/golang-stuff
// Accept forward requests
func StreamForwardRequests(x0 x.Listener) <-chan *request {
	ch := make(chan *request)
	go func() {
		for {
			conn := x0.Accept()
			req, err := conn.Read()
			if err != nil {
				conn.Write(&proto.XError{err.Error()})
				conn.Close()
				continue
			}
			fwdreq, ok := req.(*proto.XDashboardQuery)
			if !ok {
				conn.Write(&proto.XError{"unknown dashboard request"})
				conn.Close()
				continue
			}
			ch <- &request{
				Query: fwdreq,
				ReturnResponse: func(posts []*proto.Post, err error) {
					if err != nil {
						conn.Write(&proto.XError{err.Error()})
					} else {
						conn.Write(&proto.XDashboardQuerySuccess{Posts: posts})
					}
					conn.Close()
				},
			}
		}
	}()
	return ch
}
コード例 #2
0
ファイル: main.go プロジェクト: johnvilsack/golang-stuff
func StreamX(x0 x.Listener) (<-chan *createRequest, <-chan *queryRequest) {
	cch, qch := make(chan *createRequest), make(chan *queryRequest)
	go func() {
		for {
			conn := x0.Accept()
			req, err := conn.Read()
			if err != nil {
				conn.Write(&proto.XError{err.Error()})
				conn.Close()
				continue
			}
			switch q := req.(type) {
			case *proto.XCreatePost:
				cch <- &createRequest{
					Forwarded: true,
					Post:      q,
					ReturnResponse: func(err error) {
						if err != nil {
							conn.Write(&proto.XError{err.Error()})
							conn.Close()
							return
						}
						conn.Write(&proto.XSuccess{})
						conn.Close()
					},
				}
			case *proto.XTimelineQuery:
				qch <- &queryRequest{
					Query: q,
					ReturnResponse: func(posts []int64, err error) {
						if err != nil {
							conn.Write(&proto.XError{err.Error()})
							conn.Close()
							return
						}
						conn.Write(&proto.XTimelineQuerySuccess{Posts: posts})
						conn.Close()
					},
				}
			default:
				panic(fmt.Sprintf("unknown request to timeline: %#v", req))
				conn.Write(&proto.XError{"unknown request"})
				conn.Close()
			}
		}
	}()
	return cch, qch
}