| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 package bindings | |
| 6 | |
| 7 import ( | |
| 8 "fmt" | |
| 9 "sync" | |
| 10 | |
| 11 "mojo/public/go/system" | |
| 12 ) | |
| 13 | |
| 14 var errConnectionClosed = &ConnectionError{system.MOJO_RESULT_FAILED_PRECONDITIO
N} | |
| 15 | |
| 16 // ConnectionError represents a error caused by an operation on a message pipe. | |
| 17 type ConnectionError struct { | |
| 18 Result system.MojoResult | |
| 19 } | |
| 20 | |
| 21 func (e *ConnectionError) Error() string { | |
| 22 return fmt.Sprintf("message pipe error: %v", e.Result) | |
| 23 } | |
| 24 | |
| 25 // Closed returnes true iff the error was caused by an operation on a closed | |
| 26 // message pipe. | |
| 27 func (e *ConnectionError) Closed() bool { | |
| 28 return e.Result == system.MOJO_RESULT_FAILED_PRECONDITION | |
| 29 } | |
| 30 | |
| 31 // Connector owns a message pipe handle. It can read and write messages | |
| 32 // from the message pipe waiting on it if necessary. The operation are | |
| 33 // thread-safe. | |
| 34 type Connector struct { | |
| 35 mu sync.RWMutex // protects pipe handle | |
| 36 pipe system.MessagePipeHandle | |
| 37 | |
| 38 done chan struct{} | |
| 39 waitMutex sync.Mutex | |
| 40 waiter AsyncWaiter | |
| 41 waitChan chan WaitResponse | |
| 42 } | |
| 43 | |
| 44 // NewStubConnector returns a new |Connector| instance that sends and | |
| 45 // receives messages from a provided message pipe handle. | |
| 46 func NewConnector(pipe system.MessagePipeHandle, waiter AsyncWaiter) *Connector
{ | |
| 47 return &Connector{ | |
| 48 pipe: pipe, | |
| 49 waiter: waiter, | |
| 50 done: make(chan struct{}), | |
| 51 waitChan: make(chan WaitResponse, 1), | |
| 52 } | |
| 53 } | |
| 54 | |
| 55 // ReadMessage reads a message from message pipe waiting on it if necessary. | |
| 56 func (c *Connector) ReadMessage() (*Message, error) { | |
| 57 // Make sure that only one goroutine at a time waits a the handle. | |
| 58 // We use separate lock so that we can send messages to the message pipe | |
| 59 // while waiting on the pipe. | |
| 60 // | |
| 61 // It is better to acquire this lock first so that a potential queue of | |
| 62 // readers will wait while closing the message pipe in case of Close() | |
| 63 // call. | |
| 64 c.waitMutex.Lock() | |
| 65 defer c.waitMutex.Unlock() | |
| 66 // Use read lock to use pipe handle without modifying it. | |
| 67 c.mu.RLock() | |
| 68 defer c.mu.RUnlock() | |
| 69 | |
| 70 if !c.pipe.IsValid() { | |
| 71 return nil, errConnectionClosed | |
| 72 } | |
| 73 // Check if we already have a message. | |
| 74 result, bytes, handles := c.pipe.ReadMessage(system.MOJO_READ_MESSAGE_FL
AG_NONE) | |
| 75 if result == system.MOJO_RESULT_SHOULD_WAIT { | |
| 76 waitId := c.waiter.AsyncWait(c.pipe, system.MOJO_HANDLE_SIGNAL_R
EADABLE, c.waitChan) | |
| 77 select { | |
| 78 case <-c.waitChan: | |
| 79 result, bytes, handles = c.pipe.ReadMessage(system.MOJO_
READ_MESSAGE_FLAG_NONE) | |
| 80 if result != system.MOJO_RESULT_OK { | |
| 81 return nil, &ConnectionError{result} | |
| 82 } | |
| 83 case <-c.done: | |
| 84 c.waiter.CancelWait(waitId) | |
| 85 return nil, errConnectionClosed | |
| 86 } | |
| 87 } else if result != system.MOJO_RESULT_OK { | |
| 88 return nil, &ConnectionError{result} | |
| 89 } | |
| 90 return ParseMessage(bytes, handles) | |
| 91 } | |
| 92 | |
| 93 // WriteMessage writes a message to the message pipe. | |
| 94 func (c *Connector) WriteMessage(message *Message) error { | |
| 95 // Use read lock to use pipe handle without modifying it. | |
| 96 c.mu.RLock() | |
| 97 defer c.mu.RUnlock() | |
| 98 if !c.pipe.IsValid() { | |
| 99 return errConnectionClosed | |
| 100 } | |
| 101 return WriteMessage(c.pipe, message) | |
| 102 } | |
| 103 | |
| 104 // Close closes the message pipe aborting wait on the message pipe. | |
| 105 // Panics if you try to close the |Connector| more than once. | |
| 106 func (c *Connector) Close() { | |
| 107 // Stop waiting to acquire the lock. | |
| 108 close(c.done) | |
| 109 // Use write lock to modify the pipe handle. | |
| 110 c.mu.Lock() | |
| 111 c.pipe.Close() | |
| 112 c.mu.Unlock() | |
| 113 } | |
| OLD | NEW |