Merge pull request #178 from DoubleLoong/main

add interface

Author: DoubleLoong

controllers/tp_data_services_config.go

@@ -163,3 +163,31 @@ func (c *TpDataServicesConfigController) GetData() {
 		utils.SuccessWithMessage(400, rsp_err.Error(), (*context2.Context)(c.Ctx))
 	}
 }
+
+//表名查询
+func (c *TpDataServicesConfigController) TpTable() {
+
+	var tpdataservicesconfig services.TpDataServicesConfig
+	d, rsp_err := tpdataservicesconfig.GetTableNames()
+	if rsp_err == nil {
+		utils.SuccessWithDetailed(200, "success", d, map[string]string{}, (*context2.Context)(c.Ctx))
+	} else {
+		utils.SuccessWithMessage(400, rsp_err.Error(), (*context2.Context)(c.Ctx))
+	}
+}
+
+//表字段查询
+func (c *TpDataServicesConfigController) TpTableField() {
+	reqData := valid.GetTpTableFieldValidate{}
+	if err := valid.ParseAndValidate(&c.Ctx.Input.RequestBody, &reqData); err != nil {
+		utils.SuccessWithMessage(1000, err.Error(), (*context2.Context)(c.Ctx))
+		return
+	}
+	var tpdataservicesconfig services.TpDataServicesConfig
+	d, rsp_err := tpdataservicesconfig.GetTableField(reqData.TableName)
+	if rsp_err == nil {
+		utils.SuccessWithDetailed(200, "success", d, map[string]string{}, (*context2.Context)(c.Ctx))
+	} else {
+		utils.SuccessWithMessage(400, rsp_err.Error(), (*context2.Context)(c.Ctx))
+	}
+}

routers/router.go

@@ -440,12 +440,13 @@ func init() {
 
 		web.NSRouter("/notification/history/list", &controllers.TpNotification{}, "*:HistoryList"),
 		//数据服务配置
-		//本地可视化插件
 		web.NSRouter("/tp_data_services_config/list", &controllers.TpDataServicesConfigController{}, "*:List"),
 		web.NSRouter("/tp_data_services_config/add", &controllers.TpDataServicesConfigController{}, "*:Add"),
 		web.NSRouter("/tp_data_services_config/edit", &controllers.TpDataServicesConfigController{}, "*:Edit"),
 		web.NSRouter("/tp_data_services_config/del", &controllers.TpDataServicesConfigController{}, "*:Del"),
 		web.NSRouter("/tp_data_services_config/quize", &controllers.TpDataServicesConfigController{}, "*:Quize"),
+		web.NSRouter("/tp_data_services_config/tptable", &controllers.TpDataServicesConfigController{}, "*:TpTable"),
+		web.NSRouter("/tp_data_services_config/tptablefield", &controllers.TpDataServicesConfigController{}, "*:TpTableField"),
 
 		// openapi
 		web.NSRouter("/openapi/auth/list", &controllers.OpenApiController{}, "*:List"),

services/tp_data_services_config.go

@@ -23,6 +23,13 @@ type TpDataServicesConfig struct {
 	TimeField []string
 }
 
+//表面结构体
+type TableInfo struct {
+	TableName string `json:"table_name" alias:"名称" gorm:"column:table_name"`
+	Comment   string `json:"comment" alias:"描述" gorm:"column:table_description"`
+}
+
+//获取一条数详情
 func (*TpDataServicesConfig) GetTpDataServicesConfigDetail(id string) models.TpDataServicesConfig {
 	var TpDataServicesConfig models.TpDataServicesConfig
 	psql.Mydb.First(&TpDataServicesConfig, "id = ? ", id)
@@ -37,15 +44,6 @@ func (*TpDataServicesConfig) GetTpDataServicesConfigList(PaginationValidate vali
 	if PaginationValidate.Name != "" {
 		db.Where("name like ?", "%"+PaginationValidate.Name+"%")
 	}
-	// if PaginationValidate.AppKey != "" {
-	// 	db.Where("app_key like ?", "%"+PaginationValidate.AppKey+"%")
-	// }
-	// if PaginationValidate.SignatureMode != "" {
-	// 	db.Where("SignatureMode = ?", PaginationValidate.SignatureMode)
-	// }
-	// if PaginationValidate.IpWhitelist != "" {
-	// 	db.Where("ip_whitelist like ?", "%"+PaginationValidate.IpWhitelist+"%")
-	// }
 	if PaginationValidate.EnableFlag != "" {
 		db.Where("enable_flag = ?", PaginationValidate.EnableFlag)
 	}
@@ -207,3 +205,39 @@ func (*TpDataServicesConfig) GetDataByAppkey(reqData valid.GetDataPaginationVali
 	}
 	return result, nil
 }
+
+func (*TpDataServicesConfig) GetTableNames() ([]TableInfo, error) {
+	var tableNames []TableInfo
+	sql := `SELECT table_name, obj_description (c.oid) AS table_description
+	FROM information_schema.tables t
+	LEFT JOIN pg_class c ON t.table_name = c.relname
+	WHERE table_type = 'BASE TABLE' AND table_schema = 'public'`
+	result := psql.Mydb.Raw(sql).Scan(&tableNames)
+	if result.Error != nil {
+		logs.Error(result.Error, gorm.ErrRecordNotFound)
+		return tableNames, result.Error
+	}
+	return tableNames, nil
+
+}
+func (*TpDataServicesConfig) GetTableField(table string) ([]map[string]interface{}, error) {
+	sql := `SELECT 
+    a.attname AS field,
+    format_type ( a.atttypid,a.atttypmod ) AS TYPE,
+    col_description(a.attrelid, a.attnum) AS comment
+FROM 
+    pg_class as c,
+    pg_attribute as a 
+WHERE 
+    c.relname = 'asset' 
+    AND a.attnum > 0 
+    AND a.attrelid = c.oid`
+	var result []map[string]interface{}
+	db := psql.Mydb.Raw(sql).Scan(&result)
+	if db.Error != nil {
+		logs.Error(db.Error, gorm.ErrRecordNotFound)
+		return result, db.Error
+
+	}
+	return result, nil
+}

validate/tp_data_services_config.go

@@ -68,3 +68,7 @@ type TpDataServicesConfigIdValidate struct {
 type TpDataServicesConfigQuizeValidate struct {
 	DataSql string `json:"data_sql"  alias:"Sql" valid:"Required"`
 }
+
+type GetTpTableFieldValidate struct {
+	TableName string `json:"table_name"  alias:"表名" `
+}