zhangmeng
2019-08-26 3bd1f29975c0eaa6af8c99776b099faafbbfc250
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// Copyright 2016 Tom Thorogood. All rights reserved.
// Use of this source code is governed by a
// Modified BSD License license that can be found in
// the LICENSE file.
 
package net
 
import (
    "errors"
    "net"
    "sync"
 
    "github.com/tmthrgd/shm-go"
)
 
type Dialer struct {
    rw   *shm.ReadWriteCloser
    name string
 
    mut sync.Mutex
}
 
func Dial(name string) (net.Conn, error) {
    rw, err := shm.OpenDuplex(name)
    if err != nil {
        return nil, err
    }
 
    return (&Dialer{
        rw:   rw,
        name: name,
    }).Dial("shm", name)
}
 
func NewDialer(rw *shm.ReadWriteCloser, name string) *Dialer {
    return &Dialer{
        rw:   rw,
        name: name,
    }
}
 
func (d *Dialer) Dial(network, address string) (net.Conn, error) {
    if network != "shm" {
        return nil, errors.New("unrecognised network")
    }
 
    if address != d.name {
        return nil, errors.New("invalid address")
    }
 
    d.mut.Lock()
    return &Conn{d.rw, d.name, &d.mut}, nil
}