laomaDJ/backend/customer/internal/biz/cusomer.go

package biz

import (
	"context"
	"customer/api/valuation"
	"database/sql"
	"github.com/go-kratos/kratos/contrib/registry/consul/v2"
	"github.com/go-kratos/kratos/v2/transport/grpc"
	"github.com/hashicorp/consul/api"
	"gorm.io/gorm"
)

const CustomerSecret = "123456"
const CustomerDuration = 2 * 30 * 24 * 3600

type Customer struct {
	CustomerWork
	CustomerToken
	gorm.Model
}

type CustomerWork struct {
	Telephone string         `gorm:"type:varchar(15);uniqueIndex;" json:"telephone"`
	Name      sql.NullString `gorm:"type:varchar(255);uniqueIndex;" json:"name"`
	Email     sql.NullString `gorm:"type:varchar(255);uniqueIndex;" json:"email"`
	Wechat    sql.NullString `gorm:"type:varchar(255);uniqueIndex;" json:"wechat"`
	CityID    uint           `gorm:"index;" json:"city_id"`
}

type CustomerToken struct {
	Token          string       `gorm:"type:varchar(4095);" json:"token"`
	TokenCreatedAt sql.NullTime `gorm:"" json:"token_created_at"`
}

type CustomerBiz struct {
}

func NewCustomerBiz() *CustomerBiz {
	return &CustomerBiz{}
}

func (cb *CustomerBiz) GetEstimatePrice(origin, destination string) (int64, error) {
	consulConfig := api.DefaultConfig()
	consulConfig.Address = "localhost:8500"
	consulClient, err := api.NewClient(consulConfig)
	dis := consul.New(consulClient)
	if err != nil {
		return 0, err
	}
	endpoint := "discovery:///Valuation"
	conn, err := grpc.DialInsecure(
		context.Background(),
		grpc.WithEndpoint(endpoint),
		grpc.WithDiscovery(dis),
	)
	if err != nil {
		return 0, nil
	}
	defer func() {
		_ = conn.Close()
	}()
	client := valuation.NewValuationClient(conn)
	reply, err := client.GetEstimatePrice(context.Background(), &valuation.GetEstimatePriceReq{
		Origin:      origin,
		Destination: destination,
	})
	if err != nil {
		return 0, err
	}
	return reply.Price, nil
}