Procházet zdrojové kódy

feat: add token api

2637309949 před 4 roky
rodič
revize
9b4e98a701

+ 4 - 1
etc/transform.yaml

@@ -8,4 +8,7 @@ DataSource: root:gSRGZqb121TlYIbJy0@tcp(47.103.202.94:3306)/i2erp_erp1?charset=u
 Table:
 Table:
 Cache:
 Cache:
   - Host: 47.103.219.158:7001
   - Host: 47.103.219.158:7001
-    Pass: i2#333
+    Pass: i2#333
+Erp:
+  AutoRefresh: true
+  AuthServer: i2-dauth:10808

+ 4 - 0
internal/config/config.go

@@ -10,4 +10,8 @@ type Config struct {
 	DataSource string
 	DataSource string
 	Table      string
 	Table      string
 	Cache      cache.CacheConf
 	Cache      cache.CacheConf
+	Erp        struct {
+		AutoRefresh bool
+		AuthServer  string
+	}
 }
 }

+ 30 - 0
internal/logic/parsetokenlogic.go

@@ -0,0 +1,30 @@
+package logic
+
+import (
+	"context"
+
+	"git.i2edu.net/i2/i2-bill-erp/internal/svc"
+	"git.i2edu.net/i2/i2-bill-erp/internal/utils"
+	"git.i2edu.net/i2/i2-bill-erp/transform"
+
+	"git.i2edu.net/i2/go-zero/core/logx"
+)
+
+type ParseTokenLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewParseTokenLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ParseTokenLogic {
+	return &ParseTokenLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *ParseTokenLogic) ParseToken(in *transform.TokenRequest) (*transform.TokenResponse, error) {
+	erpToken := utils.GetGlobalTokenStore().Get(in.Token)
+	return &transform.TokenResponse{UserId: erpToken.UserId}, nil
+}

+ 5 - 0
internal/server/transformserver.go

@@ -25,3 +25,8 @@ func (s *TransformServer) GetUser(ctx context.Context, in *transform.UserRequest
 	l := logic.NewGetUserLogic(ctx, s.svcCtx)
 	l := logic.NewGetUserLogic(ctx, s.svcCtx)
 	return l.GetUser(in)
 	return l.GetUser(in)
 }
 }
+
+func (s *TransformServer) ParseToken(ctx context.Context, in *transform.TokenRequest) (*transform.TokenResponse, error) {
+	l := logic.NewParseTokenLogic(ctx, s.svcCtx)
+	return l.ParseToken(in)
+}

+ 634 - 0
internal/utils/token.go

