From 78687a8f42290a1f5755027473f146300188196b Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期六, 14 十二月 2019 18:13:08 +0800
Subject: [PATCH] add arg

---
 areaApi.go |   30 +++++++++++++++++++++++-------
 1 files changed, 23 insertions(+), 7 deletions(-)

diff --git a/areaApi.go b/areaApi.go
index e1d10ca..64fb84a 100644
--- a/areaApi.go
+++ b/areaApi.go
@@ -3,9 +3,25 @@
 import (
 	"encoding/json"
 	"fmt"
+	"strconv"
 )
 
 type AreaApi struct {
+	Ip string
+	Port int
+}
+
+func (api AreaApi) getBasicUrl() string {
+	if api.Ip == "" {
+		return BASIC_URL
+	}
+	if api.Ip == "" {
+		api.Ip = DEFAULT_IP
+	}
+	if api.Port == 0 {
+		api.Port = DEFAULT_PORT
+	}
+	return "http://"+api.Ip+":"+strconv.Itoa(api.Port)
 }
 
 //parentId 瑕佹煡瀵荤殑鐩綍鏍戠殑鏍硅妭鐐筰d
@@ -16,7 +32,7 @@
 
 	client := NewClient()
 
-	url := BASIC_URL + DATA_URL_PREFIX + "/area/localmenu"
+	url := api.getBasicUrl() + DATA_URL_PREFIX + "/area/localmenu"
 	params := make(map[string]string, 0)
 	params["parentId"] = parentId
 	params["searchType"] = searchType
@@ -47,7 +63,7 @@
 
 	client := NewClient()
 
-	url := BASIC_URL + DATA_URL_PREFIX + "/area/gb28181Tree"
+	url := api.getBasicUrl() + DATA_URL_PREFIX + "/area/gb28181Tree"
 	params := make(map[string]string, 0)
 	params["parentId"] = parentId
 	params["searchType"] = searchType
@@ -72,7 +88,7 @@
 
 func (api AreaApi) Gb28181TreeRefresh(id string) bool {
 	flag := false
-	url := BASIC_URL + DATA_URL_PREFIX + "/area/gb28181TreeRefresh"
+	url := api.getBasicUrl() + DATA_URL_PREFIX + "/area/gb28181TreeRefresh"
 	client := NewClient()
 	paramBody := map[string]interface{}{
 		"id":id,
@@ -91,7 +107,7 @@
 
 func (api AreaApi) Gb28181TreeDelete() bool {
 	flag := false
-	url := BASIC_URL + DATA_URL_PREFIX + "/area/gb28181TreeDelete"
+	url := api.getBasicUrl() + DATA_URL_PREFIX + "/area/gb28181TreeDelete"
 	client := NewClient()
 	respBody, err := client.DoPostRequest(url, CONTENT_TYPE_FORM, nil, nil, nil)
 	if err != nil {
@@ -108,7 +124,7 @@
 //娣诲姞鐩綍鑺傜偣
 func (api AreaApi) AreaAdd(paramBody map[string]interface{}) bool {
 	flag := false
-	url := BASIC_URL + DATA_URL_PREFIX + "/area/add"
+	url := api.getBasicUrl() + DATA_URL_PREFIX + "/area/add"
 	client := NewClient()
 	respBody, err := client.DoPostRequest(url, CONTENT_TYPE_FORM, paramBody, nil, nil)
 	if err != nil {
@@ -125,7 +141,7 @@
 //鐩綍鏇存柊
 func (api AreaApi) AreaUpdate(paramBody map[string]interface{}) bool {
 	flag := false
-	url := BASIC_URL + DATA_URL_PREFIX + "/area/update"
+	url := api.getBasicUrl() + DATA_URL_PREFIX + "/area/update"
 	client := NewClient()
 
 	respBody, err := client.DoPostRequest(url, CONTENT_TYPE_FORM, paramBody,nil, nil)
@@ -147,7 +163,7 @@
 	if id == "" {
 		return flag
 	}
-	url := BASIC_URL + DATA_URL_PREFIX + "/area/del"
+	url := api.getBasicUrl() + DATA_URL_PREFIX + "/area/del"
 	client := NewClient()
 	paramBody := make(map[string]interface{}, 0)
 	paramBody["id"] = id

--
Gitblit v1.8.0