go_mqtt/main.go

package main

import (
	"database/sql"
	"fmt"
	mqtt "github.com/eclipse/paho.mqtt.golang"
	_ "github.com/go-sql-driver/mysql"
	"github.com/google/uuid"
	"go_mqtt/models"
	_ "go_mqtt/models"
	"go_mqtt/mydb"
	_ "go_mqtt/mydb"
	"log"
	"strings"
	"time"
)

func getMySqlDB() (*sql.DB, error) {
	dsn := "root:Skyinno251,@tcp(47.242.184.139:3306)/appserver?charset=utf8mb4&parseTime=True&loc=Local"
	db, err := sql.Open("mysql", dsn)
	if err != nil {
		log.Fatal(err)
		return nil, err
	}
	db.SetConnMaxLifetime(time.Minute * 5) // 连接最大生命周期
	db.SetMaxOpenConns(50)                 // 最大连接数
	db.SetMaxIdleConns(10)                 // 最大空闲连接数

	//defer mydb.Close()

	err = db.Ping() // 检查连接是否成功建立。如果失败，则返回错误。
	if err != nil {
		// 处理连接失败的情况。例如，打印错误信息并退出程序。
		log.Println("47.242.184.139:3306连接失败")
		fmt.Println("Connection failed:", err)
		return nil, err
	} else {
		log.Println("47.242.184.139:3306连接成功")
		fmt.Println("Connected successfully")
		//复制代码到你的Go文件中，并根据你的实际情况修改DSN中的用户名、密码、主机名、端口和数据库名。
		//然后运行你的程序。如果一切正常，你应该能够看到查询结果被打印出来。
		//如果遇到任何问题，检查你的DSN是否正确，以及MySQL服务器是否正在运行并允许连接。
		//确保你的防火墙设置允许你的应用程序访问MySQL端口（默认是3306）。
		//如果你使用的是远程数据库，确保网络设置允许你从当前位置访问数据库服务器。
		return db, nil
	}
}

func moreDBConnect() {
	db1, err := mydb.NewDatabase("appserver", "root", "Skyinno251,", "47.242.184.139", 3306)
	if err != nil {
		fmt.Println("Error connecting to db1:", err)
		return
	}
	defer db1.Db.Close()

	fmt.Println("Connected to databases:", db1.Name)

	//db2, err := mydb.NewDatabase("db2", "user2", "password2", "localhost", 3306)
	//if err != nil {
	//	fmt.Println("Error connecting to db2:", err)
	//	return
	//}
	//defer db2.Conn.Close()
	//
	//fmt.Println("Connected to databases:", db1.Name, "and", db2.Name)
}

func queryOneDataByMySql(db *sql.DB) {
	//DB.QueryRow 用于查询单行数据，返回的是一个单独的 Row 对象。和 Query 不同，它只返回一行结果，适合用于查询根据条件唯一的数据：
	var temperature models.Temperature
	err := db.QueryRow("SELECT id,create_date,data_date,data_hour,data_minute,humidity,location_desc,temperature,topic FROM temperature order by id desc limit 0,?", 1).Scan(&temperature.Id, &temperature.CreateDate, &temperature.DataDate, &temperature.DataHour, &temperature.DataMinute, &temperature.Humidity, &temperature.LocationDesc, &temperature.Temperature, &temperature.Topic)
	if err != nil {
		fmt.Println("QueryRow failed:", err)
		return
	}
	fmt.Printf("Temperature Id:%d\n", temperature.Id)
	fmt.Printf("Temperature: %+v\n", temperature)
}

func queryMoreDataByMySql(db *sql.DB) {
	//DB.QueryRow 用于查询单行数据，返回的是一个单独的 Row 对象。和 Query 不同，它只返回一行结果，适合用于查询根据条件唯一的数据：
	rows, err := db.Query("SELECT id,create_date,data_date,data_hour,data_minute,humidity,location_desc,temperature,topic FROM temperature WHERE id > ? and id < ?", 0, 30)
	if err != nil {
		fmt.Println("Query failed:", err)
		return
	}
	defer rows.Close() // 确保在使用完毕后关闭资源

	items := make([]models.Temperature, 0)
	for rows.Next() {
		var temperature models.Temperature
		if err := rows.Scan(&temperature.Id, &temperature.CreateDate, &temperature.DataDate, &temperature.DataHour, &temperature.DataMinute, &temperature.Humidity, &temperature.LocationDesc, &temperature.Temperature, &temperature.Topic); err != nil {
			fmt.Println("Scan failed:", err)
			return
		}
		items = append(items, temperature)
		fmt.Println(items)
	}
}

