Exemple #1
0
func (FSR *FileServer) serverFile() {
	ch, err := FSR.RN.CommitChannel()
	if err != nil {
		panic("[Error] No CommitChannel Received")
	}

	lastAppliedIndex := 0
	for {
		commitInfo := <-ch

		if commitInfo.Err != nil {
			msg, err := decode(commitInfo.Data)
			if err != nil {
				panic("[Error] Decode error")
			}
			conn, ok := FSR.mapFind(msg.ClientId)
			if !ok {
				panic("[Error] Client not found for this msg")
			}

			switch commitInfo.Err.Error() {
			case "ERR_CONTACT_LEADER":
				ldrHost := getHost(commitInfo.LeaderId)
				ldrClientPort := getClientPort(commitInfo.LeaderId)
				reply(conn, &fs.Msg{Kind: 'R', ReDirAddr: ldrHost + ":" + strconv.Itoa(ldrClientPort)})
			case "ERR_TRY_LATER":
				reply(conn, &fs.Msg{Kind: 'T'})
			default:
				panic("[Error] Unknown error - " + commitInfo.Err.Error())
			}
		} else {
			for i := lastAppliedIndex + 1; i <= commitInfo.Index; i++ {
				data, err := FSR.RN.Get(i)
				if err != nil {
					panic("[Error] Error from FS.RN.Get - " + err.Error())
				}

				msg, err := decode(data)
				if err != nil {
					panic("[Error] Decode error")
				}

				response := fs.ProcessMsg(msg)

				conn, ok := FSR.mapFind(msg.ClientId)
				if ok {
					if commitInfo.LeaderId != FSR.Id {
						panic("[Error] Got CommitInfo & Client found, but node is not leader now")
					}
					reply(conn, response)
				}
			}
			lastAppliedIndex = max(lastAppliedIndex, commitInfo.Index)
		}
	}
}
Exemple #2
0
func (FSR *FileServer) serve(clientId int, conn *net.TCPConn) {
	reader := bufio.NewReader(conn)
	for {
		msg, msgerr, fatalerr := fs.GetMsg(reader)
		if fatalerr != nil || msgerr != nil {
			reply(conn, &fs.Msg{Kind: 'M'})
			FSR.mapDel(clientId)
			conn.Close()
			break
		}

		if msgerr != nil {
			if (!reply(conn, &fs.Msg{Kind: 'M'})) {
				FSR.mapDel(clientId)
				conn.Close()
				break
			}
		}

		//fmt.Println("["+strconv.Itoa(FSR.Id)+"] - Msg Kind ", string(msg.Kind))
		if msg.Kind == 'r' {
			response := fs.ProcessMsg(msg)
			if !reply(conn, response) {
				FSR.mapDel(clientId)
				conn.Close()
				break
			}
		} else {
			msg.ClientId = clientId
			data, err := encode(msg)
			if err != nil {
				reply(conn, &fs.Msg{Kind: 'I'})
				FSR.mapDel(clientId)
				conn.Close()
				break
			}
			err = FSR.RN.Append(data)
			if err != nil {
				reply(conn, &fs.Msg{Kind: 'I'})
				FSR.mapDel(clientId)
				conn.Close()
				break
			}
		}
	}
}