()
| 108 | } |
| 109 | |
| 110 | func (c *agentClient) read() { |
| 111 | for { |
| 112 | select { |
| 113 | case <-c.done: |
| 114 | return |
| 115 | default: |
| 116 | _, b, err := c.conn.ReadMessage() |
| 117 | if err != nil { |
| 118 | return |
| 119 | } |
| 120 | |
| 121 | msg := &livekit.ServerMessage{} |
| 122 | if err = proto.Unmarshal(b, msg); err != nil { |
| 123 | return |
| 124 | } |
| 125 | |
| 126 | switch m := msg.Message.(type) { |
| 127 | case *livekit.ServerMessage_Assignment: |
| 128 | go c.handleAssignment(m.Assignment) |
| 129 | case *livekit.ServerMessage_Availability: |
| 130 | go c.handleAvailability(m.Availability) |
| 131 | case *livekit.ServerMessage_Register: |
| 132 | go c.handleRegister(m.Register) |
| 133 | } |
| 134 | } |
| 135 | } |
| 136 | } |
| 137 | |
| 138 | func (c *agentClient) handleAssignment(req *livekit.JobAssignment) { |
| 139 | switch req.Job.Type { |
no test coverage detected