Mediator

maligh · maligh · commit 7854da99c8c0 · 2017-10-12T23:52:45.000+08:00
Mediator
diff --git a/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/ChatRoom.h b/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/ChatRoom.h
@@ -7,7 +7,10 @@
 //
 
 #import <Foundation/Foundation.h>
+#import "MediatorProtocol.h"
 
-@interface ChatRoom : NSObject
+@interface ChatRoom : NSObject<MediatorProtocol>
+
+- (void)showMessage:(NSString *)message userName:(NSString *)name;
 
 @end
diff --git a/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/ChatRoom.m b/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/ChatRoom.m
@@ -10,4 +10,13 @@
 
 @implementation ChatRoom
 
+- (void)showMessage:(NSString *)message {
+    NSLog(@"%@\n",message);
+}
+
+- (void)showMessage:(NSString *)message userName:(NSString *)name {
+    NSString *string = [NSString stringWithFormat:@"%@:%@", name, message];
+    [self showMessage:string];
+}
+
 @end
diff --git a/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/MediatorProtocol.h b/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/MediatorProtocol.h
@@ -10,4 +10,6 @@
 
 @protocol MediatorProtocol <NSObject>
 
+- (void)showMessage:(NSString *)message;
+
 @end
diff --git a/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/User.h b/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/User.h
@@ -7,7 +7,12 @@
 //
 
 #import <Foundation/Foundation.h>
+#import "ChatRoom.h"
 
 @interface User : NSObject
 
+- (instancetype)initWithName:(NSString *)name room:(ChatRoom *)room;
+
+- (void)sendMessage:(NSString *)message;
+
 @end
diff --git a/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/User.m b/MLDesignPatterns-OC/MLDesignPatterns-OC/Mediator/User.m
@@ -8,6 +8,25 @@
 
 #import "User.h"
 
+@interface User()
+
+@property (nonatomic, copy) NSString *name;     ///< 用户昵称
+@property (nonatomic, strong) ChatRoom *room;   ///< 当前聊天室
+
+@end
+
 @implementation User
 
+- (instancetype)initWithName:(NSString *)name room:(ChatRoom *)room {
+    if (self = [super init]) {
+        _name = name;
+        _room = room;
+    }
+    return self;
+}
+
+- (void)sendMessage:(NSString *)message {
+    [_room showMessage:message userName:_name];
+}
+
 @end
