-
Notifications
You must be signed in to change notification settings - Fork 0
/
localdata.go
115 lines (98 loc) · 2.4 KB
/
localdata.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
package rabbitmqpool
import (
"fmt"
"os"
"strings"
"sync"
"time"
)
// const filePath string = "data.txt"
var mutex sync.Mutex
func writeToLocalFile(data string, filePath string) error {
mutex.Lock()
defer mutex.Unlock()
file, err := os.OpenFile(filePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
fmt.Println("Failed open file: ", filePath)
return err
}
defer file.Close()
_, err = file.WriteString(data + "\n")
if err != nil {
fmt.Println("Write to file failed: ", err)
return err
}
// encoder := json.NewEncoder(file)
// if err := encoder.Encode(data); err != nil {
// return err
// }
return nil
}
func TmpMain() {
filePath := "localdata.txt"
readAndSendData(filePath)
}
func readAndSendData(filePath string) {
for {
time.Sleep(30 * time.Second)
mutex.Lock()
data, err := os.ReadFile(filePath)
if err != nil {
mutex.Unlock()
fmt.Println("Error reading file:", err)
continue
}
mutex.Unlock()
lines := splitLines(string(data))
for _, line := range lines {
// Simulate sending data (replace with actual sending logic)
success := sendData(line)
if success {
removeLineFromFile(line, filePath)
fmt.Println("Data sent successfully and line removed:", line)
} else {
fmt.Println("Failed to send data:", line)
}
}
}
}
func splitLines(input string) []string {
return strings.Split(input, "\n")
}
func sendData(data string) bool {
// Replace this with actual sending logic
// If sending is successful, return true; otherwise, return false
fmt.Println("data content is ", data)
return true
}
func removeLineFromFile(lineToRemove string, filePath string) {
mutex.Lock()
defer mutex.Unlock()
input, err := os.ReadFile(filePath)
if err != nil {
fmt.Println("Error reading file for removing line:", err)
return
}
lines := splitLines(string(input))
var output []string
for _, line := range lines {
if line != lineToRemove {
output = append(output, line)
}
}
err = os.WriteFile(filePath, []byte(strings.Join(output, "\n")), 0644)
if err != nil {
fmt.Println("Error writing file after removing line:", err)
}
}
// func testmain() {
// // Example usage
// err := writeToLocalFile("Data to be written")
// if err != nil {
// log.Fatal("Error writing to file:", err)
// }
// // Start reading and sending data from the file in the background
// go readAndSendData()
// // Keep the program running
// select {}
// }