一個簡單的socket通信小demo

寫了一個socket的程序，可以和本地的服務器進行通信，要先和服務器建立鏈接，然后發送登錄信息，驗證成功，就可以和服務器通信了

1 頁面截圖

2 點擊鏈接服務器，可以鏈接服務器，服務器的ip地址為：127.0.0.1 端口為 12345；

3 點擊鏈接服務器之后，打印信息如下

4 點擊登錄，會向服務器發送登錄信息： iam:zhangsan, 然后會客戶端會顯示登錄成功

5 比如輸入chongqingyoudian?

6 在服務器控制臺就能看到客戶端發送的信息

客戶端代碼如下

//
//  ViewController.m
//  05.聊天室
//
//  Created by apple on 14/12/5.
//  Copyright (c) 2014年 heima. All rights reserved.
//

#import "ViewController.h"

@interface ViewController ()<NSStreamDelegate,UITextFieldDelegate,UITableViewDataSource,UITableViewDelegate>{
    NSInputStream *_inputStream;//對應輸入流
    NSOutputStream *_outputStream;//對應輸出流
}
@property (weak, nonatomic) IBOutlet NSLayoutConstraint *inputViewConstraint;
@property (weak, nonatomic) IBOutlet UITableView *tableView;

@property (nonatomic, strong) NSMutableArray *chatMsgs;//聊天消息數組

@end

@implementation ViewController

-(NSMutableArray *)chatMsgs{
    if (!_chatMsgs) {
        _chatMsgs = [NSMutableArray array];
    }
    
    return _chatMsgs;
}

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view, typically from a nib.
  
    
    // 2.收發數據
    // 做一個聊天
    // 1.用戶登錄
    // 2.收發數據
    
    // 監聽鍵盤
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(kbFrmWillChange:) name:UIKeyboardWillChangeFrameNotification object:nil];
}


-(void)kbFrmWillChange:(NSNotification *)noti{
    NSLog(@"%@",noti.userInfo);
    
    // 獲取窗口的高度
    
    CGFloat windowH = [UIScreen mainScreen].bounds.size.height;
    
   
    
    // 鍵盤結束的Frm
    CGRect kbEndFrm = [noti.userInfo[UIKeyboardFrameEndUserInfoKey] CGRectValue];
     // 獲取鍵盤結束的y值
    CGFloat kbEndY = kbEndFrm.origin.y;
    
    
    self.inputViewConstraint.constant = windowH - kbEndY;
}

