日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學(xué)無(wú)先后,達(dá)者為師

網(wǎng)站首頁(yè) 編程語(yǔ)言 正文

Golang原生rpc(rpc服務(wù)端源碼解讀)_Golang

作者:鹿灝楷silves ? 更新時(shí)間: 2022-06-08 編程語(yǔ)言

創(chuàng)建rpc接口,需要幾個(gè)條件

  • 方法的類(lèi)型是可輸出的
  • 方法的本身也是可輸出的
  • 方法必須有兩個(gè)參數(shù),必須是輸出類(lèi)型或者是內(nèi)建類(lèi)型
  • 方法的第二個(gè)參數(shù)是指針類(lèi)型
  • 方法返回的類(lèi)型為error

rpc服務(wù)原理分析

server端

  • 服務(wù)注冊(cè)
  • 處理網(wǎng)絡(luò)調(diào)用

服務(wù)注冊(cè) 通過(guò)反射處理,將接口存入到map中,進(jìn)行調(diào)用 注冊(cè)服務(wù)兩個(gè)方法

func Register (rcvr interface{}) error {}
func RegisterName (rcvr interface{} , name string) error {}
//指定注冊(cè)的名稱(chēng)

注冊(cè)方法的源代碼解讀 首先,無(wú)論是Register還是RegisterName底層代碼都是調(diào)用register方法,進(jìn)行服務(wù)注冊(cè)。 server.go register方法解讀

func (server *Server) register(rcvr interface{}, name string, useName bool) error {
	//創(chuàng)建一個(gè)service實(shí)例
	s := new(service)
	s.typ = reflect.TypeOf(rcvr)
	s.rcvr = reflect.ValueOf(rcvr)
	sname := reflect.Indirect(s.rcvr).Type().Name()
	//如果服務(wù)名為空,則使用默認(rèn)的服務(wù)名
	if useName {
		sname = name
	}
	if sname == "" {
		s := "rpc.Register: no service name for type " + s.typ.String()
		log.Print(s)
		return errors.New(s)
	}
	//判斷方法名是否暴漏的,如果方法名不是暴露的,則會(huì)導(dǎo)致調(diào)用不成功,所以返回false
	if !token.IsExported(sname) && !useName {
		s := "rpc.Register: type " + sname + " is not exported"
		log.Print(s)
		return errors.New(s)
	}
	s.name = sname

	// Install the methods
	//調(diào)用suitableMethods函數(shù),進(jìn)行返回接口,在suitableMethods中判斷方法是否符合作為rpc接口的條件,如果符合,則進(jìn)行添加到services中
	s.method = suitableMethods(s.typ, true)

	if len(s.method) == 0 {
		str := ""

		// To help the user, see if a pointer receiver would work.
		//如果方法綁定到結(jié)構(gòu)體的地址上,使用reflect.TypeOf()是不會(huì)發(fā)現(xiàn)方法的,所以也要進(jìn)行查找綁定到結(jié)構(gòu)體地址上的方法
		method := suitableMethods(reflect.PtrTo(s.typ), false)
		if len(method) != 0 {
			str = "rpc.Register: type " + sname + " has no exported methods of suitable type (hint: pass a pointer to value of that type)"
		} else {
			str = "rpc.Register: type " + sname + " has no exported methods of suitable type"
		}
		log.Print(str)
		return errors.New(str)
	}
	//判斷服務(wù)接口是否已經(jīng)注冊(cè)。
	if _, dup := server.serviceMap.LoadOrStore(sname, s); dup {
		return errors.New("rpc: service already defined: " + sname)
	}
	return nil
}

suitableMethod方法解讀

func suitableMethods(typ reflect.Type, reportErr bool) map[string]*methodType {
	//創(chuàng)建一個(gè)方法的切片
	methods := make(map[string]*methodType)
	for m := 0; m < typ.NumMethod(); m++ {
		method := typ.Method(m)
		mtype := method.Type
		mname := method.Name
		// Method must be exported.
		if method.PkgPath != "" {
			continue
		}
		// Method needs three ins: receiver, *args, *reply.
		//如果傳入的參數(shù),不為三個(gè),則會(huì)報(bào)錯(cuò),這里為什么是三個(gè)?
		//golang方法體中默認(rèn)傳入結(jié)構(gòu)體實(shí)例,所以request,*response,結(jié)構(gòu)體實(shí)例一共三個(gè)參數(shù)
		if mtype.NumIn() != 3 {
			if reportErr {
				log.Printf("rpc.Register: method %q has %d input parameters; needs exactly three\n", mname, mtype.NumIn())
			}
			continue
		}
		// First arg need not be a pointer.
		argType := mtype.In(1)
		if !isExportedOrBuiltinType(argType) {
			if reportErr {
				log.Printf("rpc.Register: argument type of method %q is not exported: %q\n", mname, argType)
			}
			continue
		}
		// Second arg must be a pointer.
		//判斷第二個(gè)參數(shù)是否為指針,如果不為指針,則返回false。
		replyType := mtype.In(2)
		if replyType.Kind() != reflect.Ptr {
			if reportErr {
				log.Printf("rpc.Register: reply type of method %q is not a pointer: %q\n", mname, replyType)
			}
			continue
		}
		// Reply type must be exported.
		if !isExportedOrBuiltinType(replyType) {
			if reportErr {
				log.Printf("rpc.Register: reply type of method %q is not exported: %q\n", mname, replyType)
			}
			continue
		}
		// Method needs one out.
		//返回結(jié)果是否為一個(gè)值,且為error
		if mtype.NumOut() != 1 {
			if reportErr {
				log.Printf("rpc.Register: method %q has %d output parameters; needs exactly one\n", mname, mtype.NumOut())
			}
			continue
		}
		// The return type of the method must be error.
		if returnType := mtype.Out(0); returnType != typeOfError {
			if reportErr {
				log.Printf("rpc.Register: return type of method %q is %q, must be error\n", mname, returnType)
			}
			continue
		}
		//將接口加入service
		methods[mname] = &methodType{method: method, ArgType: argType, ReplyType: replyType}
	}
	return methods
}

