コード例 #1
0
ファイル: command.go プロジェクト: zhangshun2014/rodis
// Handle command
func Handle(v resp.Array, ex *CommandExtras) error {
	ex.Buffer.Truncate(0) // Truncate all data in the buffer

	if len(v) == 0 {
		log6.Debug("Command handler, len of the input array is 0")
		return resp.NewError(ErrFmtNoCommand).WriteTo(ex.Buffer)
	}

	args := v.ToArgs()
	//log6.Debug("Command handling:%v", humanArgs(args))

	cmd := strings.ToLower(args[0].String())
	a, err := findCmdFunc(cmd)
	if err != nil {
		log6.Debug("Command handler, cannt found command: %v", cmd)
		return resp.NewError(ErrFmtUnknownCommand, cmd).WriteTo(ex.Buffer)
	}

	if a.c != 0 && len(v) != a.c { //a.c = 0 means to check the number in f
		return resp.NewError(ErrFmtWrongNumberArgument, cmd).WriteTo(ex.Buffer)
	}

	if !ex.IsConnAuthed && ex.Password != "" && cmd != "auth" {
		return resp.NewError(ErrAuthed).WriteTo(ex.Buffer)
	}

	return a.f(args[1:], ex)
}
コード例 #2
0
ファイル: connection.go プロジェクト: rod6/rodis
func (rc *rodisConn) close() {
	err := rc.conn.Close()
	if err != nil {
		log6.Debug("Connection %v close error: %v", rc.uuid, err)
	}

	rc.server.mu.Lock()
	delete(rc.server.conns, rc.uuid)
	rc.server.mu.Unlock()

	log6.Debug("Connection %v closed.", rc.uuid)
}
コード例 #3
0
ファイル: connection.go プロジェクト: rod6/rodis
func newConnection(conn net.Conn, rs *rodisServer) {
	uuid := uuid.New()
	rc := &rodisConn{uuid: uuid, db: storage.SelectStorage(0), conn: conn, reader: bufio.NewReader(conn), server: rs}

	if rs.cfg.RequirePass == "" {
		rc.authed = true
	}

	rc.extras = &command.CommandExtras{rc.db, &rc.buffer, rc.authed, rs.cfg.RequirePass}

	rc.server.mu.Lock()
	rs.conns[uuid] = rc
	rc.server.mu.Unlock()

	log6.Debug("New connection: %v", uuid)

	go rc.handle()
}
コード例 #4
0
ファイル: connection.go プロジェクト: rod6/rodis
func (rc *rodisConn) handle() {
	for {
		respType, respValue, err := resp.Parse(rc.reader)
		if err != nil {
			select {
			case <-rc.server.quit: // Server is quit, rc.close() is called.
				return
			default:
				break
			}

			if err == io.EOF { // Client close the connection
				log6.Debug("Client close connection %v.", rc.uuid)
				rc.close()
				return
			} else {
				log6.Warn("Connection %v error: %v", rc.uuid, err)
				continue // Other error, should continue the connection
			}
		}

		rc.response(respType, respValue)
	}
}