update at 2023/08/14

tools/loadpconf/db.go

@@ -0,0 +1,88 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"strings"
+
+	_ "github.com/go-sql-driver/mysql"
+	"xorm.io/xorm"
+)
+
+var xEngine *xorm.Engine
+
+func XormConnectDatabaseWithUri(sql string) (*xorm.Engine, error) {
+	//	sqlStr := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&parseTime=true&loc=Local", db.User, db.Password, db.Host, db.Port, db.Name)
+	sqlStr := fmt.Sprintf("%s?charset=utf8&parseTime=true&loc=Local", sql)
+	//	fmt.Printf("sqlStr:%s:******@tcp(%s:%s)/%s?charset=utf8&parseTime=true&loc=Local", db.User, db.Host, db.Port, db.Name)
+	var err error
+	xEngine, err = xorm.NewEngine("mysql", sqlStr) //1、Create xorm engine
+	if err != nil {
+		fmt.Println("Failed to connect database:", err)
+		return nil, err
+	}
+	xEngine.ShowSQL(true)
+	return xEngine, nil
+}
+
+type ParamConfig struct {
+	//	Id         int    `json:"id" xorm:"pk 'id' autoincr"`
+	NeType     string `json:"neType"`
+	NeId       string `json:"neId"`
+	TopTag     string `json:"topTag"`
+	TopDisplay string `json:"topDisplay"`
+	ParamJson  string `json:"paramJson"`
+}
+
+func XormInsertParamConfig(mapJson *map[string]interface{}) (int64, error) {
+	var affected, a int64
+	var err error
+	paramConfig := new(ParamConfig)
+	for n, d := range *mapJson {
+		if d == nil {
+			break
+		}
+		fmt.Printf("n: %s", n)
+
+		for t, p := range d.(map[string]interface{}) {
+			if p == nil {
+				break
+			}
+			fmt.Println("t:", t)
+			fmt.Println("p:", p)
+			for k, v := range p.(map[string]interface{}) {
+				fmt.Println("k, v: ", k, v)
+				if k == "display" {
+					paramConfig.TopDisplay = fmt.Sprintf("%v", v)
+
+				} else {
+					pc, _ := json.Marshal(v)
+					paramConfig.ParamJson = fmt.Sprintf("{\"%v\":%v}", k, string(pc))
+				}
+			}
+
+			paramConfig.NeType = strings.ToUpper(n)
+			paramConfig.NeId = ""
+			paramConfig.TopTag = t
+			//			paramConfig.TopDisplay = p["display"]
+			//			paramConfig.ParamJson = p.(string)
+
+			fmt.Println("paramConfig:", paramConfig)
+
+			session := xEngine.NewSession()
+			defer session.Close()
+			_, err = session.Table("param_config").Where("ne_type = ? and top_tag = ?", paramConfig.NeType, paramConfig.TopTag).Delete()
+			if err != nil {
+				fmt.Println("Failed to delete param_config:", err)
+			}
+			a, err = session.Insert(paramConfig)
+			if err != nil {
+				fmt.Println("Failed to insert param_config:", err)
+			}
+			affected += a
+		}
+	}
+
+	return affected, err
+}

tools/loadpconf/load.go