// 插入单条数据
func insertDataByMySql(db *sql.DB, temperature *models.Temperature) {
	//插入操作：
	stmt, err := db.Prepare("INSERT INTO temperature(create_date,data_date,data_hour,data_minute,humidity,location_desc,temperature,topic) VALUES(?, ?,?, ?,?, ?,?, ?)")
	if err != nil {
		fmt.Println("Prepare failed:", err)
		return
	}
	res, err := stmt.Exec(&temperature.CreateDate, &temperature.DataDate, &temperature.DataHour, &temperature.DataMinute, &temperature.Humidity, &temperature.LocationDesc, &temperature.Temperature, &temperature.Topic)
	if err != nil {
		fmt.Println("Exec failed:", err)
		return
	}
	lastID, _ := res.LastInsertId()
	fmt.Printf("Inserted Temperature with ID: %d\n", lastID)
}

// 事务处理：保证数据一致性
func insertMoreDataByMySql(db *sql.DB, temperature *models.Temperature) {
	tx, err := db.Begin()
	if err != nil {
		fmt.Println("Begin transaction failed:", err)
		return
	}
	_, err = tx.Exec("UPDATE users SET age = ? WHERE id = ?", 35, 1)
	if err != nil {
		tx.Rollback() // 如果发生错误，回滚事务
		fmt.Println("Exec failed:", err)
		return
	}
	err = tx.Commit() // 提交事务
	if err != nil {
		fmt.Println("Commit failed:", err)
		return
	}
	fmt.Println("Transaction committed successfully")
}

func dbTest() {
	// 初始化数据库连接
	DB, err := sql.Open("mysql", "root:123456@tcp(127.0.0.1:3306)/test")
	if err != nil {
		fmt.Println("Error opening DB:", err)
		return
	}
	defer DB.Close()

	DB.SetConnMaxLifetime(time.Minute * 5)
	DB.SetMaxOpenConns(50)
	DB.SetMaxIdleConns(10)

	// 测试数据库连接
	if err := DB.Ping(); err != nil {
		fmt.Println("Connection failed:", err)
		return
	}
	fmt.Println("Connected successfully")

	// 查询数据
	rows, _ := DB.Query("SELECT id, name FROM users")
	defer rows.Close()

	for rows.Next() {
		var id int
		var name string
		rows.Scan(&id, &name)
		fmt.Printf("ID: %d, Name: %s\n", id, name)
	}

	// 插入数据
	stmt, _ := DB.Prepare("INSERT INTO users(name, age) VALUES(?, ?)")
	stmt.Exec("Jane Doe", 28)

	// 删除数据
	stmt, _ = DB.Prepare("DELETE FROM users WHERE id = ?")
	stmt.Exec(1)
}

func timing(client mqtt.Client) {
	//定时器，10秒钟执行一次
	ticker := time.NewTicker(5 * time.Second)
	for {
		time := <-ticker.C
		fmt.Println("定时器====>", time.String())
		if !client.IsConnectionOpen() {
			client.Connect()
		} else {
			fmt.Println("client has connect")
			//moreDBConnect()
			//db, err := getMySqlDB()
			//if err == nil {
			//	queryOneDataByMySql(db)
			//	//mydb.Close()
			//}
		}
	}
}

