更新研发

dri/aliyun/company.go

@@ -1,6 +1,7 @@
 package aliyun
 
 import (
+	. "../../common"
 	"fmt"
 	"io/ioutil"
 	"net/http"
@@ -15,13 +16,94 @@ type Company struct {
 func (this *Company) Init(apiCode string) {
 	//"06c6a07e89dd45c88de040ee1489eef7"
 	this.ApiCode = apiCode
-	this.Url = "https://api.81api.com"
+	this.Url = "http://api.81api.com"
 }
 
 // GetCompanyBaseInfo 获取企业基础信息
-func (this *Company) GetCompanyBaseInfo(name string) (string, error) {
+func (this *Company) GetCompanyOtherAll(name string) Map {
+
+	res := Map{}
+	data, e := this.GetCompanyPatentsInfo(name) //获取专利信息
+	if e != nil {
+		fmt.Println(e)
+	} else {
+		res["PatentsInfo"] = data.GetMap("data")
+	}
+	data, e = this.GetCompanyOtherCopyrightsInfo(name) //获取其他专利
+	if e != nil {
+		fmt.Println(e)
+	} else {
+		res["OtherCopyrightsInfo"] = data.GetMap("data")
+	}
+	data, e = this.GetCompanyTrademarksInfo(name) //获取商标
+	if e != nil {
+		fmt.Println(e)
+	} else {
+		res["TrademarksInfo"] = data.GetMap("data")
+	}
+	data, e = this.GetCompanySoftwareCopyrightsInfo(name) //获取软著
+	if e != nil {
+		fmt.Println(e)
+	} else {
+		res["SoftwareCopyrightsInfo"] = data.GetMap("data")
+	}
+	data, e = this.GetCompanyProfileTags(name) //获取大数据标签
+	if e != nil {
+		fmt.Println(e)
+	} else {
+		res["ProfileTags"] = data.GetMap("data")
+	}
+	return res
+}
+
+// GetCompanyBaseInfo 获取企业基础信息
+func (this *Company) GetCompanyBaseInfo(name string) (Map, error) {
 	url := "/getCompanyBaseInfo/"
 
+	body, err := this.basePost(url, name)
+	return ObjToMap(body), err
+}
+
+// GetCompanyPatentsInfo  获取专利信息
+func (this *Company) GetCompanyPatentsInfo(name string) (Map, error) {
+	url := "/getCompanyPatentsInfo/"
+
+	body, err := this.basePost(url, name)
+	return ObjToMap(body), err
+}
+
+//   获取商标信息
+func (this *Company) GetCompanyTrademarksInfo(name string) (Map, error) {
+	url := "/getCompanyTrademarksInfo/"
+
+	body, err := this.basePost(url, name)
+	return ObjToMap(body), err
+}
+
+//   获取软著信息
+func (this *Company) GetCompanySoftwareCopyrightsInfo(name string) (Map, error) {
+	url := "/getCompanySoftwareCopyrightsInfo/"
+
+	body, err := this.basePost(url, name)
+	return ObjToMap(body), err
+}
+
+//   获取其他著作信息
+func (this *Company) GetCompanyOtherCopyrightsInfo(name string) (Map, error) {
+	url := "/getCompanyOtherCopyrightsInfo/"
+
+	body, err := this.basePost(url, name)
+	return ObjToMap(body), err
+}
+
+//   获取大数据标签
+func (this *Company) GetCompanyProfileTags(name string) (Map, error) {
+	url := "/getCompanyProfileTags/"
+	body, err := this.basePost(url, name)
+	return ObjToMap(body), err
+}
+func (this *Company) basePost(url string, name string) (string, error) {
+
 	client := &http.Client{}
 
 	reqest, err := http.NewRequest("GET", this.Url+url+name+"/?isRaiseErrorCode=1", nil)
@@ -44,11 +126,9 @@ func (this *Company) GetCompanyBaseInfo(name string) (string, error) {
 	if err != nil {
 		return "", err
 	}
-
-	fmt.Println(string(body))
-
-	return string(body), err
-
+	res := string(body)
+	fmt.Println(res)
+	return res, err
 }
 
 var DefaultCompany Company

dri/tencent/company.go

@@ -0,0 +1,101 @@
+package tencent
+
+import (
+	. "../../common"
+	"crypto/hmac"
+	"crypto/sha1"
+	"encoding/base64"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"net/http"
+	gourl "net/url"
+	"strings"
+	"time"
+)
+
+func calcAuthorization(source string, secretId string, secretKey string) (auth string, datetime string, err error) {
+	timeLocation, _ := time.LoadLocation("Etc/GMT")
+	datetime = time.Now().In(timeLocation).Format("Mon, 02 Jan 2006 15:04:05 GMT")
+	signStr := fmt.Sprintf("x-date: %s\nx-source: %s", datetime, source)
+
+	// hmac-sha1
+	mac := hmac.New(sha1.New, []byte(secretKey))
+	mac.Write([]byte(signStr))
+	sign := base64.StdEncoding.EncodeToString(mac.Sum(nil))
+
+	auth = fmt.Sprintf("hmac id=\"%s\", algorithm=\"hmac-sha1\", headers=\"x-date x-source\", signature=\"%s\"",
+		secretId, sign)
+
+	return auth, datetime, nil
+}
+
+func urlencode(params map[string]string) string {
+	var p = gourl.Values{}
+	for k, v := range params {
+		p.Add(k, v)
+	}
+	return p.Encode()
+}
+
+func GetCompany(secretId, secretKey, name string) Map {
+	// 云市场分配的密钥Id
+	//secretId := "xxxx"
+	//// 云市场分配的密钥Key
+	//secretKey := "xxxx"
+	source := "market"
+
+	// 签名
+	auth, datetime, _ := calcAuthorization(source, secretId, secretKey)
+
+	// 请求方法
+	method := "GET"
+	// 请求头
+	headers := map[string]string{"X-Source": source, "X-Date": datetime, "Authorization": auth}
+
+	// 查询参数
+	queryParams := make(map[string]string)
+	queryParams["keyword"] = name
+	// body参数
+	bodyParams := make(map[string]string)
+
+	// url参数拼接
+	url := "https://service-3jnh3ku8-1256140209.gz.apigw.tencentcs.com/release/business4/geet"
+	if len(queryParams) > 0 {
+		url = fmt.Sprintf("%s?%s", url, urlencode(queryParams))
+	}
+
+	bodyMethods := map[string]bool{"POST": true, "PUT": true, "PATCH": true}
+	var body io.Reader = nil
+	if bodyMethods[method] {
+		body = strings.NewReader(urlencode(bodyParams))
+		headers["Content-Type"] = "application/x-www-form-urlencoded"
+	}
+
+	client := &http.Client{
+		Timeout: 5 * time.Second,
+	}
+	request, err := http.NewRequest(method, url, body)
+	if err != nil {
+		fmt.Println(err)
+		return nil
+	}
+	for k, v := range headers {
+		request.Header.Set(k, v)
+	}
+	response, err := client.Do(request)
+	if err != nil {
+		fmt.Println(err)
+		return nil
+	}
+	defer response.Body.Close()
+
+	bodyBytes, err := ioutil.ReadAll(response.Body)
+	if err != nil {
+		fmt.Println(err)
+		return nil
+	}
+	res := string(bodyBytes)
+	fmt.Println(res)
+	return ObjToMap(res)
+}