@@ -0,0 +1,634 @@
+// ERP 代码, light的逻辑, 废弃!!!!!
+// ERP 代码, light的逻辑, 废弃!!!!!
+
+package utils
+
+import (
+	"bytes"
+	"crypto/md5"
+	"encoding/binary"
+	"encoding/hex"
+	"errors"
+	"fmt"
+	"log"
+	"net"
+	"strconv"
+	"strings"
+	"sync"
+	"time"
+
+	"git.i2edu.net/i2/go-zero/core/logx"
+	"git.i2edu.net/i2/i2-bill-erp/internal/config"
+)
+
+type Token struct {
+	Lock         *sync.RWMutex
+	Result       int                    `json:"-"`
+	UserId       string                 `json:"user_id"`
+	AccessToken  string                 `json:"access_token"`
+	RefreshToken string                 `json:"refresh_token"`
+	LoginID      string                 `json:"login_id"`
+	TimeStamp    uint64                 `json:"time_stamp"`
+	ServerIp     string                 `json:"server_ip"`
+	Domain       string                 `json:"domain"`
+	Password     string                 `json:"-"`
+	Map          map[string]interface{} `json:"map"`
+}
+
+type TokenStore struct {
+	name   string
+	lock   *sync.RWMutex
+	tokens map[string]*Token
+}
+
+type IAuth interface {
+	Init()
+}
+
+var globalTokenStore *TokenStore = nil
+
+func init() {
+	iauthMap = make(map[string]IAuth)
+	globalTokenStore = &TokenStore{name: "sso", lock: new(sync.RWMutex), tokens: make(map[string]*Token)}
+	go globalTokenStore.startTokenCheckProcess()
+
+	lightAuth := &LightAuth{}
+	RegisterAuth("qianqiusoft.com", lightAuth)
+}
+
+type LightAuth struct {
+}
+
+func (la *LightAuth) Init() {
+
+}
+
+var iauthMap map[string]IAuth
+
+/**
+ * @brief: 注册自定登录对象, 会调用Init进行初始化
+ * @param1 t: key
+ * @param2 a: 认证对象
+ * @return none
+ */
+func RegisterAuth(t string, a IAuth) {
+	if t == "" || a == nil {
+		return
+	}
+
+	var ok bool
+	if _, ok = iauthMap[t]; ok {
+		return
+	} else {
+		a.Init()
+		iauthMap[t] = a
+	}
+}
+
+func GetGlobalTokenStore() *TokenStore {
+	return globalTokenStore
+}
+
+func (t *TokenStore) Get(key string) *Token {
+	t.lock.RLock()
+	defer t.lock.RUnlock()
+	if val, ok := t.tokens[key]; ok {
+		//log.Println(key, "获取Token:", val.AccessToken, val.RefreshToken, val.LoginID)
+		return val
+	}
+	return nil
+}
+
+func (t *TokenStore) Set(key string, v *Token) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+	if val, ok := t.tokens[key]; !ok {
+		t.tokens[key] = v
+	} else if val != v {
+		t.tokens[key] = v
+	}
+}
+
+func (t *TokenStore) Remove(key string) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+	delete(t.tokens, key)
+	log.Println(key, "删除Key")
+}
+
+func (t *TokenStore) Refresh(key string) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	val, ok := t.tokens[key]
+	if ok {
+		val.TimeStamp = uint64(time.Now().UnixNano())
+	}
+}
+
+func (t *TokenStore) startTokenCheckProcess() {
+}
+
+func Validate(accessToken, loginId string, domain string) (*Token, error) {
+	token := globalTokenStore.Get(loginId + domain)
+	if token != nil {
+		if strings.EqualFold(token.AccessToken, accessToken) {
+			logx.Info("get the token ", accessToken, " of id ", loginId+domain)
+			globalTokenStore.Refresh(loginId + domain)
+			return token, nil
+		} else {
+			logx.Error(token.AccessToken, "is not equal to", accessToken)
+			return token, errors.New(token.AccessToken + " is not equal to " + accessToken)
+		}
+	} else {
+		logx.Error("can not get the token of", loginId+domain)
+		return token, errors.New("can not get the token of " + loginId + domain)
+	}
+
+}
+
+func TokenValidate(token string) (*Token, error) {
+	user := globalTokenStore.Get(token)
+
+	if strings.EqualFold(user.AccessToken, token) {
+		logx.Info("get the token ", token, " of id ")
+		globalTokenStore.Refresh(token)
+		return user, nil
+	} else {
+		logx.Error(user.AccessToken, "is not equal to", token)
+		return user, errors.New(user.AccessToken + " is not equal to " + token)
+	}
+
+}
+
+const (
+	__KEY = "Light#dauth-@*I2"
+
+	CMD_NEW           = "new"
+	CMD_REMOVE        = "remove"
+	CMD_PINGPONG      = "pg"
+	CMD_PINGPONG_RESP = "pg_resp"
+)
+
+var TCPClient *TcpClient
+
+type authPackage struct {
+	Cmd      string
+	TokenStr string
+	Content  []byte
+}
+
+func (ap *authPackage) toBytes() []byte {
+	buf := bytes.NewBuffer([]byte{})
+
+	b := []byte(ap.Cmd)
+	buf.Write(uint32ToBytes(len(b)))
+	buf.Write(b)
+
+	buf.Write(uint32ToBytes(len(ap.Content)))
+	buf.Write(ap.Content)
+
+	return buf.Bytes()
+}
+
+type TcpClient struct {
+	conn     net.Conn // 连接
+	c        *config.Config
+	pchan    chan *authPackage // 包chan
+	done     chan bool         // 是否done
+	exited   bool              // 退出
+	verified bool              // 验证是否
+}
+
+// 创建client
+func NewTcpClient(cfg *config.Config) *TcpClient {
+	c := &TcpClient{
+		c:        cfg,
+		pchan:    make(chan *authPackage, 100),
+		done:     make(chan bool),
+		exited:   false,
+		verified: false,
+	}
+	return c
+}
+
+// 启动
+func (c *TcpClient) Start() {
+	go func() {
+		defer func() {
+			if p := recover(); p != nil {
+				fmt.Println("ecover", p)
+			}
+			if c.conn != nil {
+				c.conn.Close()
+			}
+			c.restart()
+		}()
+
+		var err error = nil
+		address := c.c.Erp.AuthServer
+		//fmt.Println("auth client start, dial address is", address)
+		c.conn, err = net.Dial("tcp", address)
+		if err != nil {
+			fmt.Println("Error dialing erp client", err.Error())
+			return
+		}
+
+		fmt.Println("发送验证")
+		sendVerify(c.conn) // 发送验证,不需要读取返回值,如果验证错误立刻关掉
+
+		fmt.Println("读取验证结果")
+		vresp, err := readString(c.conn)
+		if err != nil {
+			fmt.Println("Error dialing", err.Error())
+			return
+		}
+		if vresp != "ok" {
+			// 验证失败
+			fmt.Println("verify is not ok", vresp)
+			return
+		}
+		fmt.Println("验证成功")
+		c.verified = true
+
+		// send
+		go func() {
+			for {
+				select {
+				case data := <-c.pchan:
+					logx.Info("写入数据")
+					c.conn.SetWriteDeadline(time.Now().Add(time.Second * 2))
+					_, err := c.conn.Write(data.toBytes())
+					if err != nil {
+						fmt.Println("写入内容错误", err.Error())
+						return
+					}
+				case <-c.done:
+					logx.Info("发送数据done退出")
+					return
+				}
+			}
+		}()
+
+		// receive
+		for {
+			cmd, err := readString(c.conn) // 读取命令
+			if err != nil {
+				c.done <- true
+				fmt.Println("读取命令错误", err.Error())
+				break
+			}
+			if cmd == CMD_NEW {
+				err = c.newHandler()
+			} else if cmd == CMD_REMOVE {
+				err = c.removeHandler()
+			} else if cmd == CMD_PINGPONG_RESP {
+
+			} else {
+				fmt.Println("未知cmd", cmd)
+				continue
+			}
+			if err != nil {
+				c.done <- true
+				fmt.Println("处理错误", err.Error())
+				break
+			}
+		}
+	}()
+}
+
+// 停止
+func (c *TcpClient) Stop() {
+	c.exited = true
+	c.conn.Close()
+}
+
+// 检测
+func (c *TcpClient) restart() {
+	if c.exited {
+		// 已退出则不重启
+		return
+	}
+
+	go func() {
+		c.verified = false
+		c.done = make(chan bool)
+		c.pchan = make(chan *authPackage, 100)
+		c.exited = false
+		time.Sleep(3 * time.Second)
+		c.Start()
+	}()
+}
+
+// 发送bytes
+func (c *TcpClient) Send(cmd string, bytess []byte) {
+	if !c.verified {
+		fmt.Println("未认证")
+		return
+	}
+
+	logx.Info("发送指令1", cmd)
+	c.pchan <- &authPackage{
+		Cmd:     cmd,
+		Content: bytess,
+	}
+	logx.Info("发送指令2", cmd)
+}
+
+// 发送token
+func (c *TcpClient) SendToken(token *Token) {
+	// logx.Info("发送新建token")
+	bytess := tokenToBytes(token)
+	c.Send(CMD_NEW, bytess)
+}
+
+// 处理创建
+func (c *TcpClient) newHandler() error {
+	// fmt.Println("处理新建")
+	bytess, err := readBytes(c.conn)
+	if err != nil {
+		fmt.Println("读取token内容错误", err.Error())
+		return err
+	}
+	// 新建
+	token, err := bytesToToken(bytess)
+	if err != nil {
+		logx.Info("bytesToToken 错误", err.Error())
+		return err
+	}
+	GetGlobalTokenStore().Set(token.AccessToken, token)
+	return nil
+}
+
+// 处理删除
+func (c *TcpClient) removeHandler() error {
+	fmt.Println("处理删除")
+	bytess, err := readBytes(c.conn)
+	if err != nil {
+		fmt.Println("读取token内容错误", err.Error())
+		return err
+	}
+	// 移除,此时bytess为tokenstring
+	GetGlobalTokenStore().Remove(string(bytess))
+	return nil
+}
+
+// 读取字符串
+func readString(conn net.Conn) (string, error) {
+	// 读长度
+	size, err := readUInt32(conn)
+	if err != nil {
+		fmt.Println("读取长度失败,", err.Error())
+		return "", err
+	}
+
+	// 读字符串
+	b := make([]byte, size)
+	n, err := conn.Read(b)
+	if n != int(size) {
+		return "", errors.New("读取长度不是" + strconv.Itoa(int(size)))
+	}
+	return string(b), nil
+}
+
+// 写入字符串
+func writeString(conn net.Conn, str string) error {
+	if str == "" {
+		return errors.New("字符串为空")
+	}
+
+	bytess := []byte(str)
+	size := len(bytess)
+
+	// 发送长度
+	err := writeUInt32(conn, uint32(size))
+	if err != nil {
+		fmt.Println("发送长度失败,", err.Error())
+		return err
+	}
+
+	// 发送内容
+	n, err := conn.Write(bytess)
+	if err != nil {
+		fmt.Println("发送内容失败,", err.Error())
+		return err
+	}
+	if n != size {
+		return errors.New("发送长度不是" + strconv.Itoa(int(size)))
+	}
+
+	return nil
+}
+
+// 读取bytes
+func readBytes(conn net.Conn) ([]byte, error) {
+	// 读长度
+	size, err := readUInt32(conn)
+	if err != nil {
+		fmt.Println("读取长度失败,", err.Error())
+		return nil, err
+	}
+
+	// 读字符串
+	b := make([]byte, size)
+	n, err := conn.Read(b)
+	if n != int(size) {
+		return nil, errors.New("读取长度不是" + strconv.Itoa(int(size)))
+	}
+	return b, nil
+}
+
+// 读取uint64
+func readUInt32(conn net.Conn) (uint32, error) {
+	b := make([]byte, 4)
+	n, err := conn.Read(b)
+	if err != nil {
+		fmt.Println("读取长度失败,", err.Error())
+		return 0, err
+	}
+	if n != 4 {
+		return 0, errors.New("读取长度不是4")
+	}
+	size := binary.BigEndian.Uint32(b)
+
+	return size, nil
+}
+
+// 写入长度
+func writeUInt32(conn net.Conn, v uint32) error {
+	// 发送长度
+	b := make([]byte, 4)
+	binary.BigEndian.PutUint32(b, v)
+	n, err := conn.Write(b)
+	if err != nil {
+		fmt.Println("发送长度失败,", err.Error())
+		return err
+	}
+	if n != 4 {
+		return errors.New("发送长度不是4")
+	}
+
+	return nil
+}
+
+// 写入长度
+func writeUInt64(conn net.Conn, v uint64) error {
+	// 发送长度
+	b := make([]byte, 8)
+	binary.BigEndian.PutUint64(b, v)
+	n, err := conn.Write(b)
+	if err != nil {
+		fmt.Println("发送长度失败,", err.Error())
+		return err
+	}
+	if n != 4 {
+		return errors.New("发送长度不是4")
+	}
+
+	return nil
+}
+
+// 读取uint64
+func readStringByBytes(bytess []byte) (string, int, error) {
+	size := binary.BigEndian.Uint32(bytess)
+	return string(bytess[4 : 4+size]), int(size), nil
+}
+
+// int转bytes
+func uint32ToBytes(v int) []byte {
+	b := make([]byte, 4)
+	binary.BigEndian.PutUint32(b, uint32(v))
+
+	return b
+}
+
+// int转bytes
+func uint64ToBytes(v int) []byte {
+	b := make([]byte, 8)
+	binary.BigEndian.PutUint32(b, uint32(v))
+
+	return b
+}
+
+// 转token
+func bytesToToken(content []byte) (*Token, error) {
+	token := &Token{Lock: new(sync.RWMutex)}
+	var index int = 0
+	var size int
+	var err error = nil
+	// fmt.Println("读取userid")
+	token.UserId, size, err = readStringByBytes(content)
+	if err != nil {
+		fmt.Println("读取userid错误")
+		return nil, err
+	}
+	index += 4 + size
+
+	// fmt.Println("读取AccessToken")
+	token.AccessToken, size, err = readStringByBytes(content[index:])
+	if err != nil {
+		fmt.Println("读取AccessToken错误")
+		return nil, err
+	}
+	index += 4 + size
+
+	// fmt.Println("读取RefreshToken")
+	token.RefreshToken, size, err = readStringByBytes(content[index:])
+	if err != nil {
+		fmt.Println("读取RefreshToken错误")
+		return nil, err
+	}
+	index += 4 + size
+
+	// fmt.Println("读取LoginID")
+	token.LoginID, size, err = readStringByBytes(content[index:])
+	if err != nil {
+		fmt.Println("读取LoginID错误")
+		return nil, err
+	}
+	index += 4 + size
+
+	// fmt.Println("读取timestamp")
+	token.TimeStamp = binary.BigEndian.Uint64(content[index:])
+	index += 8
+
+	// fmt.Println("读取ServerIp")
+	token.ServerIp, size, err = readStringByBytes(content[index:])
+	if err != nil {
+		fmt.Println("读取ServerIp错误")
+		return nil, err
+	}
+	index += 4 + size
+
+	// fmt.Println("读取Domain")
+	token.Domain, size, err = readStringByBytes(content[index:])
+	if err != nil {
+		fmt.Println("读取Domain错误")
+		return nil, err
+	}
+	index += 4 + size
+
+	return token, nil
+}
+
+// 转bytes,包括token开头
+func tokenToBytes(token *Token) []byte {
+	buf := bytes.NewBuffer([]byte{})
+
+	t := []byte(token.UserId)
+	buf.Write(uint32ToBytes(len(t)))
+	buf.Write(t)
+
+	t = []byte(token.AccessToken)
+	buf.Write(uint32ToBytes(len(t)))
+	buf.Write(t)
+
+	t = []byte(token.RefreshToken)
+	buf.Write(uint32ToBytes(len(t)))
+	buf.Write(t)
+
+	t = []byte(token.LoginID)
+	buf.Write(uint32ToBytes(len(t)))
+	buf.Write(t)
+
+	buf.Write(uint64ToBytes(int(token.TimeStamp)))
+
+	fmt.Println(token.ServerIp)
+	t = []byte(token.ServerIp)
+	buf.Write(uint32ToBytes(len(t)))
+	buf.Write(t)
+
+	fmt.Println(token.Domain)
+	t = []byte(token.Domain)
+	buf.Write(uint32ToBytes(len(t)))
+	buf.Write(t)
+
+	bytess := buf.Bytes()
+	buf = bytes.NewBuffer([]byte{}) // 这里用reset是错误的
+
+	tokenstrbytes := []byte(token.AccessToken)
+	buf.Write(uint32ToBytes(len(tokenstrbytes)))
+	buf.Write(tokenstrbytes)
+	buf.Write(uint32ToBytes(len(bytess)))
+	buf.Write(bytess)
+
+	return buf.Bytes()
+}
+
+// 发送验证
+func sendVerify(conn net.Conn) {
+	timestamp := time.Now().UnixNano()
+	timestampStr := strconv.Itoa(int(timestamp))
+	seed := timestampStr + __KEY
+	hashVal := hash(seed)
+
+	writeUInt64(conn, uint64(timestamp))
+	writeString(conn, hashVal)
+}
+
+// md5 哈希
+func hash(str string) string {
+	h := md5.New()
+	h.Write([]byte(str))
+	return hex.EncodeToString(h.Sum(nil))
+}

