[59] | 1 | import threading
|
---|
| 2 | from typing import TypeVar, List, cast, Optional
|
---|
| 3 |
|
---|
| 4 | from tudelft.utilities.listener.Listener import Listener
|
---|
| 5 | from tudelft_utilities_logging.ReportToLogger import ReportToLogger
|
---|
| 6 | from tudelft_utilities_logging.Reporter import Reporter
|
---|
| 7 |
|
---|
| 8 | from geniusweb.actions.Action import Action
|
---|
| 9 | from geniusweb.connection.ConnectionEnd import ConnectionEnd
|
---|
| 10 | from geniusweb.inform.Inform import Inform
|
---|
| 11 | from geniusweb.party.Party import Party
|
---|
| 12 |
|
---|
| 13 |
|
---|
| 14 |
|
---|
| 15 | class DefaultParty (Party, Listener[Inform]):
|
---|
| 16 | '''
|
---|
| 17 | Party with default implementation to handle the connection.
|
---|
| 18 | '''
|
---|
| 19 |
|
---|
| 20 | def __init__(self, reporter:Reporter=None):
|
---|
| 21 | '''
|
---|
| 22 | @param reporter the reporter to use. Generated automatically if None
|
---|
| 23 | '''
|
---|
| 24 | if not reporter:
|
---|
| 25 | reporter = ReportToLogger(self.__class__.__name__)
|
---|
| 26 | if not isinstance(reporter, Reporter):
|
---|
| 27 | raise ValueError("reporter must be instance of reporter, got "+str(reporter))
|
---|
| 28 | self._reporter=reporter
|
---|
| 29 | self._connection: Optional[ConnectionEnd[Inform,Action]] = None
|
---|
| 30 | self._lock = threading.Lock()
|
---|
| 31 |
|
---|
| 32 | def connect(self, connection: ConnectionEnd[Inform, Action] ):
|
---|
| 33 | try:
|
---|
| 34 | self._lock.acquire()
|
---|
| 35 | if self._connection:
|
---|
| 36 | raise ValueError("Already connected")
|
---|
| 37 | self._connection = connection
|
---|
| 38 | self._connection.addListener(self)
|
---|
| 39 | finally:
|
---|
| 40 | self._lock.release()
|
---|
| 41 |
|
---|
| 42 | #Override
|
---|
| 43 | def disconnect(self):
|
---|
| 44 | self._lock.acquire()
|
---|
| 45 | if self._connection:
|
---|
| 46 | self._connection.removeListener(self)
|
---|
| 47 | self._connection.close()
|
---|
| 48 | self._connection = None #type:ignore
|
---|
| 49 | self._lock.release()
|
---|
| 50 |
|
---|
| 51 |
|
---|
| 52 | #Override
|
---|
| 53 | def terminate(self):
|
---|
| 54 | self.disconnect();
|
---|
| 55 |
|
---|
| 56 |
|
---|
| 57 | def getConnection(self) -> Optional[ConnectionEnd[Inform, Action]]:
|
---|
| 58 | '''
|
---|
| 59 | @return currently available connection, or null if not currently
|
---|
| 60 | connected.
|
---|
| 61 | '''
|
---|
| 62 | return self._connection;
|
---|
| 63 |
|
---|
| 64 |
|
---|
| 65 | def getReporter(self)->Reporter:
|
---|
| 66 | return self._reporter;
|
---|
| 67 |
|
---|