add shorturl example code

example/shorturl/rpc/transform/internal/logic/expandlogic.go

@@ -0,0 +1,35 @@
+package logic
+
+import (
+	"context"
+
+	"shorturl/rpc/transform/internal/svc"
+	transform "shorturl/rpc/transform/pb"
+
+	"github.com/tal-tech/go-zero/core/logx"
+)
+
+type ExpandLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewExpandLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ExpandLogic {
+	return &ExpandLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *ExpandLogic) Expand(in *transform.ExpandReq) (*transform.ExpandResp, error) {
+	res, err := l.svcCtx.Model.FindOne(in.Shorten)
+	if err != nil {
+		return nil, err
+	}
+
+	return &transform.ExpandResp{
+		Url: res.Url,
+	}, nil
+}

example/shorturl/rpc/transform/internal/logic/shortenlogic.go

@@ -0,0 +1,41 @@
+package logic
+
+import (
+	"context"
+
+	"shorturl/rpc/transform/internal/svc"
+	"shorturl/rpc/transform/model"
+	transform "shorturl/rpc/transform/pb"
+
+	"github.com/tal-tech/go-zero/core/hash"
+	"github.com/tal-tech/go-zero/core/logx"
+)
+
+type ShortenLogic struct {
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+	logx.Logger
+}
+
+func NewShortenLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ShortenLogic {
+	return &ShortenLogic{
+		ctx:    ctx,
+		svcCtx: svcCtx,
+		Logger: logx.WithContext(ctx),
+	}
+}
+
+func (l *ShortenLogic) Shorten(in *transform.ShortenReq) (*transform.ShortenResp, error) {
+	key := hash.Md5Hex([]byte(in.Url))[:6]
+	_, err := l.svcCtx.Model.Insert(model.Shorturl{
+		Shorten: key,
+		Url:     in.Url,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &transform.ShortenResp{
+		Shorten: key,
+	}, nil
+}