+ 6 - 0
transform.go

@@ -7,6 +7,7 @@ import (
 	"git.i2edu.net/i2/i2-bill-erp/internal/config"
 	"git.i2edu.net/i2/i2-bill-erp/internal/config"
 	"git.i2edu.net/i2/i2-bill-erp/internal/server"
 	"git.i2edu.net/i2/i2-bill-erp/internal/server"
 	"git.i2edu.net/i2/i2-bill-erp/internal/svc"
 	"git.i2edu.net/i2/i2-bill-erp/internal/svc"
+	"git.i2edu.net/i2/i2-bill-erp/internal/utils"
 	"git.i2edu.net/i2/i2-bill-erp/transform"
 	"git.i2edu.net/i2/i2-bill-erp/transform"
 
 
 	"git.i2edu.net/i2/go-zero/core/conf"
 	"git.i2edu.net/i2/go-zero/core/conf"
@@ -21,6 +22,11 @@ func main() {
 
 
 	var c config.Config
 	var c config.Config
 	conf.MustLoad(*configFile, &c)
 	conf.MustLoad(*configFile, &c)
+
+	// erp client
+	erpClient := utils.NewTcpClient(&c)
+	erpClient.Start()
+
 	ctx := svc.NewServiceContext(c)
 	ctx := svc.NewServiceContext(c)
 	srv := server.NewTransformServer(ctx)
 	srv := server.NewTransformServer(ctx)
 
 

+ 9 - 0
transform.proto

@@ -18,6 +18,15 @@ message UserResponse {
   int64 gender          = 8;
   int64 gender          = 8;
 }
 }
 
 
+message TokenRequest {
+  string token = 1;
+}
+
+message TokenResponse {
+  string UserId = 1;
+}
+
 service Transform {
 service Transform {
   rpc GetUser(UserRequest) returns(UserResponse);
   rpc GetUser(UserRequest) returns(UserResponse);
+  rpc ParseToken(TokenRequest) returns(TokenResponse);
 }
 }

+ 135 - 16
transform/transform.pb.go

@@ -166,30 +166,113 @@ func (m *UserResponse) GetGender() int64 {
 	return 0
 	return 0
 }
 }
 
 