接收到請(qǐng)求后會(huì)不斷的解析請(qǐng)求 解析請(qǐng)求的兩個(gè)方法 readRequestHeader

func (server *Server) readRequestHeader(codec ServerCodec) (svc *service, mtype *methodType, req *Request, keepReading bool, err error) {
	// Grab the request header.
	//接收到請(qǐng)求,對(duì)請(qǐng)求進(jìn)行編碼
	req = server.getRequest()
	err = codec.ReadRequestHeader(req)
	if err != nil {
		req = nil
		if err == io.EOF || err == io.ErrUnexpectedEOF {
			return
		}
		err = errors.New("rpc: server cannot decode request: " + err.Error())
		return
	}

	// We read the header successfully. If we see an error now,
	// we can still recover and move on to the next request.
	keepReading = true
//編碼后的請(qǐng)求,進(jìn)行間隔,所以只要進(jìn)行將.的左右兩邊的數(shù)據(jù)進(jìn)行分割,就能解碼
	dot := strings.LastIndex(req.ServiceMethod, ".")
	if dot < 0 {
		err = errors.New("rpc: service/method request ill-formed: " + req.ServiceMethod)
		return
	}
	serviceName := req.ServiceMethod[:dot]
	methodName := req.ServiceMethod[dot+1:]

	// Look up the request.
	svci, ok := server.serviceMap.Load(serviceName)
	if !ok {
		err = errors.New("rpc: can't find service " + req.ServiceMethod)
		return
	}
	svc = svci.(*service)
	//獲取到注冊(cè)服務(wù)時(shí),注冊(cè)的接口
	mtype = svc.method[methodName]
	if mtype == nil {
		err = errors.New("rpc: can't find method " + req.ServiceMethod)
	}
	return
}

readRequest方法

func (server *Server) readRequest(codec ServerCodec) (service *service, mtype *methodType, req *Request, argv, replyv reflect.Value, keepReading bool, err error) {
	service, mtype, req, keepReading, err = server.readRequestHeader(codec)
//調(diào)用上面的readRequestHeader方法,進(jìn)行解碼,并返返回接口數(shù)據(jù)
	if err != nil {
		if !keepReading {
			return
		}
		// discard body
		codec.ReadRequestBody(nil)
		return
	}

	// Decode the argument value.
	argIsValue := false // if true, need to indirect before calling.
	//判斷傳擦是否為指針,如果為指針,需要使用Elem()方法,進(jìn)行指向結(jié)構(gòu)體
	if mtype.ArgType.Kind() == reflect.Ptr {
		argv = reflect.New(mtype.ArgType.Elem())
	} else {
		argv = reflect.New(mtype.ArgType)
		argIsValue = true
	}
	// argv guaranteed to be a pointer now.
	if err = codec.ReadRequestBody(argv.Interface()); err != nil {
		return
	}
	if argIsValue {
		argv = argv.Elem()
	}

	replyv = reflect.New(mtype.ReplyType.Elem())

	switch mtype.ReplyType.Elem().Kind() {
	case reflect.Map:
		replyv.Elem().Set(reflect.MakeMap(mtype.ReplyType.Elem()))
	case reflect.Slice:
		replyv.Elem().Set(reflect.MakeSlice(mtype.ReplyType.Elem(), 0, 0))
	}
	return
}

call方法

func (s *service) call(server *Server, sending *sync.Mutex, wg *sync.WaitGroup, mtype *methodType, req *Request, argv, replyv reflect.Value, codec ServerCodec) {
	if wg != nil {
		defer wg.Done()
	}
	mtype.Lock()
	mtype.numCalls++
	mtype.Unlock()
	function := mtype.method.Func
	// Invoke the method, providing a new value for the reply.
	//調(diào)用call方法,并將參數(shù)轉(zhuǎn)化為valueof型參數(shù),
	returnValues := function.Call([]reflect.Value{s.rcvr, argv, replyv})
	// The return value for the method is an error.
	//將返回的error進(jìn)行讀取,轉(zhuǎn)化為interface{}型
	errInter := returnValues[0].Interface()
	errmsg := ""
	if errInter != nil {
	//將error進(jìn)行斷言
		errmsg = errInter.(error).Error()
	}
	server.sendResponse(sending, req, replyv.Interface(), codec, errmsg)
	server.freeRequest(req)
}

注冊(cè)的大概流程

  • 根據(jù)反射,進(jìn)行接口的獲取
  • 使用方法判斷接口是否符合作為rpc接口的規(guī)范(有兩個(gè)參數(shù),第二個(gè)參數(shù)為指針,返回一個(gè)參數(shù)error)
  • 如果不符合規(guī)范,將返回error,符合規(guī)范,將存入map,進(jìn)行提供調(diào)用

接收請(qǐng)求的大概流程

  • 首先,不斷的接收數(shù)據(jù)流,并進(jìn)行解碼,解碼之后為data.data,所以我們需要使用 . 作為分隔符,進(jìn)行數(shù)據(jù)的截切和讀取
  • 將讀取的數(shù)據(jù)在注冊(cè)的map中進(jìn)行查找,如果查找到,返回相關(guān)的service和其他數(shù)據(jù)
  • 進(jìn)行調(diào)用

原文鏈接:https://juejin.cn/post/7083083938603728903

欄目分類(lèi)
最近更新