コード例 #1
0
ファイル: server.go プロジェクト: hustxiaoc/tchannel
// Set sets the value for a given key.
func (h *kvHandler) Set(ctx thrift.Context, key, value string) error {
	if err := isValidKey(key); err != nil {
		return err
	}

	h.mut.Lock()
	defer h.mut.Unlock()

	h.vals[key] = value
	// Example of how to use response headers. Normally, these values should be passed via result structs.
	ctx.SetResponseHeaders(map[string]string{"count": fmt.Sprint(len(h.vals))})
	return nil
}
コード例 #2
0
ファイル: server.go プロジェクト: hustxiaoc/tchannel
func isAdmin(ctx thrift.Context) bool {
	return ctx.Headers()["user"] == "admin"
}