var messagePubHandler mqtt.MessageHandler = func(client mqtt.Client, msg mqtt.Message) {
	fmt.Printf("Received message: %s from topic: %s\n", msg.Payload(), msg.Topic())

	message := fmt.Sprintf("%s", msg.Payload())
	topic := fmt.Sprintf("%s", msg.Topic())
	fmt.Println(message)
	result := strings.Split(message, " ")
	fmt.Println(result[0])
	fmt.Println(result[1])

	now := time.Now()
	temperature := new(models.Temperature)
	temperature.CreateDate = now
	temperature.DataDate = now.Format("2006-01-02")
	temperature.DataHour = now.Format("2006-01-02 15")
	temperature.DataMinute = now.Format("2006-01-02 15:04")
	temperature.Humidity = result[0]
	temperature.Temperature = result[1]
	temperature.Topic = topic

	if topic == "WifiSHT/7C87CE9CA4E6/SHT20" {
		temperature.LocationDesc = "广东省珠海市高新区唐家湾镇东岸村水风三街28号501"
	}
	if topic == "WifiSHT/7C87CE9F5CBF/SHT20" {
		temperature.LocationDesc = "广东省珠海市金湾区三灶镇百川路1号1栋1单元1508房"
	}
	if topic == "WifiSHT/4CEBD686B6AA/SHT20" {
		temperature.LocationDesc = "广西壮族自治区崇左市天等县天等镇荣华村弄在屯113号"
	}
	fmt.Println("测温传感器地点:", temperature.LocationDesc)

	//db, err := getMySqlDB()
	//if err == nil {
	//	insertDataByMySql(db,temperature)
	//	//mydb.Close()
	//}

	models.SaveTemperature(temperature)

	//uuid := uuid.New()
	//models.SaveProduct()
	//models.SaveUser(uuid.String())
	//models.SaveOrder()

	//var users []models.User
	//var orders []models.Order
	//var user models.User
	//mydb.DB.Where("name = ?", "wenfei").First(&user)
	//fmt.Println(user)

	// 查询多条记录
	//mydb.DB.Find(&users) // 查询所有产品信息
	//mydb.DB.Order("id desc").Limit(2).Offset(0).Find(&users) // 按价格降序排序，取前 10 条记录
	//fmt.Println("users:", users)

	// 查询多条记录
	//mydb.DB.Find(&users) // 查询所有产品信息
	//mydb.DB.Where("id > ? and id < ?", 60,65).Find(&orders)
	//fmt.Println("orders:", orders)

	// 原生 SQL 查询
	//var products []models.Product
	//mydb.DB.Raw("SELECT * FROM product WHERE price > ?", 1000).Scan(&products) // 查询价格大于 1000 的产品信息
	//fmt.Println("Products:", products)

}

var connectHandler mqtt.OnConnectHandler = func(client mqtt.Client) {
	fmt.Println("Connected")
	subTemperature(client)
}

var connectLostHandler mqtt.ConnectionLostHandler = func(client mqtt.Client, err error) {
	fmt.Printf("Connect lost: %v", err)
}

func sub(client mqtt.Client) {
	topic := "gomqtt01"
	token := client.Subscribe(topic, 1, nil)
	token.Wait()
	fmt.Printf("Subscribed to topic: %s", topic)
}

func subTemperature(client mqtt.Client) {
	topic := "WifiSHT/+/SHT20"
	token := client.Subscribe(topic, 2, nil)
	token.Wait()
	fmt.Printf("Subscribed to topic: %s", topic)
}

func publish(client mqtt.Client) {
	num := 999999000000000000
	for i := 0; i < num; i++ {
		text := fmt.Sprintf("Message %d", i)
		token := client.Publish("gomqtt01", 0, false, text)
		token.Wait()
		fmt.Println(i)
		time.Sleep(6 * time.Second)
	}
}

func main() {

	uuid := uuid.New()
	fmt.Println("Generated UUID:", uuid)

	var broker = "47.242.184.139"
	var port = 1883
	opts := mqtt.NewClientOptions()
	opts.AddBroker(fmt.Sprintf("tcp://%s:%d", broker, port))
	opts.SetClientID(uuid.String())
	opts.SetUsername("admin")
	opts.SetPassword("publish452131wW452131wW$")
	opts.SetDefaultPublishHandler(messagePubHandler)
	opts.OnConnect = connectHandler
	opts.OnConnectionLost = connectLostHandler
	client := mqtt.NewClient(opts)
	if token := client.Connect(); token.Wait() && token.Error() != nil {
		fmt.Println(token.Error())
	}

	//client.IsConnectionOpen()

	go timing(client)

	//sub(client)
	subTemperature(client)
	publish(client)

	//client.Disconnect(1000)

}