-(void)stream:(NSStream *)aStream handleEvent:(NSStreamEvent)eventCode{
    NSLog(@"%@",[NSThread currentThread]);

//    NSStreamEventOpenCompleted = 1UL << 0,//輸入輸出流打開完成
//    NSStreamEventHasBytesAvailable = 1UL << 1,//有字節可讀
//    NSStreamEventHasSpaceAvailable = 1UL << 2,//可以發放字節
//    NSStreamEventErrorOccurred = 1UL << 3,// 連接出現錯誤
//    NSStreamEventEndEncountered = 1UL << 4// 連接結束
    switch (eventCode) {
        case NSStreamEventOpenCompleted:
            NSLog(@"登錄服務器成功，輸入輸出流打開完成");
            break;
        case NSStreamEventHasBytesAvailable:
            NSLog(@"有字節可讀");
            [self readData];
            break;
        case NSStreamEventHasSpaceAvailable:
            NSLog(@"可以發送字節");
            break;
        case NSStreamEventErrorOccurred:
                        NSLog(@" 連接出現錯誤");
            break;
        case NSStreamEventEndEncountered:
             NSLog(@"連接結束");
            
            // 關閉輸入輸出流
            [_inputStream close];
            [_outputStream close];
            
            // 從主運行循環移除
            [_inputStream removeFromRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
            [_outputStream removeFromRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
            break;
        default:
            break;
    }
    
}
#pragma mark 在這里先和服務器建立鏈接
- (IBAction)connectToHost:(id)sender {
    // 1.建立連接
    NSString *host = @"127.0.0.1";
    int port = 12345;
    
    // 定義C語言輸入輸出流
    CFReadStreamRef readStream;
    CFWriteStreamRef writeStream;
    CFStreamCreatePairWithSocketToHost(NULL, (__bridge CFStringRef)host, port, &readStream, &writeStream);
    
    // 把C語言的輸入輸出流轉化成OC對象
    _inputStream = (__bridge NSInputStream *)(readStream);
    _outputStream = (__bridge NSOutputStream *)(writeStream);
    
    
    // 設置代理
    _inputStream.delegate = self;
    _outputStream.delegate = self;
    
    
    // 把輸入輸入流添加到主運行循環
    // 不添加主運行循環 代理有可能不工作
    [_inputStream scheduleInRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
    [_outputStream scheduleInRunLoop:[NSRunLoop mainRunLoop] forMode:NSDefaultRunLoopMode];
    
    // 打開輸入輸出流
    [_inputStream open];
    [_outputStream open];
}

#pragma mark 在這里發送登錄的信息
- (IBAction)loginBtnClick:(id)sender {
    
    // 登錄
    // 發送用戶名和密碼
    // 在這里做的時候，只發用戶名，密碼就不用發送
    
    // 如果要登錄，發送的數據格式為 "iam:zhangsan";
    // 如果要發送聊天消息，數據格式為 "msg:did you have dinner";
    
    //登錄的指令
    NSString *loginStr = @"iam:zhangsan";
    
    //把Str轉成NSData
    NSData *data = [loginStr dataUsingEncoding:NSUTF8StringEncoding];

    
    [_outputStream write:data.bytes maxLength:data.length];
}

#pragma mark 讀了服務器返回的數據
-(void)readData{
    
    //建立一個緩沖區 可以放1024個字節
    uint8_t buf[1024];
    
    // 返回實際裝的字節數
    NSInteger len = [_inputStream read:buf maxLength:sizeof(buf)];

    // 把字節數組轉化成字符串
    NSData *data = [NSData dataWithBytes:buf length:len];
    
    // 從服務器接收到的數據
    NSString *recStr =  [[NSString alloc] initWithData:data encoding:NSUTF8StringEncoding];
    
    NSLog(@"%@",recStr);
    
    [self reloadDataWithText:recStr];
    
}

-(BOOL)textFieldShouldReturn:(UITextField *)textField{
    
    NSString *text = textField.text;
    
    NSLog(@"%@",text);
    // 聊天信息
    NSString *msgStr = [NSString stringWithFormat:@"msg:%@",text];
    
    //把Str轉成NSData
    NSData *data = [msgStr dataUsingEncoding:NSUTF8StringEncoding];
    
    // 刷新表格
    [self reloadDataWithText:msgStr];
    
    // 發送數據
    [_outputStream write:data.bytes maxLength:data.length];
    
    // 發送完數據，清空textField
    textField.text = nil;
    
    return YES;
}

-(void)reloadDataWithText:(NSString *)text{
    [self.chatMsgs addObject:text];
    
    [self.tableView reloadData];
    
    // 數據多，應該往上滾動
    NSIndexPath *lastPath = [NSIndexPath indexPathForRow:self.chatMsgs.count - 1 inSection:0];
    [self.tableView scrollToRowAtIndexPath:lastPath atScrollPosition:UITableViewScrollPositionBottom animated:YES];
}

#pragma mark 表格的數據源

-(NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section{
    return self.chatMsgs.count;
}


- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath
{
    static NSString *ID = @"Cell";
    UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:ID];
   
    cell.textLabel.text = self.chatMsgs[indexPath.row];
    
    return cell;
}

-(void)scrollViewWillBeginDragging:(UIScrollView *)scrollView{
    [self.view endEditing:YES];
}
@end

服務器端代碼

from twisted.internet.protocol import Protocol, Factory
from twisted.internet import reactor


class IphoneChat(Protocol):
    def connectionMade(self):
        #self.transport.write("""connected""")
        self.factory.clients.append(self)
        print "clients are ", self.factory.clients
    
    def connectionLost(self, reason):
        self.factory.clients.remove(self)
    
    def dataReceived(self, data):
        #print "data is ", data
        a = data.split(':')
        if len(a) > 1:
            command = a[0]
            content = a[1]
            
            msg = ""
            if command == "iam":
                self.name = content
                msg = self.name + " has joined"
                
            elif command == "msg":
                msg = self.name + ": " + content
            
            print msg
                        
            for c in self.factory.clients:
                c.message(msg)
                
    def message(self, message):
        self.transport.write(message + '\n')


factory = Factory()
factory.protocol = IphoneChat
factory.clients = []

reactor.listenTCP(12345, factory)
print "Iphone Chat server started"
reactor.run()

?