@@ -0,0 +1,124 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"io/ioutil"
+	"os"
+
+	"gopkg.in/yaml.v3"
+)
+
+// Yaml struct of config
+type YamlConfig struct {
+	Logger struct {
+		File  string `yaml:"file"`
+		Level string `yaml:"level"`
+	}
+
+	Rest struct {
+		Addr string `yaml:"addr"`
+		Port uint16 `yaml:"port"`
+	}
+
+	Database struct {
+		Type     string `yaml:"type"`
+		User     string `yaml:"user"`
+		Password string `yaml:"password"`
+		Host     string `yaml:"host"`
+		Port     string `yaml:"port"`
+		Name     string `yaml:"name"`
+	}
+	ParamFileDir string `yaml:"paramfiledir"`
+}
+
+var yamlConfig YamlConfig
+
+func ReadConfig(configFile string) {
+	yamlFile, err := ioutil.ReadFile(configFile)
+	if err != nil {
+		fmt.Printf("ioutil.ReadFile %s err %v", configFile, err)
+	}
+	//	fmt.Println("yamlfile:", string(yamlFile))
+
+	err = yaml.Unmarshal(yamlFile, &yamlConfig)
+	if err != nil {
+		fmt.Printf("Unmarshal: %v when to struct", err)
+	}
+}
+
+func GetYamlConfig() *YamlConfig {
+	return &yamlConfig
+}
+
+var mapYaml map[string]interface{}
+
+func ReadParamConfig(fileName string) *map[string]interface{} {
+	file, err := ioutil.ReadFile(fileName)
+	if err != nil {
+		fmt.Println("yamlFile.Get err", err)
+	}
+
+	mapYaml = make(map[string]interface{})
+
+	err = yaml.Unmarshal(file, &mapYaml)
+	if err != nil {
+		fmt.Println("yaml.Unmarshal: %v when to struct", err)
+	}
+	//	fmt.Println("mapYaml:", mapYaml)
+
+	return &mapYaml
+}
+
+func GetAllFile(dir string, s []string) ([]string, error) {
+	rd, err := ioutil.ReadDir(dir)
+	if err != nil {
+		fmt.Println("read dir fail:", err)
+		return s, err
+	}
+
+	for _, fi := range rd {
+		if !fi.IsDir() {
+			fullName := dir + "/" + fi.Name()
+			s = append(s, fullName)
+		}
+	}
+	return s, nil
+}
+
+const defaultConfigFile = "./loadpconf.yaml"
+
+var (
+	version   string
+	buildTime string
+	goVer     string
+)
+
+var pfiles []string
+
+func init() {
+	cfile := flag.String("c", defaultConfigFile, "config file")
+	pfile := flag.String("p", "", "param file")
+	pv := flag.Bool("v", false, "print version")
+	ph := flag.Bool("h", false, "print help")
+
+	flag.Parse()
+	if *pv {
+		fmt.Printf("OMC initems version: %s\n%s\n%s\n\n", version, buildTime, goVer)
+		os.Exit(0)
+	}
+	if *ph {
+		flag.Usage()
+		os.Exit(0)
+	}
+
+	ReadConfig(*cfile)
+
+	fmt.Println("pfile:", *pfile)
+	if *pfile != "" {
+		pfiles = append(pfiles, *pfile)
+	} else {
+		pfiles, _ = GetAllFile(yamlConfig.ParamFileDir, pfiles)
+	}
+	fmt.Println("pfiles:", pfiles)
+}

tools/loadpconf/loadpconf.go

@@ -0,0 +1,21 @@
+package main
+
+import (
+	"fmt"
+)
+
+func main() {
+	fmt.Println("========================= OMC loadpconf startup =========================")
+	fmt.Println("OMC loadpconf version: %s %s %s", version, buildTime, goVer)
+
+	for _, f := range pfiles {
+		fmt.Println("f:", f)
+		paramMap := ReadParamConfig(f)
+		fmt.Println("paramMap:", paramMap)
+		db := yamlConfig.Database
+		sqlStr := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", db.User, db.Password, db.Host, db.Port, db.Name)
+		XormConnectDatabaseWithUri(sqlStr)
+		XormInsertParamConfig(paramMap)
+	}
+
+}

tools/loadpconf/loadpconf.yaml

@@ -0,0 +1,17 @@
+logger:
+  file: ./initems.log
+  level: trace
+
+rest:
+  addr: 0.0.0.0
+  port: 3040
+
+database:
+  type: mysql
+  user: root
+  password: 1000omc@kp!
+  host: 127.0.0.1
+  port: 33066
+  name: omc_db
+
+paramfiledir: ../../config/param

tools/loadpconf/makefile

@@ -0,0 +1,22 @@
+# Makefile for rest agent project
+
+PROJECT = OMC
+VERSION = 5GC16.1.1
+PLATFORM = amd64
+DEBDIR = ../../../deb
+ETCDIR = ../../../etc
+RELEASEDIR = ../../../release
+LIBDIR = ../lib
+BINNAME = loadpconf
+
+.PHONY: build $(BINNAME)
+build $(BINNAME):
+	go build -o $(BINNAME) -v -ldflags "-X 'main.version=$(VERSION)' \
+		-X 'main.buildTime=`date`' \
+		-X 'main.goVer=`go version`'"
+
+run: $(BINNAME)	
+	./$(BINNAME)
+
+clean:
+	rm ./$(BINNAME)