+type TokenRequest struct {
+	Token                string   `protobuf:"bytes,1,opt,name=token,proto3" json:"token,omitempty"`
+	XXX_NoUnkeyedLiteral struct{} `json:"-"`
+	XXX_unrecognized     []byte   `json:"-"`
+	XXX_sizecache        int32    `json:"-"`
+}
+
+func (m *TokenRequest) Reset()         { *m = TokenRequest{} }
+func (m *TokenRequest) String() string { return proto.CompactTextString(m) }
+func (*TokenRequest) ProtoMessage()    {}
+func (*TokenRequest) Descriptor() ([]byte, []int) {
+	return fileDescriptor_cb4a498eeb2ba07d, []int{2}
+}
+
+func (m *TokenRequest) XXX_Unmarshal(b []byte) error {
+	return xxx_messageInfo_TokenRequest.Unmarshal(m, b)
+}
+func (m *TokenRequest) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
+	return xxx_messageInfo_TokenRequest.Marshal(b, m, deterministic)
+}
+func (m *TokenRequest) XXX_Merge(src proto.Message) {
+	xxx_messageInfo_TokenRequest.Merge(m, src)
+}
+func (m *TokenRequest) XXX_Size() int {
+	return xxx_messageInfo_TokenRequest.Size(m)
+}
+func (m *TokenRequest) XXX_DiscardUnknown() {
+	xxx_messageInfo_TokenRequest.DiscardUnknown(m)
+}
+
+var xxx_messageInfo_TokenRequest proto.InternalMessageInfo
+
+func (m *TokenRequest) GetToken() string {
+	if m != nil {
+		return m.Token
+	}
+	return ""
+}
+
+type TokenResponse struct {
+	UserId               string   `protobuf:"bytes,1,opt,name=UserId,proto3" json:"UserId,omitempty"`
+	XXX_NoUnkeyedLiteral struct{} `json:"-"`
+	XXX_unrecognized     []byte   `json:"-"`
+	XXX_sizecache        int32    `json:"-"`
+}
+
+func (m *TokenResponse) Reset()         { *m = TokenResponse{} }
+func (m *TokenResponse) String() string { return proto.CompactTextString(m) }
+func (*TokenResponse) ProtoMessage()    {}
+func (*TokenResponse) Descriptor() ([]byte, []int) {
+	return fileDescriptor_cb4a498eeb2ba07d, []int{3}
+}
+
+func (m *TokenResponse) XXX_Unmarshal(b []byte) error {
+	return xxx_messageInfo_TokenResponse.Unmarshal(m, b)
+}
+func (m *TokenResponse) XXX_Marshal(b []byte, deterministic bool) ([]byte, error) {
+	return xxx_messageInfo_TokenResponse.Marshal(b, m, deterministic)
+}
+func (m *TokenResponse) XXX_Merge(src proto.Message) {
+	xxx_messageInfo_TokenResponse.Merge(m, src)
+}
+func (m *TokenResponse) XXX_Size() int {
+	return xxx_messageInfo_TokenResponse.Size(m)
+}
+func (m *TokenResponse) XXX_DiscardUnknown() {
+	xxx_messageInfo_TokenResponse.DiscardUnknown(m)
+}
+
+var xxx_messageInfo_TokenResponse proto.InternalMessageInfo
+
+func (m *TokenResponse) GetUserId() string {
+	if m != nil {
+		return m.UserId
+	}
+	return ""
+}
+
 func init() {
 func init() {
 	proto.RegisterType((*UserRequest)(nil), "transform.UserRequest")
 	proto.RegisterType((*UserRequest)(nil), "transform.UserRequest")
 	proto.RegisterType((*UserResponse)(nil), "transform.UserResponse")
 	proto.RegisterType((*UserResponse)(nil), "transform.UserResponse")
+	proto.RegisterType((*TokenRequest)(nil), "transform.TokenRequest")
+	proto.RegisterType((*TokenResponse)(nil), "transform.TokenResponse")
 }
 }
 
 
 func init() { proto.RegisterFile("transform.proto", fileDescriptor_cb4a498eeb2ba07d) }
 func init() { proto.RegisterFile("transform.proto", fileDescriptor_cb4a498eeb2ba07d) }
 
 
 var fileDescriptor_cb4a498eeb2ba07d = []byte{
 var fileDescriptor_cb4a498eeb2ba07d = []byte{
-	// 232 bytes of a gzipped FileDescriptorProto
-	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x90, 0xb1, 0x4e, 0xc3, 0x30,
-	0x10, 0x86, 0xe5, 0x94, 0xa6, 0xcd, 0x15, 0x81, 0x74, 0x42, 0xc5, 0xea, 0x14, 0x75, 0xea, 0xd4,
-	0x81, 0x6e, 0xbc, 0x40, 0x37, 0x06, 0xab, 0x3c, 0x40, 0x20, 0x07, 0x44, 0x6a, 0x6d, 0x73, 0xe7,
-	0x0c, 0x7d, 0x4f, 0x1e, 0x08, 0xd9, 0xa6, 0x21, 0x12, 0x0b, 0xe3, 0xf7, 0xfd, 0x3a, 0xff, 0xf2,
-	0x0f, 0xb7, 0x81, 0x1b, 0x2b, 0x6f, 0x8e, 0x4f, 0x5b, 0xcf, 0x2e, 0x38, 0xac, 0x06, 0xb1, 0xde,
-	0xc1, 0xe2, 0x59, 0x88, 0x0d, 0x7d, 0xf6, 0x24, 0x01, 0x6f, 0xa0, 0xe8, 0x5a, 0xad, 0x6a, 0xb5,
-	0xa9, 0x4c, 0xd1, 0xb5, 0x78, 0x07, 0x53, 0xff, 0xe1, 0x2c, 0xe9, 0x22, 0xa9, 0x0c, 0xeb, 0x2f,
-	0x05, 0xd7, 0xf9, 0x4a, 0xbc, 0xb3, 0x42, 0xff, 0x3b, 0x8b, 0x96, 0xdd, 0x91, 0x44, 0x4f, 0xb2,
-	0x4d, 0x80, 0x4b, 0x28, 0x25, 0x34, 0xa1, 0x17, 0x7d, 0x55, 0xab, 0xcd, 0xc4, 0xfc, 0x10, 0xd6,
-	0xb0, 0x20, 0xf6, 0xc6, 0x1d, 0xe9, 0x70, 0xf6, 0xa4, 0xa7, 0x29, 0x1c, 0x2b, 0x5c, 0xc1, 0xbc,
-	0x17, 0xe2, 0xa7, 0xe6, 0x44, 0xba, 0x4c, 0x4f, 0x0e, 0x1c, 0xb3, 0xd7, 0x2e, 0x9c, 0x53, 0x36,
-	0xcb, 0xd9, 0x85, 0x63, 0xe3, 0x3b, 0xd9, 0x96, 0x58, 0xcf, 0x73, 0x63, 0xa6, 0x87, 0x3d, 0x54,
-	0x87, 0xcb, 0x30, 0xf8, 0x08, 0xb3, 0x3d, 0x85, 0xf8, 0x4b, 0x5c, 0x6e, 0x7f, 0x07, 0x1c, 0x8d,
-	0xb5, 0xba, 0xff, 0xe3, 0xf3, 0x1c, 0x2f, 0x65, 0x9a, 0x79, 0xf7, 0x1d, 0x00, 0x00, 0xff, 0xff,
-	0x03, 0x6c, 0x0e, 0x00, 0x79, 0x01, 0x00, 0x00,
+	// 287 bytes of a gzipped FileDescriptorProto
+	0x1f, 0x8b, 0x08, 0x00, 0x00, 0x00, 0x00, 0x00, 0x02, 0xff, 0x8c, 0x91, 0xbd, 0x4e, 0xc3, 0x30,
+	0x14, 0x85, 0x95, 0x94, 0xa4, 0xcd, 0x6d, 0x01, 0xc9, 0xaa, 0x82, 0x95, 0x29, 0x8a, 0x90, 0xe8,
+	0xd4, 0x81, 0x6e, 0x2c, 0x8c, 0x88, 0x05, 0xa1, 0x28, 0x3c, 0x40, 0x20, 0x17, 0x88, 0x68, 0x63,
+	0xe3, 0xeb, 0x0c, 0x7d, 0x03, 0x1e, 0x90, 0x07, 0x42, 0xfe, 0x49, 0x89, 0xd4, 0xa5, 0xe3, 0x77,
+	0x8e, 0xed, 0xe3, 0x7b, 0x2e, 0x5c, 0x6a, 0x55, 0x77, 0xf4, 0x2e, 0xd4, 0x6e, 0x2d, 0x95, 0xd0,
+	0x82, 0x25, 0x07, 0xa1, 0xd8, 0xc0, 0xfc, 0x85, 0x50, 0x95, 0xf8, 0xdd, 0x23, 0x69, 0x76, 0x01,
+	0x61, 0xdb, 0xf0, 0x20, 0x0f, 0x56, 0x49, 0x19, 0xb6, 0x0d, 0x5b, 0x42, 0x24, 0x3f, 0x45, 0x87,
+	0x3c, 0xb4, 0x92, 0x83, 0xe2, 0x37, 0x80, 0x85, 0xbb, 0x45, 0x52, 0x74, 0x84, 0xa7, 0x5d, 0x33,
+	0xaa, 0x12, 0x5b, 0x24, 0x3e, 0x71, 0xaa, 0x05, 0x96, 0x42, 0x4c, 0xba, 0xd6, 0x3d, 0xf1, 0xb3,
+	0x3c, 0x58, 0x4d, 0x4a, 0x4f, 0x2c, 0x87, 0x39, 0x2a, 0x59, 0x8a, 0x2d, 0x56, 0x7b, 0x89, 0x3c,
+	0xb2, 0xe6, 0x58, 0x62, 0x19, 0xcc, 0x7a, 0x42, 0xf5, 0x54, 0xef, 0x90, 0xc7, 0xf6, 0xc9, 0x03,
+	0x1b, 0xef, 0xad, 0xd5, 0x7b, 0xeb, 0x4d, 0x9d, 0x37, 0xb0, 0x49, 0xfc, 0xc0, 0xae, 0x41, 0xc5,
+	0x67, 0x2e, 0xd1, 0x51, 0x71, 0x0d, 0x8b, 0x4a, 0x7c, 0x61, 0x37, 0x94, 0xb1, 0x84, 0x48, 0x1b,
+	0xf6, 0x83, 0x39, 0x28, 0x6e, 0xe0, 0xdc, 0x9f, 0xf2, 0xc3, 0xa7, 0x10, 0x9b, 0x32, 0x1e, 0x87,
+	0x02, 0x3c, 0xdd, 0xfe, 0x04, 0x90, 0x54, 0x43, 0xd1, 0xec, 0x0e, 0xa6, 0x0f, 0xa8, 0x8d, 0xc5,
+	0xd2, 0xf5, 0xff, 0x42, 0x46, 0xe5, 0x67, 0x57, 0x47, 0xba, 0x4f, 0xb8, 0x07, 0x78, 0xae, 0x15,
+	0xa1, 0xcd, 0x65, 0xe3, 0x63, 0xe3, 0xff, 0x66, 0xfc, 0xd8, 0x70, 0x0f, 0xbc, 0xc6, 0x76, 0xef,
+	0x9b, 0xbf, 0x00, 0x00, 0x00, 0xff, 0xff, 0x6f, 0x8d, 0xf1, 0x12, 0x0a, 0x02, 0x00, 0x00,
 }
 }
 
 
 // Reference imports to suppress errors if they are not otherwise used.
 // Reference imports to suppress errors if they are not otherwise used.
@@ -205,6 +288,7 @@ const _ = grpc.SupportPackageIsVersion4
 // For semantics around ctx use and closing/ending streaming RPCs, please refer to https://godoc.org/google.golang.org/grpc#ClientConn.NewStream.
 // For semantics around ctx use and closing/ending streaming RPCs, please refer to https://godoc.org/google.golang.org/grpc#ClientConn.NewStream.
 type TransformClient interface {
 type TransformClient interface {
 	GetUser(ctx context.Context, in *UserRequest, opts ...grpc.CallOption) (*UserResponse, error)
 	GetUser(ctx context.Context, in *UserRequest, opts ...grpc.CallOption) (*UserResponse, error)
+	ParseToken(ctx context.Context, in *TokenRequest, opts ...grpc.CallOption) (*TokenResponse, error)
 }
 }
 
 
 type transformClient struct {
 type transformClient struct {
@@ -224,9 +308,19 @@ func (c *transformClient) GetUser(ctx context.Context, in *UserRequest, opts ...
 	return out, nil
 	return out, nil
 }
 }
 
 
+func (c *transformClient) ParseToken(ctx context.Context, in *TokenRequest, opts ...grpc.CallOption) (*TokenResponse, error) {
+	out := new(TokenResponse)
+	err := c.cc.Invoke(ctx, "/transform.Transform/ParseToken", in, out, opts...)
+	if err != nil {
+		return nil, err
+	}
+	return out, nil
+}
+
 // TransformServer is the server API for Transform service.
 // TransformServer is the server API for Transform service.
 type TransformServer interface {
 type TransformServer interface {
 	GetUser(context.Context, *UserRequest) (*UserResponse, error)
 	GetUser(context.Context, *UserRequest) (*UserResponse, error)
+	ParseToken(context.Context, *TokenRequest) (*TokenResponse, error)
 }
 }
 
 
 // UnimplementedTransformServer can be embedded to have forward compatible implementations.
 // UnimplementedTransformServer can be embedded to have forward compatible implementations.
@@ -236,6 +330,9 @@ type UnimplementedTransformServer struct {
 func (*UnimplementedTransformServer) GetUser(ctx context.Context, req *UserRequest) (*UserResponse, error) {
 func (*UnimplementedTransformServer) GetUser(ctx context.Context, req *UserRequest) (*UserResponse, error) {
 	return nil, status.Errorf(codes.Unimplemented, "method GetUser not implemented")
 	return nil, status.Errorf(codes.Unimplemented, "method GetUser not implemented")
 }
 }
+func (*UnimplementedTransformServer) ParseToken(ctx context.Context, req *TokenRequest) (*TokenResponse, error) {
+	return nil, status.Errorf(codes.Unimplemented, "method ParseToken not implemented")
+}
 
 
 func RegisterTransformServer(s *grpc.Server, srv TransformServer) {
 func RegisterTransformServer(s *grpc.Server, srv TransformServer) {
 	s.RegisterService(&_Transform_serviceDesc, srv)
 	s.RegisterService(&_Transform_serviceDesc, srv)
@@ -259,6 +356,24 @@ func _Transform_GetUser_Handler(srv interface{}, ctx context.Context, dec func(i
 	return interceptor(ctx, in, info, handler)
 	return interceptor(ctx, in, info, handler)
 }
 }
 
 
+func _Transform_ParseToken_Handler(srv interface{}, ctx context.Context, dec func(interface{}) error, interceptor grpc.UnaryServerInterceptor) (interface{}, error) {
+	in := new(TokenRequest)
+	if err := dec(in); err != nil {
+		return nil, err
+	}
+	if interceptor == nil {
+		return srv.(TransformServer).ParseToken(ctx, in)
+	}
+	info := &grpc.UnaryServerInfo{
+		Server:     srv,
+		FullMethod: "/transform.Transform/ParseToken",
+	}
+	handler := func(ctx context.Context, req interface{}) (interface{}, error) {
+		return srv.(TransformServer).ParseToken(ctx, req.(*TokenRequest))
+	}
+	return interceptor(ctx, in, info, handler)
+}
+
 var _Transform_serviceDesc = grpc.ServiceDesc{
 var _Transform_serviceDesc = grpc.ServiceDesc{
 	ServiceName: "transform.Transform",
 	ServiceName: "transform.Transform",
 	HandlerType: (*TransformServer)(nil),
 	HandlerType: (*TransformServer)(nil),
@@ -267,6 +382,10 @@ var _Transform_serviceDesc = grpc.ServiceDesc{
 			MethodName: "GetUser",
 			MethodName: "GetUser",
 			Handler:    _Transform_GetUser_Handler,
 			Handler:    _Transform_GetUser_Handler,
 		},
 		},
+		{
+			MethodName: "ParseToken",
+			Handler:    _Transform_ParseToken_Handler,
+		},
 	},
 	},
 	Streams:  []grpc.StreamDesc{},
 	Streams:  []grpc.StreamDesc{},
 	Metadata: "transform.proto",
 	Metadata: "transform.proto",

+ 10 - 2
transformclient/transform.go

@@ -14,11 +14,14 @@ import (
 )
 )
 
 
 type (
 type (
-	UserRequest  = transform.UserRequest
-	UserResponse = transform.UserResponse
+	TokenResponse = transform.TokenResponse
+	UserRequest   = transform.UserRequest
+	UserResponse  = transform.UserResponse
+	TokenRequest  = transform.TokenRequest
 
 
 	Transform interface {
 	Transform interface {
 		GetUser(ctx context.Context, in *UserRequest) (*UserResponse, error)
 		GetUser(ctx context.Context, in *UserRequest) (*UserResponse, error)
+		ParseToken(ctx context.Context, in *TokenRequest) (*TokenResponse, error)
 	}
 	}
 
 
 	defaultTransform struct {
 	defaultTransform struct {
@@ -36,3 +39,8 @@ func (m *defaultTransform) GetUser(ctx context.Context, in *UserRequest) (*UserR
 	client := transform.NewTransformClient(m.cli.Conn())
 	client := transform.NewTransformClient(m.cli.Conn())
 	return client.GetUser(ctx, in)
 	return client.GetUser(ctx, in)
 }
 }
+
+func (m *defaultTransform) ParseToken(ctx context.Context, in *TokenRequest) (*TokenResponse, error) {
+	client := transform.NewTransformClient(m.cli.Conn())
+	return client.ParseToken(ctx, in)
+}